Pytest动态为用例添加mark标记

1. Pytest动态为用例添加mark标记

在Pytest中，我们可以使用mark标记来对测试用例进行分类和标记。mark标记可以用来进行用例的过滤、分组和跳过等操作。而有时候，我们可能需要动态地为测试用例添加mark标记。下面我们将介绍如何实现这一功能。

1.1 使用fixture动态为用例添加mark标记

在Pytest中，fixture是一种用于管理测试用例执行前后的资源的机制。使用fixture可以在测试用例执行前后进行一些操作，如准备测试数据、清理资源等。fixture还可以通过使用pytest.mark标记来动态地为用例添加标记。

首先，我们需要定义一个fixture来实现动态添加mark标记。我们可以使用pytest的hook函数 pytest_collection_modifyitems来实现这个功能。该函数在收集到测试用例后，可以改变测试用例上的mark标记。

def pytest_collection_modifyitems(config, items):

    for item in items:
        if 'condition' in item.keywords:
            item.add_marker(pytest.mark.slow)
        elif 'data' in item.keywords:
            item.add_marker(pytest.mark.data)

上述代码中，我们通过判断测试用例的keywords是否包含特定的关键字，然后动态地为用例添加对应的mark标记。比如，如果测试用例的keywords中包含'condition'，则为该用例添加pytest.mark.slow标记；如果keywords中包含'data'，则为该用例添加pytest.mark.data标记。

通过这种方式，我们可以根据自定义的条件来动态地为测试用例添加标签。

1.2 使用装饰器动态为用例添加mark标记

除了使用fixture外，我们还可以通过定义装饰器来动态地为测试用例添加mark标记。

import pytest

def slow_test(func):
    func = pytest.mark.slow(func)
    return func
def test_example():
    assert 1 + 1 == 2
@slow_test
def test_slow_example():
    assert 2 * 2 == 4

在上述代码中，我们定义了一个装饰器slow_test，用于将被装饰的测试用例添加pytest.mark.slow标记。我们可以使用@slow_test装饰器来动态地为用例添加标记。在上面的例子中，test_slow_example函数被@slow_test装饰器修饰，所以它会被添加pytest.mark.slow标记。

通过使用装饰器，我们可以根据不同的需求为不同的测试用例动态地添加标记，以便进行分类和管理。

2. 示例代码

下面是一个完整的示例代码，演示了如何使用fixture和装饰器动态地为测试用例添加mark标记：

import pytest

def pytest_collection_modifyitems(config, items):
    for item in items:
        if 'condition' in item.keywords:
            item.add_marker(pytest.mark.slow)
        elif 'data' in item.keywords:
            item.add_marker(pytest.mark.data)
def slow_test(func):
    func = pytest.mark.slow(func)
    return func
def test_normal_example():
    assert 1 + 1 == 2
def test_condition_example():
    assert 2 * 2 == 4
@slow_test
def test_slow_example():
    assert 3 - 1 == 2
@pytest.mark.data
def test_data_example():
    assert 4 / 2 == 2

在这个示例代码中，我们定义了四个测试用例，分别为test_normal_example、test_condition_example、test_slow_example和test_data_example。其中test_slow_example和test_data_example用了装饰器slow_test和pytest.mark.data来动态地添加mark标记。

通过上述代码，我们可以在运行测试用例时，根据不同的条件和需求，动态地为用例添加mark标记，以便进行分类和管理。

综上所述，我们可以通过使用fixture和装饰器的方式，动态地为测试用例添加mark标记。这样可以方便地对用例进行分类和管理，提高测试用例的可读性和可维护性。