changes as per review

libp2p · Apr 2, 2020 · 9c3dce7 · 9c3dce7
1 parent 6623e4d
commit 9c3dce7
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/dht_test.go b/dht_test.go
@@ -1414,7 +1414,7 @@ func testFindPeerQuery(t *testing.T,
 	}
 
 	// give some time for things to settle down
-	time.Sleep(2 * time.Second)
+	waitForWellFormedTables(t, dhts, minRTRefreshThreshold, minRTRefreshThreshold, 5*time.Second)
 
 	for _, d := range dhts {
 		if err := <-d.RefreshRoutingTable(); err != nil {

diff --git a/subscriber_notifee.go b/subscriber_notifee.go
@@ -120,11 +120,10 @@ func handlePeerProtocolsUpdatedEvent(dht *IpfsDHT, e event.EvtPeerProtocolsUpdat
 
 	if !valid {
 		dht.peerStoppedDHT(dht.ctx, e.Peer)
-	} else {
-		// we just might have discovered a peer that supports the DHT protocol
-		dht.fixLowPeers()
 	}
 
+	// we just might have discovered a peer that supports the DHT protocol
+	dht.fixLowPeers()
 }
 
 func handleLocalReachabilityChangedEvent(dht *IpfsDHT, e event.EvtLocalReachabilityChanged) {