varun-r-mallya/py-libp2p
1
0
Fork 0
mirror of https://github.com/varun-r-mallya/py-libp2p.git synced 2026-02-10 07:00:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update tests for new logic

Browse Source
This commit is contained in:
Alex Stokes
2019-08-02 18:36:58 -07:00
parent 9e18d7561d
commit cda74dd382
4 changed files with 122 additions and 49 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
tests/libp2p/test_notify.py
View File

@ -19,6 +19,7 @@ from tests.utils import (
cleanup,
echo_stream_handler,
perform_two_host_set_up_custom_handler,
generate_new_private_key,
)
@ -172,14 +173,18 @@ async def test_one_notifier_on_two_nodes():
async def test_one_notifier_on_two_nodes_with_listen():
events_b = []
node_a_key = generate_new_private_key()
node_a_transport_opt = ["/ip4/127.0.0.1/tcp/0"]
node_a = await new_node(transport_opt=node_a_transport_opt)
node_a = await new_node(node_a_key, transport_opt=node_a_transport_opt)
await node_a.get_network().listen(multiaddr.Multiaddr(node_a_transport_opt[0]))
# Set up node_b swarm to pass into host
node_b_key = generate_new_private_key()
node_b_transport_opt = ["/ip4/127.0.0.1/tcp/0"]
node_b_multiaddr = multiaddr.Multiaddr(node_b_transport_opt[0])
node_b_swarm = initialize_default_swarm(transport_opt=node_b_transport_opt)
node_b_swarm = initialize_default_swarm(
node_b_key, transport_opt=node_b_transport_opt
)
node_b = BasicHost(node_b_swarm)
async def my_stream_handler(stream):