add coverage of NodeInfoServiceTest

Author: 317787106

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

@@ -2,17 +2,28 @@
 
 import com.alibaba.fastjson.JSON;
 import com.google.protobuf.ByteString;
+import java.net.InetSocketAddress;
 import javax.annotation.Resource;
 import lombok.extern.slf4j.Slf4j;
+import org.junit.After;
 import org.junit.Assert;
 import org.junit.BeforeClass;
 import org.junit.Test;
+import org.springframework.context.ApplicationContext;
 import org.tron.common.BaseTest;
 import org.tron.common.entity.NodeInfo;
+import org.tron.common.utils.PublicMethod;
+import org.tron.common.utils.ReflectUtils;
 import org.tron.common.utils.Sha256Hash;
 import org.tron.core.Constant;
 import org.tron.core.capsule.BlockCapsule;
 import org.tron.core.config.args.Args;
+import org.tron.core.net.P2pEventHandlerImpl;
+import org.tron.core.net.TronNetService;
+import org.tron.core.net.peer.PeerConnection;
+import org.tron.core.net.peer.PeerManager;
+import org.tron.p2p.P2pConfig;
+import org.tron.p2p.connection.Channel;
 import org.tron.program.Version;
 
 
@@ -23,6 +34,10 @@ public class NodeInfoServiceTest extends BaseTest {
   protected NodeInfoService nodeInfoService;
   @Resource
   protected WitnessProductBlockService witnessProductBlockService;
+  @Resource
+  private P2pEventHandlerImpl p2pEventHandler;
+  @Resource
+  private TronNetService tronNetService;
 
 
   @BeforeClass
@@ -31,6 +46,13 @@ public static void init() {
         Constant.TEST_CONF);
   }
 
+  @After
+  public void clearPeers() {
+    for (PeerConnection p : PeerManager.getPeers()) {
+      PeerManager.remove(p.getChannel());
+    }
+  }
+
   @Test
   public void test() {
     BlockCapsule blockCapsule1 = new BlockCapsule(1, Sha256Hash.ZERO_HASH,
@@ -40,11 +62,31 @@ public void test() {
     witnessProductBlockService.validWitnessProductTwoBlock(blockCapsule1);
     witnessProductBlockService.validWitnessProductTwoBlock(blockCapsule2);
 
+    addPeer();
+
     //test setConnectInfo
     NodeInfo nodeInfo = nodeInfoService.getNodeInfo();
     Assert.assertEquals(nodeInfo.getConfigNodeInfo().getCodeVersion(), Version.getVersion());
     Assert.assertEquals(1, nodeInfo.getCheatWitnessInfoMap().size());
     logger.info("{}", JSON.toJSONString(nodeInfo));
   }
 
+  private void addPeer() {
+    int port = PublicMethod.chooseRandomPort();
+    P2pConfig p2pConfig = new P2pConfig();
+    p2pConfig.setIp("127.0.0.1");
+    p2pConfig.setPort(port);
+    ReflectUtils.setFieldValue(tronNetService, "p2pConfig", p2pConfig);
+    TronNetService.getP2pService().start(p2pConfig);
+
+    ApplicationContext ctx = (ApplicationContext) ReflectUtils.getFieldObject(p2pEventHandler,
+        "ctx");
+    InetSocketAddress inetSocketAddress1 =
+        new InetSocketAddress("127.0.0.1", 10001);
+    Channel c1 = new Channel();
+    ReflectUtils.setFieldValue(c1, "inetSocketAddress", inetSocketAddress1);
+    ReflectUtils.setFieldValue(c1, "inetAddress", inetSocketAddress1.getAddress());
+
+    PeerManager.add(ctx, c1);
+  }
 }