Add tests for CallProcedure and CallProcedureAsync

redis · Jul 27, 2023 · 89c039f · 89c039f
1 parent f9b8acb
commit 89c039f
Showing 1 changed file with 86 additions and 0 deletions.
diff --git a/tests/NRedisStack.Tests/Graph/GraphTests.cs b/tests/NRedisStack.Tests/Graph/GraphTests.cs
@@ -956,6 +956,49 @@ public void TestModulePrefixs()
         Assert.NotEqual(graph1.GetHashCode(), graph2.GetHashCode());
     }
 
+    [Fact]
+    public void TestCallProcedureDbLabels()
+    {
+        var db = redisFixture.Redis.GetDatabase();
+        db.Execute("FLUSHALL");
+
+        const string graphName = "social";
+
+        var graph = db.GRAPH();
+        // Create empty graph, otherwise call procedure will throw exception
+        graph.Query(graphName, "RETURN 1"); 
+
+        var labels0 = graph.CallProcedure(graphName, "db.labels");
+        Assert.Empty(labels0);
+
+        graph.Query(graphName, "CREATE (:Person { name: 'Bob' })");
+
+        var labels1 = graph.CallProcedure(graphName, "db.labels");
+        Assert.Single(labels1);
+    }
+
+    [Fact]
+    public void TestCallProcedureReadOnly()
+    {
+        var db = redisFixture.Redis.GetDatabase();
+        db.Execute("FLUSHALL");
+
+        const string graphName = "social";
+
+        var graph = db.GRAPH();
+        // throws RedisServerException when executing a ReadOnly procedure against non-existing graph.
+        Assert.Throws<RedisServerException>(() => graph.CallProcedure(graphName, "db.labels", ProcedureMode.Read));
+
+        graph.Query(graphName, "CREATE (:Person { name: 'Bob' })");
+        var procedureArgs = new List<string>
+        {
+            "Person",
+            "name"
+        };
+        // throws RedisServerException when executing a Write procedure with Read procedure mode.
+        Assert.Throws<RedisServerException>(() => graph.CallProcedure(graphName, "db.idx.fulltext.createNodeIndex", procedureArgs, ProcedureMode.Read));
+    }
+
     #endregion
 
     #region AsyncTests
@@ -1886,6 +1929,49 @@ public async Task TestModulePrefixsAsync()
         Assert.NotEqual(graph1.GetHashCode(), graph2.GetHashCode());
     }
 
+    [Fact]
+    public async Task TestCallProcedureDbLabelsAsync()
+    {
+        var db = redisFixture.Redis.GetDatabase();
+        db.Execute("FLUSHALL");
+
+        const string graphName = "social";
+
+        var graph = db.GRAPH();
+        // Create empty graph, otherwise call procedure will throw exception
+        await graph.QueryAsync(graphName, "RETURN 1"); 
+
+        var labels0 = await graph.CallProcedureAsync(graphName, "db.labels");
+        Assert.Empty(labels0);
+
+        await graph.QueryAsync(graphName, "CREATE (:Person { name: 'Bob' })");
+
+        var labels1 = await graph.CallProcedureAsync(graphName, "db.labels");
+        Assert.Single(labels1);
+    }
+
+    [Fact]
+    public async Task TestCallProcedureReadOnlyAsync()
+    {
+        var db = redisFixture.Redis.GetDatabase();
+        db.Execute("FLUSHALL");
+
+        const string graphName = "social";
+
+        var graph = db.GRAPH();
+        // throws RedisServerException when executing a ReadOnly procedure against non-existing graph.
+        await Assert.ThrowsAsync<RedisServerException>(() => graph.CallProcedureAsync(graphName, "db.labels", ProcedureMode.Read));
+
+        await graph.QueryAsync(graphName, "CREATE (:Person { name: 'Bob' })");
+        var procedureArgs = new List<string>
+        {
+            "Person",
+            "name"
+        };
+        // throws RedisServerException when executing a Write procedure with Read procedure mode.
+        await Assert.ThrowsAsync<RedisServerException>(() => graph.CallProcedureAsync(graphName, "db.idx.fulltext.createNodeIndex", procedureArgs, ProcedureMode.Read));
+    }
+
     [Fact]
     public void TestParseInfinity()
     {