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

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

14
tests/pubsub/factories.py
View File

@ -1,17 +1,21 @@
import factory
from libp2p import initialize_default_swarm
from libp2p.host.basic_host import BasicHost
from libp2p.pubsub.floodsub import FloodSub
from libp2p.pubsub.gossipsub import GossipSub
from libp2p.pubsub.pubsub import Pubsub
from tests.configs import LISTEN_MADDR
from .configs import FLOODSUB_PROTOCOL_ID, GOSSIPSUB_PARAMS, GOSSIPSUB_PROTOCOL_ID
from tests.configs import (
FLOODSUB_PROTOCOL_ID,
GOSSIPSUB_PARAMS,
GOSSIPSUB_PROTOCOL_ID,
LISTEN_MADDR,
)
from tests.utils import generate_new_private_key
def swarm_factory():
return initialize_default_swarm(transport_opt=[str(LISTEN_MADDR)])
private_key = generate_new_private_key()
return initialize_default_swarm(private_key, transport_opt=[str(LISTEN_MADDR)])
class HostFactory(factory.Factory):