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

tests: multiple trackers, simple case

Browse source
This commit is contained in:
Victor Shyba 2022-03-09 17:01:49 -03:00
parent cc4a578578
commit e299a9c159
1 changed files with 10 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -52,10 +52,11 @@ class UDPTrackerClientTestCase(AsyncioTestCase):
async def add_server(self, port=None, add_to_client=True): async def add_server(self, port=None, add_to_client=True):
port = port or len(self.servers) + 59990 port = port or len(self.servers) + 59990
assert port not in self.servers
server = UDPTrackerServerProtocol() server = UDPTrackerServerProtocol()
self.servers[port] = server
transport, _ = await self.loop.create_datagram_endpoint(lambda: server, local_addr=("127.0.0.1", port)) transport, _ = await self.loop.create_datagram_endpoint(lambda: server, local_addr=("127.0.0.1", port))
self.addCleanup(transport.close) self.addCleanup(transport.close)
self.servers[port] = server
if add_to_client: if add_to_client:
self.client.servers.append(("127.0.0.1", port)) self.client.servers.append(("127.0.0.1", port))
@ -82,3 +83,11 @@ class UDPTrackerClientTestCase(AsyncioTestCase):
with self.assertRaises(Exception) as err: with self.assertRaises(Exception) as err:
await self.client.get_peer_list(info_hash) await self.client.get_peer_list(info_hash)
self.assertEqual(err.exception.args[0], b'Connection ID missmatch.\x00') self.assertEqual(err.exception.args[0], b'Connection ID missmatch.\x00')
async def test_multiple(self):
await asyncio.gather(*[self.add_server() for _ in range(10)])
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]))