varun-r-mallya/py-libp2p
1
0
Fork 0
mirror of https://github.com/varun-r-mallya/py-libp2p.git synced 2026-02-11 23:51:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add tests for MplexStream

Browse Source
This commit is contained in:
mhchia
2019-09-18 21:51:09 +08:00
parent 02c55e5d14
commit 313ae45b45
4 changed files with 212 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
tests/stream_muxer/conftest.py
View File

@ -2,7 +2,7 @@ import asyncio
import pytest
from tests.factories import mplex_conn_pair_factory
from tests.factories import mplex_conn_pair_factory, mplex_stream_pair_factory
@pytest.fixture
@ -16,3 +16,14 @@ async def mplex_conn_pair(is_host_secure):
yield mplex_conn_0, mplex_conn_1
finally:
await asyncio.gather(*[swarm_0.close(), swarm_1.close()])
@pytest.fixture
async def mplex_stream_pair(is_host_secure):
mplex_stream_0, swarm_0, mplex_stream_1, swarm_1 = await mplex_stream_pair_factory(
is_host_secure
)
try:
yield mplex_stream_0, mplex_stream_1
finally:
await asyncio.gather(*[swarm_0.close(), swarm_1.close()])