pytest后置处理方式

在 pytest 中，确保后置清理代码一定会执行（无论测试通过、失败还是发生异常），推荐使用以下两种最稳健的方式：

1. 使用yield的 fixture（推荐）

在 fixture 的yield之后编写清理代码，pytest 会在测试结束后自动运行这些代码，即使测试断言失败或抛出异常。

import pytest @pytest.fixture def db_connection(): # 前置：建立连接 conn = create_connection() yield conn # 后置：一定会执行（除非 fixture 前置阶段就挂了） conn.close() def test_query(db_connection): assert db_connection.query("SELECT 1") == 1

⚠️ 注意：如果yield之前的代码（建立连接）抛出了异常，那么yield之后的清理代码不会执行，因为 fixture 本身没有成功初始化。这是合理的——资源从未被成功获取，自然无需释放。

2. 使用request.addfinalizer

功能与yield类似，但以注册回调函数的方式显式声明清理逻辑。

import pytest @pytest.fixture def db_connection(request): conn = create_connection() def cleanup(): conn.close() request.addfinalizer(cleanup) # 注册后置函数 return conn

同样，addfinalizer注册成功后的清理函数一定会执行。

3. 在测试类或模块中使用传统teardown方法

适用于经典的 xUnit 风格，pytest 完全支持：

• 方法级：def teardown_method(self, method):

• 类级：def teardown_class(cls):

• 模块级：def teardown_module(module):

class TestDatabase: def setup_method(self): self.conn = create_connection() def teardown_method(self): self.conn.close() # 每个测试方法结束后一定会执行 def test_query(self): assert self.conn.query("SELECT 1") == 1

4. 如果需要“无论如何（即使 fixture 设置失败）都执行清理”

如果存在必须执行的全局资源回收（例如临时文件、进程等），可以在 fixture 中使用try/finally：

@pytest.fixture def temp_file(): f = None try: f = open("/tmp/test.txt", "w") yield f finally: if f: f.close() # 或者无条件删除文件

但这种情况较少见，因为通常设置失败时资源尚未分配，无需清理。