fix test: testCreateGraphsWithMultiDisksForRocksDB

javeme · javeme · commit 68b6de5c3f61 · 2022-06-14T00:15:54.000+08:00
Change-Id: Ia0069c683a61ff9cbcb312aeb75d6622bc0e1003
diff --git a/hugegraph-test/src/main/java/com/baidu/hugegraph/core/MultiGraphsTest.java b/hugegraph-test/src/main/java/com/baidu/hugegraph/core/MultiGraphsTest.java
@@ -24,7 +24,6 @@
 import java.util.Iterator;
 import java.util.List;
 
-import com.google.common.collect.ImmutableList;
 import org.apache.commons.configuration2.BaseConfiguration;
 import org.apache.commons.configuration2.Configuration;
 import org.apache.commons.configuration2.PropertiesConfiguration;
@@ -36,11 +35,11 @@
 
 import com.baidu.hugegraph.HugeException;
 import com.baidu.hugegraph.HugeGraph;
-import com.baidu.hugegraph.backend.BackendException;
 import com.baidu.hugegraph.backend.id.IdGenerator;
 import com.baidu.hugegraph.backend.store.BackendStoreInfo;
 import com.baidu.hugegraph.backend.store.rocksdb.RocksDBOptions;
 import com.baidu.hugegraph.config.CoreOptions;
+import com.baidu.hugegraph.exception.ConnectionException;
 import com.baidu.hugegraph.exception.ExistedException;
 import com.baidu.hugegraph.schema.EdgeLabel;
 import com.baidu.hugegraph.schema.IndexLabel;
@@ -50,6 +49,7 @@
 import com.baidu.hugegraph.testutil.Assert;
 import com.baidu.hugegraph.testutil.Utils;
 import com.baidu.hugegraph.type.define.NodeRole;
+import com.google.common.collect.ImmutableList;
 
 public class MultiGraphsTest {
 
@@ -333,7 +333,7 @@ public void testCreateGraphsWithMultiDisksForRocksDB() {
         g1.clearBackend();
 
         final HugeGraph[] g2 = new HugeGraph[1];
-        Assert.assertThrows(BackendException.class, () -> {
+        Assert.assertThrows(ConnectionException.class, () -> {
             g2[0] = openGraphWithBackend("g2", "rocksdb", "binary",
                                          "rocksdb.data_disks",
                                          "[g/range_int_index:rocksdb-index1]");
@@ -348,7 +348,7 @@ public void testCreateGraphsWithMultiDisksForRocksDB() {
         });
 
         final HugeGraph[] g3 = new HugeGraph[1];
-        Assert.assertThrows(BackendException.class, () -> {
+        Assert.assertThrows(ConnectionException.class, () -> {
             g3[0] = openGraphWithBackend("g3", "rocksdb", "binary",
                                          "rocksdb.data_disks",
                                          "[g/secondary_index:/]");
