modify brock test

Author: lvs007

src/test/java/org/tron/core/net/node/BroadTest.java

@@ -8,6 +8,7 @@
 import java.util.concurrent.ConcurrentHashMap;
 import java.util.concurrent.ExecutorService;
 import lombok.extern.slf4j.Slf4j;
+import org.apache.commons.collections4.MapUtils;
 import org.junit.After;
 import org.junit.Assert;
 import org.junit.Before;
@@ -18,6 +19,7 @@
 import org.tron.common.overlay.client.PeerClient;
 import org.tron.common.overlay.discover.Node;
 import org.tron.common.overlay.message.Message;
+import org.tron.common.overlay.server.Channel;
 import org.tron.common.overlay.server.ChannelManager;
 import org.tron.common.overlay.server.MessageQueue;
 import org.tron.common.overlay.server.SyncPool;
@@ -27,6 +29,7 @@
 import org.tron.core.capsule.BlockCapsule;
 import org.tron.core.config.DefaultConfig;
 import org.tron.core.config.args.Args;
+import org.tron.core.db.ByteArrayWrapper;
 import org.tron.core.db.Manager;
 import org.tron.core.net.message.BlockMessage;
 import org.tron.core.net.message.MessageTypes;
@@ -234,7 +237,6 @@ public void run() {
     }
   }
 
-  private static int trys = 0;
   private void prepare() {
     try {
       ExecutorService advertiseLoopThread = ReflectUtils.getFieldValue(node, "broadPool");
@@ -251,7 +253,13 @@ public void run() {
           peerClient.connect(node.getHost(), node.getPort(), node.getHexId());
         }
       }).start();
-      Thread.sleep(5000);
+      Thread.sleep(2000);
+      Map<ByteArrayWrapper, Channel> activePeers = ReflectUtils
+          .getFieldValue(channelManager, "activePeers");
+      int tryTimes = 0;
+      while (MapUtils.isEmpty(activePeers) && ++tryTimes < 10) {
+        Thread.sleep(1000);
+      }
       go = true;
     } catch (Exception e) {
       e.printStackTrace();