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

Add tests for SwarmConn

Browse Source
This commit is contained in:
mhchia
2019-09-17 23:38:11 +08:00
parent b8b5ac5e06
commit d61327f5f9
5 changed files with 79 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
tests/network/conftest.py
View File

@ -2,7 +2,11 @@ import asyncio
import pytest
from tests.factories import net_stream_pair_factory, swarm_pair_factory
from tests.factories import (
net_stream_pair_factory,
swarm_conn_pair_factory,
swarm_pair_factory,
)
@pytest.fixture
@ -21,3 +25,12 @@ async def swarm_pair(is_host_secure):
yield swarm_0, swarm_1
finally:
await asyncio.gather(*[swarm_0.close(), swarm_1.close()])
@pytest.fixture
async def swarm_conn_pair(is_host_secure):
conn_0, swarm_0, conn_1, swarm_1 = await swarm_conn_pair_factory(is_host_secure)
try:
yield conn_0, conn_1
finally:
await asyncio.gather(*[swarm_0.close(), swarm_1.close()])