py-libp2p/tests/stream_muxer/conftest.py

import asyncio

import pytest

from libp2p.tools.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()])