根据使用的夹具添加 pytest 标记进行测试

Question

我正在寻找一种方法来使用标记标记 pytest 测试，具体取决于该测试使用的固定装置。

我想用它来根据他们使用的功能过滤测试。 例如：使用“database_connection”fixture 的测试应该自动标记为“database”。 这样，我可以根据是否有数据库凭据来包含或排除所有此类测试。

这是我目前拥有的：

def pytest_configure(config):
    """Register hook to extend the list of available markers."""
    config.addinivalue_line("markers", "database: mark test that need a database connection.")

def pytest_collection_modifyitems(config, items):  # pylint: disable=unused-argument
    """Register hook to map used fixtures to markers."""
    for item in items:
        if "database_connection" in item.fixturenames:
            database_marker = pytest.mark.database()
            item.add_marker(database_marker)

@pytest.fixture
def database_connection():
    """Fixture providing a database connection."""

这正是我想要的方式，但我不喜欢必须维护从夹具到标记的映射与夹具本身分开。 我想做的是用应该在所有使用它们的测试上设置的标记来装饰灯具。 它应该如下所示：

def pytest_configure(config):
    """Register hook to extend the list of available markers."""
    config.addinivalue_line("markers", "database: mark test that need a database connection.")

def pytest_collection_modifyitems(config, items):  # pylint: disable=unused-argument
    """Register hook to map used fixtures to markers."""
    for item in items:
        for fixture in item.fixtures:
            item.add_markers(fixture.markers)         

@pytest.fixture(markers=["database"])
def database_connection():
    """Fixture providing a database connection."""

当然，我可以构建一个将映射存储在全局变量中的装饰器：

_fixture_marker_map = {}

def set_markers(*markers):
    def decorator_set_markers(func):
        _fixture_marker_map[func.__name__] = markers
        
        @functools.wraps(func)
        def wrapper_set_markers(*args, **kwargs):
            return func(*args, **kwargs)
        return wrapper_set_markers
    return decorator_set_markers

@set_markers("database")
@pytest.fixture
def database_connection():
    """Fixture providing a database connection."""

但是，这感觉有点hacky。 我很确定这不是一个奇怪的用例，可能有一些 pytest 功能已经提供了我需要的东西。

有谁知道如何以简单的方式实现这一点？

Answer 1

有一个简单的解决方案似乎有效。 诀窍是注意您可以参数化夹具并标记参数。 您可以忽略该参数。 例如：

@pytest.fixture(params=[pytest.param(None, marks=pytest.mark.db))
def database_connection(request):
    """Fixture providing a database connection."""
    ...


def test_db_connection(database_connection):
    ...

将使用db标记自动标记任何使用database_connection的测试。