30 lines
802 B
Python
30 lines
802 B
Python
import asyncio
|
|
|
|
import pytest
|
|
|
|
from tests.factories import mplex_conn_pair_factory, mplex_stream_pair_factory
|
|
|
|
|
|
@pytest.fixture
|
|
async def mplex_conn_pair(is_host_secure):
|
|
mplex_conn_0, swarm_0, mplex_conn_1, swarm_1 = await mplex_conn_pair_factory(
|
|
is_host_secure
|
|
)
|
|
assert mplex_conn_0.is_initiator
|
|
assert not mplex_conn_1.is_initiator
|
|
try:
|
|
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()])
|