diff options
Diffstat (limited to 'tests/multi_net/asyncio_tcp_readall.py')
-rw-r--r-- | tests/multi_net/asyncio_tcp_readall.py | 57 |
1 files changed, 57 insertions, 0 deletions
diff --git a/tests/multi_net/asyncio_tcp_readall.py b/tests/multi_net/asyncio_tcp_readall.py new file mode 100644 index 0000000000..66ad97549c --- /dev/null +++ b/tests/multi_net/asyncio_tcp_readall.py @@ -0,0 +1,57 @@ +# Test asyncio stream read(-1) method using TCP server/client + +try: + import asyncio +except ImportError: + print("SKIP") + raise SystemExit + +PORT = 8000 + + +async def handle_connection(reader, writer): + writer.write(b"a") + await writer.drain() + + # Split the first 2 bytes up so the client must wait for the second one + await asyncio.sleep(0.1) + + writer.write(b"b") + await writer.drain() + + writer.write(b"c") + await writer.drain() + + print("close") + writer.close() + await writer.wait_closed() + + print("done") + ev.set() + + +async def tcp_server(): + global ev + ev = asyncio.Event() + server = await asyncio.start_server(handle_connection, "0.0.0.0", PORT) + print("server running") + multitest.next() + async with server: + await asyncio.wait_for(ev.wait(), 2) + + +async def tcp_client(): + reader, writer = await asyncio.open_connection(IP, PORT) + print(await reader.read()) + print(await reader.read()) # should be empty + print(await reader.read(1)) # should be empty + + +def instance0(): + multitest.globals(IP=multitest.get_network_ip()) + asyncio.run(tcp_server()) + + +def instance1(): + multitest.next() + asyncio.run(tcp_client()) |