forked from LBRYCommunity/lbry-sdk
71 lines
2.4 KiB
Python
71 lines
2.4 KiB
Python
|
import logging
|
||
|
|
from twisted.internet import defer
|
||
|
|
from dht_test_environment import TestKademliaBase
|
||
|
|
|
||
|
|
log = logging.getLogger()
|
||
|
|
|
||
|
|
|
||
|
|
class TestReJoin(TestKademliaBase):
|
||
|
|
network_size = 40
|
||
|
|
|
||
|
|
@defer.inlineCallbacks
|
||
|
|
def test_re_join(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
|
||
|
|
self.pump_clock(3600)
|
||
|
|
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_pingable()
|
||
|
|
|
||
|
|
@defer.inlineCallbacks
|
||
|
|
def test_re_join_with_new_ip(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()
|
||
|
|
removed_node.externalIP = "10.43.43.43"
|
||
|
|
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_pingable()
|
||
|
|
|
||
|
|
@defer.inlineCallbacks
|
||
|
|
def test_re_join_with_new_node_id(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()
|
||
|
|
removed_node.node_id = removed_node._generateID()
|
||
|
|
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_pingable()
|