如何使用Python pytest测试带有上下文管理器（with）的代码？-Python教程-PHP中文网

最稳妥方式是直接 patch __enter__ 和 __exit__ 方法，而非 mock 整个类或实例；需确保 patch 路径为被测模块中导入该类的位置，__enter__ 返回值匹配 as 变量类型，__exit__ 默认设 return_value=false 以支持异常传播。

如何使用python pytest测试带有上下文管理器（with）的代码？

如何用 pytest 模拟 with 语句中的上下文管理器？

直接 patch __enter__ 和 __exit__ 是最稳妥的方式，而不是 mock 整个类或实例。因为 with 语句底层依赖这两个方法的调用协议，mock 错对象会导致 AttributeError: __enter__ 或静默失败。

用 unittest.mock.patch 作用于被测模块中导入的类（不是定义处），例如 @patch('mymodule.MyContextManager.__enter__')
__enter__ 的返回值要匹配实际代码中 as xxx 绑定的变量类型，比如返回一个 mock 对象或具体值
__exit__ 默认返回 False（表示不抑制异常），若被测逻辑依赖异常传播，别忘了设 return_value=False
避免 patch 类本身——那样 __enter__ 可能根本不会被调用，pytest 会跳过上下文协议

测试 with 块内抛出异常的场景怎么写？

必须显式触发 __exit__ 并验证它是否收到对应异常信息，否则无法确认清理逻辑是否执行。

在 test 函数里手动调用 ctx.__enter__()，再用 ctx.__exit__(exc_type, exc_val, exc_tb) 模拟异常退出
更自然的做法是：用 with pytest.raises(ExpectedError): 包裹整个 with 块，并在 __exit__ 的 mock 上断言 mock_call_args 中的三个参数
注意：如果上下文管理器自身在 __exit__ 里吞掉了异常（返回 True），则 pytest.raises 会失败，此时应检查 __exit__ 的返回值逻辑

为什么 patch 路径写错就报 AttributeError: enter？

错误通常不是因为没 mock，而是 patch 目标位置不对——pytest 找不到你 mock 的那个 __enter__ 方法，于是运行时调用原始类的未 patch 版本，而该版本可能没实现协议。

Python 3.14.2

Python 3.14.2是Python编程语言在2025年12月5日发布的稳定版本，属于3.14系列的第二个维护更新。该版本包含了18项修复，重点解决了多进程、数据类及正则表达式等模块的回归问题，并修复了CVE-2025-12084等安全漏洞。此版本标志着自由线程模式（移除GIL）正式获得官方支持，是Python发展的重要里程碑。

下载

确认 patch 路径是「被测函数所在模块中引用该类的位置」，不是类定义所在的模块。例如 mymodule.py 里写了 from lib import DBConn，然后 with DBConn() as db:，那就要 patch 'mymodule.DBConn.__enter__'
用 print 或 debugger 检查被测函数内部实际构造的对象类型，防止路径指向了另一个同名但不同模块的类
如果类是动态生成或用了 __call__ 伪装成上下文管理器，__enter__/__exit__ 可能根本不存在，需改用其他方式验证行为