如何用单个Mock变量统一Mock多文件中的同一Python函数？

不需要写多个独立的Mock，可以通过复用单个Mock实例+批量上下文管理的方式实现统一Mock，避免代码冗余。

方法1：使用ExitStack批量管理patch上下文 #

这是最灵活的方式，适合需要Mock大量跨模块目标的场景：

from contextlib import ExitStack
from unittest.mock import patch, Mock
from script import main

def test_unified_mock():
    # 创建单个Mock实例，统一控制所有被替换的get_one
    mock_get_one = Mock(return_value=0)
    # 列出所有需要Mock的目标路径，可按需扩展
    mock_targets = ["file1.get_one", "file2.get_one"]

    with ExitStack() as stack:
        # 遍历目标列表，逐个绑定同一个Mock实例
        for target in mock_targets:
            stack.enter_context(patch(target, mock_get_one))
        # 执行测试逻辑
        main(2, 3)

    # 验证统一Mock的调用次数
    assert mock_get_one.call_count == 5

方法2：复用Mock的装饰器写法 #

如果Mock目标数量不多，也可以用多个@patch装饰器绑定同一个Mock实例：

from unittest.mock import patch, Mock
from script import main

# 定义统一Mock实例
mock_get_one = Mock(return_value=0)

@patch("file1.get_one", mock_get_one)
@patch("file2.get_one", mock_get_one)
def test_mock_with_decorators():
    main(2, 3)
    assert mock_get_one.call_count == 5

关键说明 #

• 两种方法的核心都是让所有需要Mock的get_one引用指向同一个Mock实例，这样就能统一统计调用次数、设置返回值等。
• 之所以不能直接像你示例中那样给patch传多个目标，是因为unittest.mock.patch的接口设计只支持单个目标，但通过批量绑定同一个Mock可以实现等效的统一控制效果。

内容的提问来源于stack exchange，提问作者Johnny Metz