From 72a8d55faa1892dda20512313f7e1d14bc057ec7 Mon Sep 17 00:00:00 2001 From: zixuanzh Date: Sun, 17 Mar 2019 19:33:10 -0400 Subject: [PATCH] refactor host setup helper --- tests/utils.py | 21 +++++++++++++++++++-- 1 file changed, 19 insertions(+), 2 deletions(-) diff --git a/tests/utils.py b/tests/utils.py index c995412..e7f9b85 100644 --- a/tests/utils.py +++ b/tests/utils.py @@ -1,6 +1,8 @@ -import asyncio - from contextlib import suppress +import asyncio +import multiaddr + +from libp2p import new_node async def cleanup(): @@ -12,3 +14,18 @@ async def cleanup(): # Cancelled task raises asyncio.CancelledError that we can suppress: with suppress(asyncio.CancelledError): await task + +async def set_up_nodes_by_transport_opt(transport_opt_list): + nodes_list = [] + for transport_opt in transport_opt_list: + node = await new_node(transport_opt=transport_opt) + await node.get_network().listen(multiaddr.Multiaddr(transport_opt[0])) + nodes_list.append(node) + return tuple(nodes_list) + +async def echo_stream_handler(stream): + while True: + read_string = (await stream.read()).decode() + + resp = "ack:" + read_string + await stream.write(resp.encode())