Python的pytest框架(6)--测试钩子 (hooks)

import pytest

@pytest.hookimpl(tryfirst=True)
def pytest_runtest_logstart(nodeid):
    print(f"Test starting: {nodeid.split('::')[-1]}")

@pytest.hookimpl(tryfirst=True)
def pytest_runtest_logfinish(nodeid):
    print(f"Test finished: {nodeid.split('::')[-1]}")

这里的tryfirst=True参数表示这个钩子函数将在其他钩子函数之前执行。

4. 具体使用实例

在conftest.py文件中定义钩子：

我们通过一个完整的例子来展示如何在pytest中使用钩子函数来控制测试流程。

创建一个简单的测试场景，其中包括：

自定义fixture：用于演示基础的测试资源管理。

测试钩子：在测试开始前打印消息、在每个测试用例前后执行特定操作，并在测试结束时生成一个简短的报告。

确保我们的环境中安装了pytest。接下来，创建以下文件结构：

python 复制代码

my_project/
│
├── tests/
│   ├── conftest.py
│   └── test_sample.py
└── requirements.txt

conftest.py

在tests/conftest.py文件中，我们定义钩子函数和一个fixture：

python 复制代码

# tests/conftest.py

import pytest

# 自定义fixture
@pytest.fixture(scope='function')
def my_fixture():
    print("\nSetting up my_fixture...")
    yield
    print("Tearing down my_fixture...")

# 测试开始前的钩子
def pytest_sessionstart(session):
    print("pytest_sessionstart: 开始整个测试会话，准备环境...")

# 每个测试用例开始前的钩子
def pytest_runtest_setup(item):
    print(f"pytest_runtest_setup: 准备执行测试 {item.nodeid}")

# 每个测试用例结束后的钩子
def pytest_runtest_teardown(item, nextitem):
    print(f"pytest_runtest_teardown: 清理测试 {item.nodeid}, 下一个测试: {nextitem.nodeid if nextitem else '无'}")

# 整个测试会话结束后的钩子
def pytest_sessionfinish(session, exitstatus):
    print(f"pytest_sessionfinish: 测试会话结束，退出状态码: {exitstatus}")
    print("执行了一些清理工作...")

# 使用pytest_report_teststatus来改变测试结果的输出文本和颜色：
def pytest_report_teststatus(report, config):
    if report.when == 'call':
        if report.outcome == 'passed':
            # 绿色表示通过
            return ("通过", f"\033[32m{report.outcome}\033[0m", "")
        elif report.outcome == 'failed':
            # 红色表示失败，并附加失败信息的第一行
            return ("失败", f"\033[31m{report.outcome}: {report.longrepr.splitlines()[0]}\033[0m", "")
        elif report.outcome == 'skipped':
            # 蓝色表示跳过
            return ("跳过", f"\033[34m{report.outcome}: {report.wasxfail}\033[0m", "")
#颜色代码\033[XXm和\033[0m是ANSI转义序列，用于控制终端文本的颜色和样式

test_sample.py

然后，在tests/test_sample.py中编写我们的测试用例，使用上面定义的fixture：

python 复制代码

# tests/test_sample.py

import pytest

def test_case1(my_fixture):
    assert True, "测试用例1通过"

def test_case2(my_fixture):
    assert False, "测试用例2失败，期望断言失败"

执行测试

保存以上文件后，在项目根目录下运行pytest：

python 复制代码

pytest tests/

输出示例

执行上述命令后，你将看到类似以下的输出，展示了不同钩子函数在测试过程中的调用时机：

python 复制代码

pytest_sessionstart: 开始整个测试会话，准备环境...
collecting ... collected 2 items

test_sample.py 
Setting up my_fixture...
pytest_runtest_setup: 准备执行测试 tests/test_sample.py::test_case1
Tearing down my_fixture...
pytest_runtest_teardown: 清理测试 tests/test_sample.py::test_case1, 下一个测试: tests/test_sample.py::test_case2
Setting up my_fixture...
pytest_runtest_setup: 准备执行测试 tests/test_sample.py::test_case2
Tearing down my_fixture...
pytest_runtest_teardown: 清理测试 tests/test_sample.py::test_case2, 下一个测试: 无
pytest_sessionfinish: 测试会话结束，退出状态码: 1
执行了一些清理工作...

============================== 1 passed, 1 failed in 0.05s ==============================

这个例子展示了如何使用pytest的钩子函数来控制测试流程，包括设置和清理测试环境、以及在测试会话的各个阶段执行自定义操作。

自定义插件来定义和使用测试钩子

1-创建插件

首先，创建一个新的Python模块（例如，命名为my_custom_plugin.py），并在其中定义你的钩子函数：

python 复制代码

# my_custom_plugin.py

import pytest

def pytest_sessionstart(session):
    """
    在整个测试会话开始时被调用。
    """
    print(">>> 自定义插件：测试会话开始，执行初始化操作...")

def pytest_sessionfinish(session, exitstatus):
    """
    在整个测试会话结束时被调用。
    """
    print(">>> 自定义插件：测试会话结束，执行清理操作...")

2-使用自定义插件

要在测试套件中使用这个插件，有几种方法：

方法 A: 在命令行中指定插件

当你从命令行运行pytest时，可以通过-p或--plugin选项指定你的插件：

python 复制代码

pytest -p my_custom_plugin tests/

方法 B: 在pytest配置文件中指定插件

你也可以在pytest的配置文件（如pytest.ini、tox.ini或setup.cfg）中声明你的插件：

python 复制代码

# pytest.ini 或其他配置文件
[pytest]
addopts = -p my_custom_plugin

为了保持项目的组织性和钩子的可重用性，推荐的做法还是将它们放在适当的conftest.py文件中。这样不仅便于管理，也使得项目结构更加清晰。如果需要对特定测试文件或目录应用更为细致的控制，可以在该目录下放置一个conftest.py来局部覆盖或扩展全局的钩子定义。

希望以上内容能帮助大家合理有效使用测试钩子