From ebe5dd0e6862f399c7e7fedb59ccd60fce783959 Mon Sep 17 00:00:00 2001
From: Jack Robison <jackrobison@lbry.io>
Date: Mon, 26 Feb 2018 11:53:10 -0500
Subject: [PATCH] better ping test

---
 lbrynet/tests/functional/test_dht.py | 46 +++++++++++++++++++++-------
 1 file changed, 35 insertions(+), 11 deletions(-)

diff --git a/lbrynet/tests/functional/test_dht.py b/lbrynet/tests/functional/test_dht.py
index 142bffc8b..ac8572193 100644
--- a/lbrynet/tests/functional/test_dht.py
+++ b/lbrynet/tests/functional/test_dht.py
@@ -107,6 +107,37 @@ class TestKademliaBase(unittest.TestCase):
             routable.update(contact_addresses)
         self.assertSetEqual(routable, node_addresses)
 
+    @defer.inlineCallbacks
+    def verify_all_nodes_are_pingable(self):
+        ping_replies = {}
+        ping_dl = []
+        contacted = set()
+
+        def _ping_cb(result, node, replies):
+            replies[node] = result
+
+        for node in self._seeds:
+            contact_addresses = set()
+            for contact in node.contacts:
+                contact_addresses.add(contact.address)
+                d = contact.ping()
+                d.addCallback(_ping_cb, contact.address, ping_replies)
+                contacted.add(contact.address)
+                ping_dl.append(d)
+        for node in self.nodes:
+            contact_addresses = set()
+            for contact in node.contacts:
+                contact_addresses.add(contact.address)
+                d = contact.ping()
+                d.addCallback(_ping_cb, contact.address, ping_replies)
+                contacted.add(contact.address)
+                ping_dl.append(d)
+        self.run_reactor(2, *ping_dl)
+        yield threads.deferToThread(time.sleep, 0.1)
+        node_addresses = {node.externalIP for node in self.nodes}.union({seed.externalIP for seed in self._seeds})
+        self.assertSetEqual(node_addresses, contacted)
+        self.assertDictEqual(ping_replies, {node: "pong" for node in contacted})
+
 
 class TestKademliaBootstrap(TestKademliaBase):
     """
@@ -134,17 +165,7 @@ class TestKademliaBootstrap(TestKademliaBase):
         )
 
     def test_all_nodes_are_pingable(self):
-        def _ping_cb(result):
-            self.assertEqual(result, "pong")
-
-        dl = []
-        for seed in self._seeds:
-            self.assertEqual(len(seed.contacts), 2)
-            for contact in seed.contacts:
-                d = contact.ping()
-                d.addCallback(_ping_cb)
-                dl.append(d)
-        self.run_reactor(1, *dl)
+        return self.verify_all_nodes_are_pingable()
 
 
 class TestKademliaBootstrapSixteenSeeds(TestKademliaBase):
@@ -208,6 +229,9 @@ class TestKademliaBootstrapSixteenSeeds(TestKademliaBase):
     def test_bootstrap_network(self):
         pass
 
+    def test_all_nodes_are_pingable(self):
+        return self.verify_all_nodes_are_pingable()
+
 
 class Test250NodeNetwork(TestKademliaBase):
     network_size = 250