mirror of https://github.com/celery/kombu.git
36 lines
1.0 KiB
Python
36 lines
1.0 KiB
Python
from __future__ import annotations
|
|
|
|
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)
|