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) 14. 如果需要“无论如何即使 fixture 设置失败都执行清理”如果存在必须执行的全局资源回收例如临时文件、进程等可以在 fixture 中使用try/finallypytest.fixture def temp_file(): f None try: f open(/tmp/test.txt, w) yield f finally: if f: f.close() # 或者无条件删除文件但这种情况较少见因为通常设置失败时资源尚未分配无需清理。方式是否保证后置执行备注yieldfixture✅前置成功时最推荐代码简洁request.addfinalizer✅前置成功时与 yield 等价teardown_*方法✅对应作用域适用于传统风格try/finally✅无条件适合必须清理的场景