LBRYCommunity/lbry-sdk
1
0
Fork 1
Code Issues Pull requests Projects Releases 282 Packages Wiki Activity

update contact expiration test and add re-join after expiration test

Browse source
This commit is contained in:
Jack Robison 2018-05-23 20:44:28 -04:00
parent 877da78505
commit b5f3ed5542
No known key found for this signature in database
GPG key ID: DF25C68FE0239BB2
1 changed files with 26 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

30
lbrynet/tests/functional/dht/test_contact_expiration.py
View file

@ -14,6 +14,7 @@ class TestPeerExpiration(TestKademliaBase):
removed_nodes = [] removed_nodes = []
self.show_info() self.show_info()
# stop 5 nodes
for _ in range(5): for _ in range(5):
n = self.nodes[0] n = self.nodes[0]
removed_nodes.append(n) removed_nodes.append(n)
@ -23,11 +24,15 @@ class TestPeerExpiration(TestKademliaBase):
offline_addresses = self.get_routable_addresses().difference(self.get_online_addresses()) offline_addresses = self.get_routable_addresses().difference(self.get_online_addresses())
self.assertSetEqual(offline_addresses, removed_addresses) self.assertSetEqual(offline_addresses, removed_addresses)
get_nodes_with_stale_contacts = lambda: filter(lambda node: any(contact.address in offline_addresses get_nodes_with_stale_contacts = lambda: filter(lambda node: any(contact.address in offline_addresses
for contact in node.contacts), self.nodes + self._seeds) for contact in node.contacts), self.nodes + self._seeds)
self.assertRaises(AssertionError, self.verify_all_nodes_are_routable) self.assertRaises(AssertionError, self.verify_all_nodes_are_routable)
self.assertTrue(len(get_nodes_with_stale_contacts()) > 1) self.assertTrue(len(get_nodes_with_stale_contacts()) > 1)
for _ in range(90):
# run the network for an hour, which should expire the removed nodes
for _ in range(60):
log.info("Time is %f, nodes with stale contacts: %i/%i", self.clock.seconds(), log.info("Time is %f, nodes with stale contacts: %i/%i", self.clock.seconds(),
len(get_nodes_with_stale_contacts()), len(self.nodes + self._seeds)) len(get_nodes_with_stale_contacts()), len(self.nodes + self._seeds))
self.pump_clock(60) self.pump_clock(60)
@ -35,9 +40,26 @@ class TestPeerExpiration(TestKademliaBase):
self.verify_all_nodes_are_routable() self.verify_all_nodes_are_routable()
self.verify_all_nodes_are_pingable() self.verify_all_nodes_are_pingable()
restarted_node = removed_nodes[0]
yield self.run_reactor(1, [restarted_node.start([(seed_name, 4444)
for seed_name in sorted(self.seed_dns.keys())])])
class TestReJoinExpiredPeer(TestKademliaBase):
network_size = 40
@defer.inlineCallbacks
def test_re_join_expired_peer(self):
removed_node = self.nodes[0]
self.nodes.remove(removed_node)
yield self.run_reactor(1, [removed_node.stop()])
# run the network for an hour, which should expire the removed node
for _ in range(60):
self.pump_clock(60)
self.verify_all_nodes_are_routable()
self.verify_all_nodes_are_pingable()
self.nodes.append(removed_node)
yield self.run_reactor(
31, [removed_node.start([(seed_name, 4444) for seed_name in sorted(self.seed_dns.keys())])]
)
self.pump_clock(901)
self.verify_all_nodes_are_routable() self.verify_all_nodes_are_routable()
self.verify_all_nodes_are_pingable() self.verify_all_nodes_are_pingable()