Madiator2011/lbry-sdk
1
0
Fork 0
forked from LBRYCommunity/lbry-sdk
Code Pull requests Activity

fix drop/reconnect in test_losing_connection

Browse source
This commit is contained in:
Jack Robison 2019-11-22 01:19:22 -05:00
parent 5951186463
commit d7fe46dbde
No known key found for this signature in database
GPG key ID: DF25C68FE0239BB2
1 changed files with 3 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
lbry/tests/unit/dht/test_node.py
View file

@ -130,24 +130,15 @@ class TestTemporarilyLosingConnection(AsyncioTestCase):
self.assertTrue(len(await node._storage.get_persisted_kademlia_peers()) > num_seeds) self.assertTrue(len(await node._storage.get_persisted_kademlia_peers()) > num_seeds)
# We lost internet connection - all the peers stop responding # We lost internet connection - all the peers stop responding
popped_protocols = [] dht_network.pop((node.protocol.external_ip, node.protocol.udp_port))
for peer_address in peer_addresses[:-1]:
popped_protocols.append(dht_network.pop(peer_address))
# The peers are cleared on refresh from RT and storage # The peers are cleared on refresh from RT and storage
await advance(4000) await advance(4000)
self.assertListEqual([], node.protocol.routing_table.get_peers()) self.assertListEqual([], node.protocol.routing_table.get_peers())
self.assertListEqual([], await node._storage.get_persisted_kademlia_peers()) self.assertListEqual([], await node._storage.get_persisted_kademlia_peers())
# Reconnect some of the previously stored - node shouldn't connect # Reconnect
for peer_address, protocol in zip(peer_addresses[num_seeds+1:-2], popped_protocols[num_seeds+1:-2]): dht_network[(node.protocol.external_ip, node.protocol.udp_port)] = node.protocol
dht_network[peer_address] = protocol
await advance(1000)
self.assertListEqual([], node.protocol.routing_table.get_peers())
# Reconnect some of the seed nodes
for peer_address, protocol in zip(peer_addresses[:num_seeds], popped_protocols[:num_seeds]):
dht_network[peer_address] = protocol
# Check that node reconnects at least to them # Check that node reconnects at least to them
await advance(1000) await advance(1000)