varun-r-mallya/py-libp2p
1
0
Fork 0
mirror of https://github.com/varun-r-mallya/py-libp2p.git synced 2025-12-31 20:36:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Lint test and add cleanup to the end

Browse Source
This commit is contained in:
NIC619
2019-07-19 19:43:12 +08:00
parent 41141c028b
commit 14ee44c549
1 changed files with 3 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
tests/pubsub/test_gossipsub.py
View File

@ -2,10 +2,6 @@ import asyncio
import pytest import pytest
import random import random
from libp2p.pubsub.gossipsub import GossipSub
from libp2p.pubsub.floodsub import FloodSub
from libp2p.pubsub.pb import rpc_pb2
from libp2p.pubsub.pubsub import Pubsub
from utils import message_id_generator, generate_RPC_packet, \ from utils import message_id_generator, generate_RPC_packet, \
create_libp2p_hosts, create_pubsub_and_gossipsub_instances, sparse_connect, dense_connect, \ create_libp2p_hosts, create_pubsub_and_gossipsub_instances, sparse_connect, dense_connect, \
connect, one_to_all_connect connect, one_to_all_connect
@ -17,8 +13,6 @@ SUPPORTED_PROTOCOLS = ["/gossipsub/1.0.0"]
@pytest.mark.asyncio @pytest.mark.asyncio
async def test_join(): async def test_join():
# Create libp2p hosts # Create libp2p hosts
next_msg_id_func = message_id_generator(0)
num_hosts = 10 num_hosts = 10
hosts_indices = list(range(num_hosts)) hosts_indices = list(range(num_hosts))
libp2p_hosts = await create_libp2p_hosts(num_hosts) libp2p_hosts = await create_libp2p_hosts(num_hosts)
@ -37,7 +31,7 @@ async def test_join():
# All pubsub except the one of central node subscribe to topic # All pubsub except the one of central node subscribe to topic
for i in subscribed_peer_indices: for i in subscribed_peer_indices:
q = await pubsubs[i].subscribe(topic) await pubsubs[i].subscribe(topic)
# Connect central host to all other hosts # Connect central host to all other hosts
await one_to_all_connect(libp2p_hosts, central_node_index) await one_to_all_connect(libp2p_hosts, central_node_index)
@ -58,6 +52,8 @@ async def test_join():
else: else:
assert str(libp2p_hosts[i].get_id()) not in gossipsubs[central_node_index].mesh[topic] assert str(libp2p_hosts[i].get_id()) not in gossipsubs[central_node_index].mesh[topic]
await cleanup()
@pytest.mark.asyncio @pytest.mark.asyncio
async def test_leave(): async def test_leave():
@ -128,11 +124,9 @@ async def test_dense():
await asyncio.sleep(0.5) await asyncio.sleep(0.5)
# Assert that all blocking queues receive the message # Assert that all blocking queues receive the message
items = []
for queue in queues: for queue in queues:
msg = await queue.get() msg = await queue.get()
assert msg.data == packet.publish[0].data assert msg.data == packet.publish[0].data
items.append(msg.data)
await cleanup() await cleanup()
@pytest.mark.asyncio @pytest.mark.asyncio