fix test

vesoft-inc · Oct 19, 2021 · 50ef23e · 50ef23e
1 parent 7758b91
commit 50ef23e
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/...spark-connector/src/test/scala/com/vesoft/nebula/connector/nebula/GraphProviderTest.scala b/...spark-connector/src/test/scala/com/vesoft/nebula/connector/nebula/GraphProviderTest.scala
@@ -34,7 +34,7 @@ class GraphProviderTest extends AnyFunSuite with BeforeAndAfterAll {
   }
 
   test("submit") {
-    val result = graphProvider.submit("insert vertex person(col1) values 100:(\"test\")")
+    val result = graphProvider.submit("fetch prop on person 1")
     assert(result.isSucceeded)
   }
 }
diff --git a/nebula-spark-connector/src/test/scala/com/vesoft/nebula/connector/reader/ReadSuite.scala b/nebula-spark-connector/src/test/scala/com/vesoft/nebula/connector/reader/ReadSuite.scala
@@ -197,7 +197,7 @@ class ReadSuite extends AnyFunSuite with BeforeAndAfterAll {
     val nebulaReadEdgeConfig: ReadNebulaConfig = ReadNebulaConfig
       .builder()
       .withSpace("test_int")
-      .withLabel("person")
+      .withLabel("friend")
       .withNoColumn(false)
       .withReturnCols(List("col1"))
       .withLimit(10)

diff --git a/...-spark-connector/src/test/scala/com/vesoft/nebula/connector/writer/WriteDeleteSuite.scala b/...-spark-connector/src/test/scala/com/vesoft/nebula/connector/writer/WriteDeleteSuite.scala
@@ -32,7 +32,7 @@ class WriteDeleteSuite extends AnyFunSuite with BeforeAndAfterAll {
     val resultSet: ResultSet =
       graphProvider.submit("use test_write_string;match (v:person_connector) return v;")
     assert(resultSet.getColumnNames.size() == 0)
-    assert(resultSet.getRows.size() == 0)
+    assert(resultSet.isEmpty)
   }
 
   test("write edge into test_write_string space with delete mode") {
@@ -44,6 +44,6 @@ class WriteDeleteSuite extends AnyFunSuite with BeforeAndAfterAll {
     val resultSet: ResultSet =
       graphProvider.submit("use test_write_string;fetch prop on friend_connector 1->2@10")
     assert(resultSet.getColumnNames.size() == 0)
-    assert(resultSet.getRows.size() == 0)
+    assert(resultSet.isEmpty)
   }
 }