py-libp2p/tests/utils.py

43 lines
1.3 KiB
Python
Raw Normal View History

2019-03-18 07:33:10 +08:00
import multiaddr
2019-03-18 07:33:10 +08:00
from libp2p import new_node
2019-07-25 23:11:27 +08:00
from libp2p.peer.peerinfo import info_from_p2p_addr
2019-09-06 17:26:40 +08:00
from tests.constants import MAX_READ_LEN
2019-07-25 23:11:27 +08:00
async def connect(node1, node2):
"""
Connect node1 to node2
"""
addr = node2.get_addrs()[0]
info = info_from_p2p_addr(addr)
await node1.connect(info)
2019-03-18 07:33:10 +08:00
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)
2019-08-01 06:00:12 +08:00
2019-03-18 07:33:10 +08:00
async def echo_stream_handler(stream):
while True:
2019-09-06 17:26:40 +08:00
read_string = (await stream.read(MAX_READ_LEN)).decode()
2019-03-18 07:33:10 +08:00
resp = "ack:" + read_string
await stream.write(resp.encode())
2019-03-18 09:15:14 +08:00
2019-08-01 06:00:12 +08:00
2019-09-06 17:26:40 +08:00
async def perform_two_host_set_up(handler=echo_stream_handler):
2019-03-18 09:15:14 +08:00
transport_opt_list = [["/ip4/127.0.0.1/tcp/0"], ["/ip4/127.0.0.1/tcp/0"]]
2019-03-18 09:30:56 +08:00
(node_a, node_b) = await set_up_nodes_by_transport_opt(transport_opt_list)
2019-03-18 09:15:14 +08:00
node_b.set_stream_handler("/echo/1.0.0", handler)
# Associate the peer with local ip address (see default parameters of Libp2p())
node_a.get_peerstore().add_addrs(node_b.get_id(), node_b.get_addrs(), 10)
2019-03-18 09:30:56 +08:00
return node_a, node_b