kombu/conftest.py

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)