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

tests: lower timeout, add test with bad and good mixed

Browse source
This commit is contained in:
Victor Shyba 2022-03-09 17:07:16 -03:00
parent e299a9c159
commit 407c570f8b
2 changed files with 12 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
lbry/torrent/tracker.py
View file

@ -124,8 +124,8 @@ class UDPTrackerClientProtocol(asyncio.DatagramProtocol):
class TrackerClient: class TrackerClient:
EVENT_CONTROLLER = StreamController() EVENT_CONTROLLER = StreamController()
def __init__(self, node_id, announce_port, servers): def __init__(self, node_id, announce_port, servers, timeout=10.0):
self.client = UDPTrackerClientProtocol() self.client = UDPTrackerClientProtocol(timeout=timeout)
self.transport = None self.transport = None
self.node_id = node_id or random.getrandbits(160).to_bytes(20, "big", signed=False) self.node_id = node_id or random.getrandbits(160).to_bytes(20, "big", signed=False)
self.announce_port = announce_port self.announce_port = announce_port

11
tests/unit/torrent/test_tracker.py
View file

@ -45,7 +45,7 @@ class UDPTrackerServerProtocol(asyncio.DatagramProtocol): # for testing. Not su
class UDPTrackerClientTestCase(AsyncioTestCase): class UDPTrackerClientTestCase(AsyncioTestCase):
async def asyncSetUp(self): async def asyncSetUp(self):
self.servers = {} self.servers = {}
self.client = TrackerClient(b"\x00" * 48, 4444, []) self.client = TrackerClient(b"\x00" * 48, 4444, [], timeout=0.1)
await self.client.start() await self.client.start()
self.addCleanup(self.client.stop) self.addCleanup(self.client.stop)
await self.add_server() await self.add_server()
@ -91,3 +91,12 @@ class UDPTrackerClientTestCase(AsyncioTestCase):
for server in self.servers.values(): for server in self.servers.values():
self.assertEqual(1, len(server.peers)) self.assertEqual(1, len(server.peers))
self.assertEqual(1, len(server.peers[info_hash])) self.assertEqual(1, len(server.peers[info_hash]))
async def test_multiple_with_bad_one(self):
await asyncio.gather(*[self.add_server() for _ in range(10)])
self.client.servers.append(("127.0.0.2", 7070))
info_hash = random.getrandbits(160).to_bytes(20, "big", signed=False)
await self.client.get_peer_list(info_hash)
for server in self.servers.values():
self.assertEqual(1, len(server.peers))
self.assertEqual(1, len(server.peers[info_hash]))