Merge pull request #5080 from lurais/feature/auto_stop_opt

fix(test): add wallet unit test

Author: halibobo1205

framework/src/test/java/org/tron/core/WalletTest.java

@@ -1066,5 +1066,17 @@ public void testEstimateEnergyOutOfTime() {
       Assert.assertTrue(true);
     }
   }
+
+  @Test
+  public void testListNodes() {
+    try {
+      GrpcAPI.NodeList nodeList = wallet.listNodes();
+    } catch (Exception e) {
+      Assert.assertTrue(e instanceof NullPointerException);
+    }
+    Args.getInstance().setP2pDisable(true);
+    GrpcAPI.NodeList nodeList = wallet.listNodes();
+    Assert.assertTrue(nodeList.getNodesList().size() == 0);
+  }
 }
 

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

@@ -0,0 +1,60 @@
+package org.tron.core.services;
+
+import io.grpc.ManagedChannelBuilder;
+import java.io.File;
+import lombok.extern.slf4j.Slf4j;
+import org.junit.After;
+import org.junit.Assert;
+import org.junit.Before;
+import org.junit.Test;
+import org.tron.api.GrpcAPI.EmptyMessage;
+import org.tron.api.WalletGrpc;
+import org.tron.common.application.Application;
+import org.tron.common.application.ApplicationFactory;
+import org.tron.common.application.TronApplicationContext;
+import org.tron.common.utils.FileUtil;
+import org.tron.core.Constant;
+import org.tron.core.config.DefaultConfig;
+import org.tron.core.config.args.Args;
+import stest.tron.wallet.common.client.Configuration;
+
+@Slf4j
+public class WalletApiTest {
+
+  private static TronApplicationContext context;
+  private static String dbPath = "output_wallet_api_test";
+  private String fullnode = Configuration.getByPath("testng.conf")
+      .getStringList("fullnode.ip.list").get(0);
+  private RpcApiService rpcApiService;
+  private Application appT;
+
+  @Before
+  public void init() {
+    Args.setParam(new String[]{ "-d", dbPath, "--p2p-disable", "true"}, Constant.TEST_CONF);
+    context = new TronApplicationContext(DefaultConfig.class);
+    appT = ApplicationFactory.create(context);
+    rpcApiService = context.getBean(RpcApiService.class);
+    appT.addService(rpcApiService);
+    appT.initServices(Args.getInstance());
+    appT.startServices();
+    appT.startup();
+  }
+
+  @Test
+  public void listNodesTest() {
+    WalletGrpc.WalletBlockingStub walletStub = WalletGrpc
+        .newBlockingStub(ManagedChannelBuilder.forTarget(fullnode)
+            .usePlaintext(true)
+            .build());
+    Assert.assertTrue(walletStub.listNodes(EmptyMessage.getDefaultInstance())
+        .getNodesList().size() == 0);
+  }
+
+  @After
+  public void destroy() {
+    Args.clearParam();
+    context.destroy();
+    FileUtil.deleteDir(new File(dbPath));
+  }
+
+}