throw error for failed use space execution (#63)

* throw error for failed use space execution * upload logs * add sleep after create schema
vesoft-inc · Oct 24, 2022 · 44015bb · 44015bb
1 parent 1091639
commit 44015bb
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 2 deletions.
diff --git a/nebula-spark-common/src/main/scala/com/vesoft/nebula/connector/nebula/GraphProvider.scala b/nebula-spark-common/src/main/scala/com/vesoft/nebula/connector/nebula/GraphProvider.scala
@@ -90,7 +90,11 @@ class GraphProvider(addresses: List[Address],
     val switchStatment = s"use $space"
     LOG.info(s"switch space $space")
     val result = submit(switchStatment)
-    result.isSucceeded
+    if(!result.isSucceeded){
+      LOG.error(s"switch space $space failed, ${result.getErrorMessage}")
+      throw new RuntimeException(s"switch space $space failed, ${result.getErrorMessage}")
+    }
+    true
   }
 
   /**

diff --git a/...la-spark-common/src/test/scala/com/vesoft/nebula/connector/nebula/GraphProviderTest.scala b/...la-spark-common/src/test/scala/com/vesoft/nebula/connector/nebula/GraphProviderTest.scala
@@ -30,7 +30,7 @@ class GraphProviderTest extends AnyFunSuite with BeforeAndAfterAll {
   }
 
   test("switchSpace") {
-    assert(!graphProvider.switchSpace("root", "nebula", "not_exit_space"))
+    assertThrows[RuntimeException](graphProvider.switchSpace("root", "nebula", "space_not_exist"))
     assert(graphProvider.switchSpace("root", "nebula", "test_int"))
   }
 

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
@@ -21,6 +21,7 @@ class WriteDeleteSuite extends AnyFunSuite with BeforeAndAfterAll {
     graphMock.mockStringIdGraphSchema()
     graphMock.mockIntIdGraphSchema()
     graphMock.close()
+    Thread.sleep(10000)
     SparkMock.writeVertex()
     SparkMock.writeEdge()
   }

diff --git a/...-spark-connector/src/test/scala/com/vesoft/nebula/connector/writer/WriteInsertSuite.scala b/...-spark-connector/src/test/scala/com/vesoft/nebula/connector/writer/WriteInsertSuite.scala
@@ -21,6 +21,7 @@ class WriteInsertSuite extends AnyFunSuite with BeforeAndAfterAll {
     graphMock.mockStringIdGraphSchema()
     graphMock.mockIntIdGraphSchema()
     graphMock.close()
+    Thread.sleep(10000)
   }
 
   test("write vertex into test_write_string space with insert mode") {