pytest_collection_modifyitem对用例进行过滤

z917185537

于 2023-12-14 18:03:31 发布

阅读量699

点赞数 8

本文链接：https://blog.youkuaiyun.com/z917185537/article/details/135000577

版权

python 同时被 2 个专栏收录

43 篇文章

订阅专栏

backup

8 篇文章

订阅专栏

文章介绍了pytest_collection_modifyitems钩子在pytest中的应用，如何根据配置和用例特性进行过滤（如在headless模式下跳过only_headed用例），以及如何动态地为用例添加interface和event标记。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

pytest_collection_modifyitem在收集完用例后执行，通过此hooks，在conftest.py中可以实现对用例的过滤和重新排序

根据用例标记和执行参数标记用例是否跳过( --headless运行时，并且标记only_headed的用例跳过）

def pytest_collection_modifyitems(config, items):
    '''
    skip the only_headed test case, if run in headless mode
    '''
    only_headed = config.getoption("--headed")
    skip_headed_parameters = pytest.mark.skip(reason="only run in headed mode")
    for item in items:  # only skip the test which mark.only_headed
        if not only_headed and "only_headed" in item.keywords:
            item.add_marker(skip_headed_parameters)

根据用例动态的批量添加标记

def pytest_collection_modifyitems(items):
    for item in items:
        if "interface" in item.nodeid: #用例nodeid中包含interface时标记interface
            item.add_marker(pytest.mark.interface)
        elif "event" in item.nodeid:  #用例nodeid中包含event时标记event
            item.add_marker(pytest.mark.event)