add close in test (#282)

vesoft-inc · Mar 26, 2021 · 55dd09a · 55dd09a
1 parent 9531f4a
commit 55dd09a
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/client/src/test/java/com/vesoft/nebula/client/storage/MockStorageData.java b/client/src/test/java/com/vesoft/nebula/client/storage/MockStorageData.java
@@ -53,6 +53,8 @@ public static void initGraph() {
                 | IOErrorException | AuthFailedException
                 | UnsupportedEncodingException e) {
             e.printStackTrace();
+        } finally {
+            pool.close();
         }
     }
 

diff --git a/client/src/test/java/com/vesoft/nebula/client/storage/StorageClientTest.java b/client/src/test/java/com/vesoft/nebula/client/storage/StorageClientTest.java
@@ -18,13 +18,14 @@
 import java.io.UnsupportedEncodingException;
 import java.util.Arrays;
 import java.util.List;
+import org.junit.After;
 import org.junit.Before;
 import org.junit.Test;
 
 public class StorageClientTest {
 
     private final String ip = "127.0.0.1";
-    private StorageClient client;
+    private StorageClient client = null;
 
     @Before
     public void before() {
@@ -39,6 +40,12 @@ public void before() {
         client = new StorageClient(address);
     }
 
+    @After
+    public void after() {
+        if (client != null) {
+            client.close();
+        }
+    }
 
     @Test
     public void testScanVertexWithNoCol() {