本文介绍如何使用 unittest.mock.patch 在跨模块调用链中全局拦截并统一替换目标函数（如 get_args），确保所有层级的调用均返回相同模拟对象，避免因模块级导入导致的 patch 失效问题。
本文介绍如何使用 `unittest.mock.patch` 在跨模块调用链中全局拦截并统一替换目标函数（如 `get_args`），确保所有层级的调用均返回相同模拟对象，避免因模块级导入导致的 patch 失效问题。
在 Python 单元测试或临时调试中，常需对全局状态函数（如 get_args()）进行模拟，使其在一次执行上下文中始终返回定制化的对象。但若该函数被多个模块分别导入（如 mod1.py 和 mod2.py 均执行 from mod0 import get_args），直接 patch 某一模块中的引用（如 'mod1.get_args'）仅影响该模块内调用，无法覆盖其他模块中独立导入的同名函数——这是由 Python 的导入机制决定的：每个模块持有自己对 get_args 的引用副本。
要实现“一次 patch，处处生效”，关键在于：对所有实际调用 get_args 的模块，分别 patch 其本地引用。即：不是 patch 函数定义处（mod0.get_args），而是 patch 所有 使用方模块 中的导入路径。
以下为推荐方案（基于原示例优化）：
from argparse import Namespace
import unittest.mock
from mod0 import get_args
from mod1 import f1
class _MockArgsProvider:
def __init__(self):
# 复用原始 args 结构，仅覆盖所需字段
base = get_args()
self._args = Namespace(**{**base.__dict__, 'a': 'a'})
def __call__(self):
print("new_get_args was called")
return self._args
def main():
provider = _MockArgsProvider() # 单例实例，确保返回同一对象
# 同时 patch 所有调用方模块中的 get_args 引用
with unittest.mock.patch('mod1.get_args', new=provider), \
unittest.mock.patch('mod2.get_args', new=provider):
f1()✅ 优势说明：
- 使用同一可调用实例 provider（而非重复调用 provider()）保证 f1 和 f2 获取的是完全相同的 Namespace 对象（id 一致），支持后续原地修改（如 args.c = 3）被所有函数共享；
- 语法简洁：patch(..., new=...) 直接注入，无需手动管理 return_value；
- 易扩展：新增调用模块（如 mod3.py）只需追加一行 unittest.mock.patch('mod3.get_args', new=provider)。
⚠️ 注意事项：
- ❌ 不要 patch 定义模块（如 'mod0.get_args'）——除非所有模块都通过 import mod0; mod0.get_args() 方式调用，而本例中是 from mod0 import get_args，此时 patch mod0.get_args 无效；
- ✅ 确保 patch 路径与模块中实际使用的导入方式严格匹配（from mod0 import get_args → patch 'mod1.get_args'，而非 'mod0.get_args'）；
- ? 若需动态行为（如每次调用返回不同值），可将 provider 改为带状态的类，或使用 side_effect；
- ? 在真实测试中，建议将 patch 逻辑封装进 unittest.TestCase 的 setUp 或使用 @patch 装饰器，提升可维护性。
综上，跨模块函数模拟的本质是“按需 patch 导入点”，而非“patch 定义点”。理解 Python 模块导入与符号绑定机制，是精准控制 mock 行为的基础。
