kombu/conftest.py

34 lines
1.0 KiB
Python

import pytest
def pytest_addoption(parser):
parser.addoption(
"-E",
action="append",
metavar="NAME",
help="only run tests matching the environment NAME.",
)
def pytest_configure(config):
# register an additional marker
config.addinivalue_line(
"markers",
"env(name): mark test to run only on named environment",
)
config.addinivalue_line("markers", "replace_module_value")
config.addinivalue_line("markers", "masked_modules")
config.addinivalue_line("markers", "ensured_modules")
config.addinivalue_line("markers", "sleepdeprived_patched_module")
def pytest_runtest_setup(item):
envnames = [mark.args[0] for mark in item.iter_markers(name='env')]
if envnames:
if (
item.config.getoption("-E") is None
or len(set(item.config.getoption("-E")) & set(envnames)) == 0
):
# We skip test if does not mentioned by -E param
pytest.skip("test requires env in %r" % envnames)