commit 89bc644ea55f1443e6bdc0279a401b9fef5f154a Author: Illia Volochii <illia.volochii@gmail.com> Date: Sat Mar 14 22:05:15 2020 +0200 Add temporary scripts for testing async communication with sockets --- test_control_port.py | 30 ++++++++++++++++++++++++++++++ test_control_socket_file.py | 30 ++++++++++++++++++++++++++++++ test_relay_socket.py | 28 ++++++++++++++++++++++++++++ 3 files changed, 88 insertions(+) diff --git a/test_control_port.py b/test_control_port.py new file mode 100644 index 00000000..85fe26d4 --- /dev/null +++ b/test_control_port.py @@ -0,0 +1,30 @@ +import asyncio + +from stem.async_socket import ControlPort + + +async def run_command(i, command): + async with ControlPort() as cp: + print(f'{i} Connecting') + await cp.connect() + print(f'{i} Authenticating') + await cp.send('AUTHENTICATE "password"') + print(f'{i} Receiving auth message') + await cp.recv() + print(f'{i} Sending the command') + await cp.send(command) + print(f'{i} Receiving result of the command') + result = await cp.recv() + print(f'{i} {result.content()}') + + +if __name__ == '__main__': + loop = asyncio.get_event_loop() + tasks = asyncio.gather( + run_command(1, 'PROTOCOLINFO 1'), + run_command(2, 'GETINFO traffic/read'), + ) + try: + loop.run_until_complete(tasks) + finally: + loop.close() diff --git a/test_control_socket_file.py b/test_control_socket_file.py new file mode 100644 index 00000000..4c6816f4 --- /dev/null +++ b/test_control_socket_file.py @@ -0,0 +1,30 @@ +import asyncio + +from stem.async_socket import ControlSocketFile + + +async def run_command(i, command): + async with ControlSocketFile() as cp: + print(f'{i} Connecting') + await cp.connect() + print(f'{i} Authenticating') + await cp.send('AUTHENTICATE "password"') + print(f'{i} Receiving auth message') + await cp.recv() + print(f'{i} Sending the command') + await cp.send(command) + print(f'{i} Receiving result of the command') + result = await cp.recv() + print(f'{i} {result.content()}') + + +if __name__ == '__main__': + loop = asyncio.get_event_loop() + tasks = asyncio.gather( + run_command(1, 'PROTOCOLINFO 1'), + run_command(2, 'GETINFO traffic/read'), + ) + try: + loop.run_until_complete(tasks) + finally: + loop.close() diff --git a/test_relay_socket.py b/test_relay_socket.py new file mode 100644 index 00000000..76289d46 --- /dev/null +++ b/test_relay_socket.py @@ -0,0 +1,28 @@ +import asyncio + +from stem.client import DEFAULT_LINK_PROTOCOLS +from stem.client.cell import VersionsCell +from stem.async_socket import RelaySocket + + +async def run_command(i, command): + async with RelaySocket(address='127.0.0.1', port=443) as cp: + print(f'{i} Connecting') + await cp.connect() + print(f'{i} Sending the command') + await cp.send(command) + print(f'{i} Receiving result of the command') + result = await cp.recv(2) + print(result) + + +if __name__ == '__main__': + loop = asyncio.get_event_loop() + tasks = asyncio.gather( + run_command(1, VersionsCell(DEFAULT_LINK_PROTOCOLS).pack(2)), + run_command(2, VersionsCell(DEFAULT_LINK_PROTOCOLS).pack(2)), + ) + try: + loop.run_until_complete(tasks) + finally: + loop.close()