feat: remove db after test

framework/src/test/java/org/tron/common/overlay/discover/node/NodeHandlerTest.java

@@ -9,24 +9,21 @@
 import org.junit.Test;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
-import org.tron.common.application.Application;
 import org.tron.common.application.TronApplicationContext;
-import org.tron.common.parameter.CommonParameter;
 import org.tron.common.utils.FileUtil;
 import org.tron.core.ChainBaseManager;
 import org.tron.core.Constant;
 import org.tron.core.config.DefaultConfig;
 import org.tron.core.config.args.Args;
-import org.tron.core.db.Manager;
 
 
 public class NodeHandlerTest {
 
   private static final Logger logger = LoggerFactory.getLogger("Test");
-  private static Manager dbManager;
+  // private static Manager dbManager;
   private static TronApplicationContext context;
-  private Application appTest;
-  private CommonParameter argsTest;
+  // private Application appTest;
+  // private CommonParameter argsTest;
   private static Node currNode;
   private static Node oldNode;
   private static Node replaceNode;
@@ -67,7 +64,7 @@ public static void destroy() {
    * init nodes.
    */
   public static void initNodes() {
-    dbManager = context.getBean(Manager.class);
+    // dbManager = context.getBean(Manager.class);
     nodeManager = new NodeManager(context.getBean(ChainBaseManager.class));
     String currNodeId = "74c11ffad1d59d7b1a56691a0b84a53f0791c92361357364f1d2537"
         + "898407ef0249bbbf5a4ce8cff9e34e2fdf8bac883540e026d1e5d6ebf536414bdde81198e";
@@ -97,7 +94,7 @@ public void stateNonActiveTest() throws Exception {
     nh.changeState(NodeHandler.State.NONACTIVE);
     replaceHandler.changeState(NodeHandler.State.ALIVE);
 
-    Assert.assertTrue(!nodeManager.getTable().contains(oldNode));
+    Assert.assertFalse(nodeManager.getTable().contains(oldNode));
     Assert.assertTrue(nodeManager.getTable().contains(replaceNode));
   }
 }

framework/src/test/java/org/tron/common/overlay/discover/node/NodeManagerTest.java

@@ -79,15 +79,16 @@ public static void destroy() {
   // @Before
   public static void initManager() throws Exception {
     nodeManagerClazz = NodeManager.class;
-    Constructor<NodeManager> handlerConstructor
-        = nodeManagerClazz.getConstructor(ChainBaseManager.class);
+    // Constructor<NodeManager> handlerConstructor
+    //     = nodeManagerClazz.getConstructor(ChainBaseManager.class);
     manager = context.getBean(Manager.class);
-    nodeManager = handlerConstructor.newInstance(context.getBean(ChainBaseManager.class));
+    // nodeManager = handlerConstructor.newInstance(context.getBean(ChainBaseManager.class));
+    nodeManager = new NodeManager(context.getBean(ChainBaseManager.class));
   }
 
   @Test
   public void isNodeAliveTest() {
-    Node node = new Node(new byte[64], "128.0.0.1", 18888, 18888);
+    Node node = new Node(new byte[64], "128.0.0.1", 18889, 18889);
     nodeManager.getTable().addNode(node);
     NodeHandler nodeHandler = new NodeHandler(node, nodeManager);
     nodeHandler.changeState(NodeHandler.State.ACTIVE);
@@ -167,7 +168,7 @@ public int insertValues(int totalNodes) throws Exception {
       Class nodeClazz = Node.class;
       Constructor<Node> nodeConstructor
           = nodeClazz.getConstructor(byte[].class, String.class, int.class, int.class);
-      Node node = nodeConstructor.newInstance(bytes, stringBuilder.toString(), 18888, 18888);
+      Node node = nodeConstructor.newInstance(bytes, stringBuilder.toString(), 18889, 18889);
       Field isConnectableField = nodeClazz.getDeclaredField("p2pVersion");
       isConnectableField.setAccessible(true);
       isConnectableField.set(node, Args.getInstance().getNodeP2pVersion());
@@ -183,10 +184,10 @@ public void insertNotConnectibleNodes() throws Exception {
     Class nodeClazz = Node.class;
     Constructor<Node> nodeConstructor
         = nodeClazz.getConstructor(byte[].class, String.class, int.class, int.class);
-    Node wrongNode1 = nodeConstructor.newInstance(new byte[64], "128.0.0.1", 1111, 18888);
+    Node wrongNode1 = nodeConstructor.newInstance(new byte[64], "128.0.0.1", 1111, 18889);
     byte[] id = new byte[64];
     id[63] = 1;
-    Node wrongNode2 = nodeConstructor.newInstance(id, "128.0.0.2", 1111, 18888);
+    Node wrongNode2 = nodeConstructor.newInstance(id, "128.0.0.2", 1111, 18889);
     Field isConnectableField = nodeClazz.getDeclaredField("p2pVersion");
     isConnectableField.setAccessible(true);
     isConnectableField.set(wrongNode1, 999);
@@ -210,9 +211,9 @@ public int getHandlerMapSize() throws Exception {
 
   @Test
   public void dumpActiveNodesTest() {
-    Node node1 = new Node(new byte[64], "128.0.0.1", 18888, 18888);
-    Node node2 = new Node(new byte[64], "128.0.0.2", 18888, 18888);
-    Node node3 = new Node(new byte[64], "128.0.0.3", 18888, 18888);
+    Node node1 = new Node(new byte[64], "128.0.0.1", 18889, 18889);
+    Node node2 = new Node(new byte[64], "128.0.0.2", 18889, 18889);
+    Node node3 = new Node(new byte[64], "128.0.0.3", 18889, 18889);
     NodeHandler nodeHandler1 = nodeManager.getNodeHandler(node1);
     NodeHandler nodeHandler2 = nodeManager.getNodeHandler(node2);
     NodeHandler nodeHandler3 = nodeManager.getNodeHandler(node3);

framework/src/test/java/org/tron/common/runtime/vm/ValidateMultiSignContractTest.java

@@ -1,12 +1,14 @@
 package org.tron.common.runtime.vm;
 
 import com.google.protobuf.ByteString;
+import java.io.File;
 import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.List;
 import lombok.extern.slf4j.Slf4j;
 import org.apache.commons.lang3.tuple.Pair;
 import org.bouncycastle.util.encoders.Hex;
+import org.junit.AfterClass;
 import org.junit.Test;
 import org.testng.Assert;
 import org.tron.common.application.Application;
@@ -17,6 +19,7 @@
 import org.tron.common.parameter.CommonParameter;
 import org.tron.common.utils.ByteArray;
 import org.tron.common.utils.ByteUtil;
+import org.tron.common.utils.FileUtil;
 import org.tron.common.utils.Sha256Hash;
 import org.tron.common.utils.StringUtil;
 import org.tron.core.Constant;
@@ -55,6 +58,20 @@ public class ValidateMultiSignContractTest {
 
   ValidateMultiSign contract = new ValidateMultiSign();
 
+  /**
+   * Release resources.
+   */
+  @AfterClass
+  public static void destroy() {
+    Args.clearParam();
+    context.destroy();
+    if (FileUtil.deleteDir(new File(dbPath))) {
+      logger.info("Release resources successful.");
+    } else {
+      logger.info("Release resources failure.");
+    }
+  }
+
   @Test
   public void testAddressNonExist() {
     byte[] hash = Hash.sha3(longData);

framework/src/test/java/org/tron/common/storage/leveldb/RocksDbDataSourceImplTest.java

@@ -18,6 +18,7 @@
 import org.junit.AfterClass;
 import org.junit.Assert;
 import org.junit.Before;
+import org.junit.BeforeClass;
 import org.junit.Test;
 import org.tron.common.storage.rocksdb.RocksDbDataSourceImpl;
 import org.tron.common.utils.ByteArray;
@@ -49,16 +50,23 @@ public class RocksDbDataSourceImplTest {
    */
   @AfterClass
   public static void destroy() {
+    String directory = Args.getInstance().getStorage().getDbDirectory();
     Args.clearParam();
     if (FileUtil.deleteDir(new File(dbPath))) {
       logger.info("Release resources successful.");
     } else {
       logger.info("Release resources failure.");
     }
+
+    if (FileUtil.deleteDir(new File(dbPath  + directory))) {
+      logger.info("Release resources successful.");
+    } else {
+      logger.info("Release resources failure.");
+    }
   }
 
-  @Before
-  public void initDb() {
+  @BeforeClass
+  public static void initDb() {
     Args.setParam(new String[]{"--output-directory", dbPath}, "config-test-dbbackup.conf");
     dataSourceTest = new RocksDbDataSourceImpl(dbPath + File.separator, "test_rocksDb");
   }

framework/src/test/java/org/tron/core/services/ProposalServiceTest.java

@@ -6,9 +6,10 @@
 import java.io.File;
 import java.util.HashSet;
 import java.util.Set;
-import org.junit.After;
+import lombok.extern.slf4j.Slf4j;
+import org.junit.AfterClass;
 import org.junit.Assert;
-import org.junit.Before;
+import org.junit.BeforeClass;
 import org.junit.Test;
 import org.tron.common.application.TronApplicationContext;
 import org.tron.common.utils.FileUtil;
@@ -21,14 +22,15 @@
 import org.tron.core.utils.ProposalUtil.ProposalType;
 import org.tron.protos.Protocol.Proposal;
 
+@Slf4j
 public class ProposalServiceTest {
 
-  private TronApplicationContext context;
-  private Manager manager;
-  private String dbPath = "output_proposal_test";
+  private static TronApplicationContext context;
+  private static Manager manager;
+  private static String dbPath = "output_proposal_test";
 
-  @Before
-  public void init() {
+  @BeforeClass
+  public static void init() {
     Args.setParam(new String[]{"-d", dbPath}, Constant.TEST_CONF);
     context = new TronApplicationContext(DefaultConfig.class);
     manager = context.getBean(Manager.class);
@@ -82,11 +84,14 @@ public void testUpdateEnergyFee() {
         currentHistory);
   }
 
-
-  @After
-  public void removeDb() {
+  @AfterClass
+  public static void removeDb() {
     Args.clearParam();
     context.destroy();
-    FileUtil.deleteDir(new File(dbPath));
+    if (FileUtil.deleteDir(new File(dbPath))) {
+      logger.info("Release resources successful.");
+    } else {
+      logger.info("Release resources failure.");
+    }
   }
 }