From f4ec35573e2f0c9fd58ee0bb032c5eaa55d4f3e5 Mon Sep 17 00:00:00 2001
From: Jacob Heun <jacobheun@gmail.com>
Date: Fri, 28 Feb 2020 15:30:43 +0100
Subject: [PATCH] fix(test): improve flakey random walk discovery test (#574)

* test: improve stability of dht discovery test

* test: isolate who is random walking
---
 test/peer-discovery/index.node.js | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/test/peer-discovery/index.node.js b/test/peer-discovery/index.node.js
index b346f80e..2f315e29 100644
--- a/test/peer-discovery/index.node.js
+++ b/test/peer-discovery/index.node.js
@@ -139,17 +139,21 @@ describe('peer discovery scenarios', () => {
         },
         dht: {
           randomWalk: {
-            enabled: true,
+            enabled: false,
             delay: 1000, // start the first query quickly
             interval: 10000,
-            timeout: 1000
+            timeout: 5000
           },
           enabled: true
         }
       }
     })
 
-    libp2p = new Libp2p(getConfig(peerInfo))
+    const localConfig = getConfig(peerInfo)
+    // Only run random walk on our local node
+    localConfig.config.dht.randomWalk.enabled = true
+    libp2p = new Libp2p(localConfig)
+
     const remoteLibp2p1 = new Libp2p(getConfig(remotePeerInfo1))
     const remoteLibp2p2 = new Libp2p(getConfig(remotePeerInfo2))
 
@@ -161,6 +165,7 @@ describe('peer discovery scenarios', () => {
     })
 
     await Promise.all([
+      libp2p.start(),
       remoteLibp2p1.start(),
       remoteLibp2p2.start()
     ])
@@ -173,8 +178,6 @@ describe('peer discovery scenarios', () => {
       remoteLibp2p2.dial(remotePeerInfo1)
     ])
 
-    libp2p.start()
-
     await deferred.promise
     return Promise.all([
       remoteLibp2p1.stop(),