python - 仅在使用 --collect-only 时运行方法

Question

我正在尝试收集一些关于每个功能或团队有多少测试用例的统计数据，并且只想在收集期间执行此操作，而不是在我公司的每个人都运行他们的测试时执行此操作。我正在寻找一种仅当 --collect-only 在 py.test 命令中使用时才运行函数的方法。

目标:

• 仅当使用了--collect-only 时才运行一些代码
• 我需要访问测试 items 数据结构(即我需要所有的标记用过)

目前，我正在通过 pytest_collection_modifyitems Hook 执行此操作:

def pytest_collection_modifyitems(config, items):
    # This hook runs after collection
    for item in items:
        teams = [mark.name for mark in item.iter_markers() if mark.name.startswith('team_')]
        features = [mark.name for mark in item.iter_markers() if mark.name.startswith('feature_')]

        # do something with these markers

当且仅当 py.test 使用 --collect-only 运行时，是否有办法运行上面的代码？如果有人对执行此操作的更好方法有建议，请提供帮助!

谢谢!

Answer 1

只需检查是否在 config 对象中设置了标志。使用您的 hookimpl 的示例:

def pytest_collection_modifyitems(config, items):
    if config.option.collectonly:
        print("I will run only with --collectonly flag")
    else:
        print("I will run only without --collectonly flag")