如何通过pytest-mock对已导入函数实施桩化（patching）以模拟行为？

本文共计661个文字，预计阅读时间需要3分钟。

相关主题

在 pytest 中使用 mocker.patch 时，若函数已被 `from ... import ...` 导入，需 patch 当前模块中该函数的引用名（如 `f"{__name__}.get_a"`），而非其原始定义位置（如 `"mypackage.mymodule.get_a"`），否则打桩失效。

这是 Python 导入机制与 unittest.mock.patch 工作原理共同决定的关键行为：mock 总是作用于“被调用处所引用的对象”，而非“对象最初定义的位置”。

? 问题根源解析

当你执行：

from mypackage.mymodule import A, get_a

Python 将 mypackage.mymodule.get_a 的引用复制到当前模块的命名空间中，变量 get_a 成为一个独立的本地标识符。后续代码中调用 get_a() 实际访问的是 当前模块中的 get_a 名称绑定，而非动态查找 mypackage.mymodule.get_a。

因此，若你 patch "mypackage.mymodule.get_a"，仅修改了源模块中该属性（对当前模块无影响）；而真正被调用的 get_a 仍指向原函数 —— 所以 mock 失效。

✅ 正确做法是 patch 当前模块中该名称的绑定位置：

import pytest from mypackage.mymodule import A, get_a # ← get_a 现在是本模块的局部名 @pytest.fixture def mocked_A(mocker): a = A(2) # ✅ Patch where it's used: current module's 'get_a' mocker.patch(f"{__name__}.get_a", return_value=a) def test_mocked_a(mocked_A) -> None: a = get_a() # ← 调用的是本模块的 get_a（已被 patch） assert a.value == 2 # ✅ 通过

⚠️ 常见误区与对比

写法	是否生效	说明
mocker.patch("mypackage.mymodule.get_a", ...)	❌	仅 patch 源模块，不影响已导入的本地引用
mocker.patch(f"{__name__}.get_a", ...)	✅	正确 patch 当前模块中实际被调用的符号
mocker.patch("test_a.get_a", ...)	✅（但不推荐）	硬编码模块名，可读性差且易出错；推荐用 __name__ 动态获取

? 补充建议

• 若需 patch 多个导入函数，统一使用 f"{__name__}.xxx" 模式，保持一致性；
• 避免在测试文件中过度使用 from ... import ... —— 改用 import mypackage.mymodule 后显式调用 mypackage.mymodule.get_a()，此时可安全 patch "mypackage.mymodule.get_a"；
• 使用 pytest-mock 的 mocker fixture 时，其底层仍是 unittest.mock.patch，行为完全一致，无需额外适配。

掌握这一原则，即可精准控制 mock 作用域，避免“看似写了 patch 却毫无效果”的调试困境。