forked from goodboy/tractor
69 lines
2.0 KiB
Python
69 lines
2.0 KiB
Python
"""
|
|
``tractor`` testing!!
|
|
"""
|
|
import random
|
|
import platform
|
|
|
|
import pytest
|
|
import tractor
|
|
from tractor.testing import tractor_test
|
|
|
|
|
|
pytest_plugins = ['pytester']
|
|
_arb_addr = '127.0.0.1', random.randint(1000, 9999)
|
|
|
|
|
|
def pytest_addoption(parser):
|
|
parser.addoption(
|
|
"--ll", action="store", dest='loglevel',
|
|
default=None, help="logging level to set when testing"
|
|
)
|
|
|
|
parser.addoption(
|
|
"--spawn-backend", action="store", dest='spawn_backend',
|
|
default='trio_run_in_process',
|
|
help="Processing spawning backend to use for test run",
|
|
)
|
|
|
|
|
|
def pytest_configure(config):
|
|
backend = config.option.spawn_backend
|
|
if backend == 'mp':
|
|
tractor._spawn.try_set_start_method('spawn')
|
|
elif backend == 'trio_run_in_process':
|
|
tractor._spawn.try_set_start_method(backend)
|
|
|
|
|
|
@pytest.fixture(scope='session', autouse=True)
|
|
def loglevel(request):
|
|
orig = tractor.log._default_loglevel
|
|
level = tractor.log._default_loglevel = request.config.option.loglevel
|
|
yield level
|
|
tractor.log._default_loglevel = orig
|
|
|
|
|
|
@pytest.fixture(scope='session')
|
|
def arb_addr():
|
|
return _arb_addr
|
|
|
|
|
|
def pytest_generate_tests(metafunc):
|
|
spawn_backend = metafunc.config.getoption("spawn_backend")
|
|
assert spawn_backend in ('mp', 'trio_run_in_process')
|
|
|
|
if 'start_method' in metafunc.fixturenames:
|
|
if spawn_backend == 'mp':
|
|
from multiprocessing import get_all_start_methods
|
|
methods = get_all_start_methods()
|
|
if 'fork' in methods: # fork not available on windows, so check before removing
|
|
# XXX: the fork method is in general incompatible with
|
|
# trio's global scheduler state
|
|
methods.remove('fork')
|
|
elif spawn_backend == 'trio_run_in_process':
|
|
if platform.system() == "Windows":
|
|
pytest.fail("Only `--spawn-backend=mp` is supported on Windows")
|
|
|
|
methods = ['trio_run_in_process']
|
|
|
|
metafunc.parametrize("start_method", methods, scope='module')
|