37 lines
914 B
Python
37 lines
914 B
Python
import asyncio
|
|
|
|
import pytest
|
|
|
|
from tests.factories import (
|
|
net_stream_pair_factory,
|
|
swarm_conn_pair_factory,
|
|
swarm_pair_factory,
|
|
)
|
|
|
|
|
|
@pytest.fixture
|
|
async def net_stream_pair(is_host_secure):
|
|
stream_0, host_0, stream_1, host_1 = await net_stream_pair_factory(is_host_secure)
|
|
try:
|
|
yield stream_0, stream_1
|
|
finally:
|
|
await asyncio.gather(*[host_0.close(), host_1.close()])
|
|
|
|
|
|
@pytest.fixture
|
|
async def swarm_pair(is_host_secure):
|
|
swarm_0, swarm_1 = await swarm_pair_factory(is_host_secure)
|
|
try:
|
|
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()])
|