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

Fix test_handle_prune

Browse Source
This commit is contained in:
NIC619
2019-07-24 11:35:14 +08:00
parent 99eabe49eb
commit b0b4ddd0ca
1 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
tests/pubsub/test_gossipsub.py
View File

@ -164,9 +164,9 @@ async def test_handle_prune():
libp2p_hosts = await create_libp2p_hosts(num_hosts) libp2p_hosts = await create_libp2p_hosts(num_hosts)
# Create pubsub, gossipsub instances # Create pubsub, gossipsub instances
_, gossipsubs = create_pubsub_and_gossipsub_instances(libp2p_hosts, \ pubsubs, gossipsubs = create_pubsub_and_gossipsub_instances(libp2p_hosts, \
SUPPORTED_PROTOCOLS, \ SUPPORTED_PROTOCOLS, \
10, 9, 11, 30, 3, 5, 2) 10, 9, 11, 30, 3, 5, 3)
index_alice = 0 index_alice = 0
id_alice = str(libp2p_hosts[index_alice].get_id()) id_alice = str(libp2p_hosts[index_alice].get_id())
@ -174,13 +174,13 @@ async def test_handle_prune():
id_bob = str(libp2p_hosts[index_bob].get_id()) id_bob = str(libp2p_hosts[index_bob].get_id())
topic = "test_handle_prune" topic = "test_handle_prune"
await gossipsubs[index_alice].join(topic) for pubsub in pubsubs:
await gossipsubs[index_bob].join(topic) await pubsub.subscribe(topic)
await connect(libp2p_hosts[index_alice], libp2p_hosts[index_bob]) await connect(libp2p_hosts[index_alice], libp2p_hosts[index_bob])
# Wait 2 seconds for heartbeat to allow mesh to connect # Wait 3 seconds for heartbeat to allow mesh to connect
await asyncio.sleep(2) await asyncio.sleep(3)
# Check that they are each other's mesh peer # Check that they are each other's mesh peer
assert id_alice in gossipsubs[index_bob].mesh[topic] assert id_alice in gossipsubs[index_bob].mesh[topic]
@ -191,7 +191,7 @@ async def test_handle_prune():
await gossipsubs[index_alice].emit_prune(topic, id_bob) await gossipsubs[index_alice].emit_prune(topic, id_bob)
# FIXME: This test currently works because the heartbeat interval # FIXME: This test currently works because the heartbeat interval
# is increased to 2 seconds, so alice won't get add back into # is increased to 3 seconds, so alice won't get add back into
# bob's mesh peer during heartbeat. # bob's mesh peer during heartbeat.
await asyncio.sleep(1) await asyncio.sleep(1)