编写 pytest 插件的解决方法非常简单,下面是一个示例:
pytest_example_plugin
。pytest_example_plugin.py
。pytest_example_plugin.py
文件中添加以下代码:import pytest
def pytest_report_header(config):
return "这是一个 pytest 插件示例"
@pytest.hookimpl(tryfirst=True)
def pytest_collection_modifyitems(config, items):
for item in items:
if "example" in item.nodeid:
item.add_marker(pytest.mark.example)
def pytest_terminal_summary(terminalreporter):
passed = len(terminalreporter.stats.get("passed", []))
failed = len(terminalreporter.stats.get("failed", []))
skipped = len(terminalreporter.stats.get("skipped", []))
terminalreporter.write_line(f"通过的测试用例:{passed}")
terminalreporter.write_line(f"失败的测试用例:{failed}")
terminalreporter.write_line(f"跳过的测试用例:{skipped}")
在上面的示例中,我们定义了三个 pytest 插件的钩子函数:
pytest_report_header
:用于向测试报告的头部添加一行信息。pytest_collection_modifyitems
:用于修改收集到的测试用例集合。在这个示例中,我们给包含 "example" 的测试用例添加了一个自定义的 marker,名为 example
。pytest_terminal_summary
:用于在测试结果的总结部分添加一些自定义的信息,比如通过的测试用例数量、失败的测试用例数量和跳过的测试用例数量。tests
文件夹,并在其中创建一个测试文件,命名为test_example.py
。test_example.py
文件中添加以下代码:import pytest
@pytest.mark.example
def test_example():
assert 1 + 1 == 2
def test_another_example():
assert 2 * 2 == 4
在上面的示例中,我们使用了之前定义的 example
marker 来标记一个测试用例,这个测试用例会被 pytest 插件识别并执行。
pytest_example_plugin
文件夹,并执行以下命令运行测试:pytest
你会发现在测试结果的头部添加了一行自定义的信息 "这是一个 pytest 插件示例",并且在总结部分显示了通过、失败和跳过的测试用例数量。
这就是一个简单的 pytest 插件的示例,你可以根据自己的需求编写更复杂的插件。