StarMirror/py-libp2p
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add test

Browse Source
This commit is contained in:
NIC619 2019-07-22 19:28:12 +08:00
parent 67f9edb77d
commit cdbeb63879
No known key found for this signature in database
GPG Key ID: 570C35F5C2D51B17
1 changed files with 16 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
tests/pubsub/test_gossipsub.py
View File

@ -39,13 +39,27 @@ async def test_join():
# Wait 2 seconds for heartbeat to allow mesh to connect # Wait 2 seconds for heartbeat to allow mesh to connect
await asyncio.sleep(2) await asyncio.sleep(2)
# Check that the pubsub of central node does not have mesh or fanout for the topic # Central node publish to the topic so that this topic
assert topic not in gossipsubs[central_node_index].fanout # is added to central node's fanout
next_msg_id_func = message_id_generator(0)
msg_content = ""
host_id = str(libp2p_hosts[central_node_index].get_id())
# Generate message packet
packet = generate_RPC_packet(host_id, [topic], msg_content, next_msg_id_func())
# publish from the randomly chosen host
await gossipsubs[central_node_index].publish(host_id, packet.SerializeToString())
# Check that the gossipsub of central node has fanout for the topic
assert topic in gossipsubs[central_node_index].fanout
# Check that the gossipsub of central node does not has mesh for the topic
assert topic not in gossipsubs[central_node_index].mesh assert topic not in gossipsubs[central_node_index].mesh
# Central node subscribe message origin # Central node subscribe message origin
await pubsubs[central_node_index].subscribe(topic) await pubsubs[central_node_index].subscribe(topic)
# Check that the gossipsub of central node no longer has fanout for the topic
assert topic not in gossipsubs[central_node_index].fanout
for i in hosts_indices: for i in hosts_indices:
if i in subscribed_peer_indices: if i in subscribed_peer_indices:
assert str(libp2p_hosts[i].get_id()) in gossipsubs[central_node_index].mesh[topic] assert str(libp2p_hosts[i].get_id()) in gossipsubs[central_node_index].mesh[topic]