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

fix udp ping test

Browse source
This commit is contained in:
Jack Robison 2021-05-21 13:12:23 -04:00 committed by Victor Shyba
parent 96c318ee88
commit c8d0d765d1
1 changed files with 18 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

26
tests/integration/blockchain/test_network.py
View file

@ -176,10 +176,19 @@ class UDPServerFailDiscoveryTest(AsyncioTestCase):
class ServerPickingTestCase(AsyncioTestCase): class ServerPickingTestCase(AsyncioTestCase):
async def _make_udp_server(self, port): async def _make_udp_server(self, port, latency) -> StatusServer:
s = StatusServer() s = StatusServer()
await s.start(0, b'\x00' * 32, '127.0.0.1', port) await s.start(0, b'\x00' * 32, '127.0.0.1', port)
s.set_available()
sendto = s._protocol.transport.sendto
def mock_sendto(data, addr):
self.loop.call_later(latency, sendto, data, addr)
s._protocol.transport.sendto = mock_sendto
self.addCleanup(s.stop) self.addCleanup(s.stop)
return s
async def _make_fake_server(self, latency=1.0, port=1): async def _make_fake_server(self, latency=1.0, port=1):
# local fake server with artificial latency # local fake server with artificial latency
@ -191,23 +200,24 @@ class ServerPickingTestCase(AsyncioTestCase):
return {'height': 1} return {'height': 1}
server = await self.loop.create_server(lambda: FakeSession(), host='127.0.0.1', port=port) server = await self.loop.create_server(lambda: FakeSession(), host='127.0.0.1', port=port)
self.addCleanup(server.close) self.addCleanup(server.close)
await self._make_udp_server(port) await self._make_udp_server(port, latency)
return '127.0.0.1', port return '127.0.0.1', port
async def _make_bad_server(self, port=42420): async def _make_bad_server(self, port=42420):
async def echo(reader, writer): async def echo(reader, writer):
while True: while True:
writer.write(await reader.read()) writer.write(await reader.read())
server = await asyncio.start_server(echo, host='127.0.0.1', port=port) server = await asyncio.start_server(echo, host='127.0.0.1', port=port)
self.addCleanup(server.close) self.addCleanup(server.close)
await self._make_udp_server(port) await self._make_udp_server(port, 0)
return '127.0.0.1', port return '127.0.0.1', port
async def _test_pick_fastest(self): async def test_pick_fastest(self):
ledger = Mock(config={ ledger = Mock(config={
'default_servers': [ 'default_servers': [
# fast but unhealthy, should be discarded # fast but unhealthy, should be discarded
await self._make_bad_server(), # await self._make_bad_server(),
('localhost', 1), ('localhost', 1),
('example.that.doesnt.resolve', 9000), ('example.that.doesnt.resolve', 9000),
await self._make_fake_server(latency=1.0, port=1340), await self._make_fake_server(latency=1.0, port=1340),
@ -223,7 +233,7 @@ class ServerPickingTestCase(AsyncioTestCase):
await asyncio.wait_for(network.on_connected.first, timeout=10) await asyncio.wait_for(network.on_connected.first, timeout=10)
self.assertTrue(network.is_connected) self.assertTrue(network.is_connected)
self.assertTupleEqual(network.client.server, ('127.0.0.1', 1337)) self.assertTupleEqual(network.client.server, ('127.0.0.1', 1337))
self.assertTrue(all([not session.is_closing() for session in network.session_pool.available_sessions])) # self.assertTrue(all([not session.is_closing() for session in network.session_pool.available_sessions]))
# ensure we are connected to all of them after a while # ensure we are connected to all of them after a while
await asyncio.sleep(1) # await asyncio.sleep(1)
self.assertEqual(len(list(network.session_pool.available_sessions)), 3) # self.assertEqual(len(list(network.session_pool.available_sessions)), 3)