多线程环境下Mock外部库函数的集成测试问题

我明白你遇到的困扰了——在多线程场景下想用unittest.mock.patchMock外部库函数，结果子线程里还是跑了原函数。这背后主要是模块导入顺序和缓存的问题，咱们一步步来解决：

问题根源 #

当你在测试代码里先导入Bar（比如from bar import Bar），Python会立刻加载bar模块，而bar又会加载foo模块，这时候foo里已经把external_lib的原函数function_i_want_to_mock导入并绑定了。之后你再执行patch，虽然修改了foo模块的function_i_want_to_mock属性，但子线程里的Foo类已经持有了原函数的引用，所以Mock没生效。

解决方案 #

方案一：调整导入顺序，清除模块缓存（最推荐，无需修改被测代码） #

我们要确保在patch生效之后再导入被测模块，同时清除之前缓存的模块，让Python重新加载它们，这样新导入的Foo会直接使用Mock后的函数。

修改测试代码如下：

import sys
from unittest.mock import patch

def test_run():
    # 先清除已缓存的foo和bar模块，避免旧的导入影响
    for module_name in ['foo', 'bar']:
        if module_name in sys.modules:
            del sys.modules[module_name]
    
    with patch("foo.function_i_want_to_mock") as mock_func:
        mock_func.return_value = "Mocked Result"
        # 在patch生效后才导入Bar，确保Foo加载时用的是Mock函数
        from bar import Bar
        bar = Bar()
        bar.start()
        bar.join()
        
        # 验证Mock是否被调用，方便排查问题
        assert mock_func.called, "Mocked函数居然没被调用！"
        print(f"Mocked函数被调用了{mock_func.call_count}次")

if __name__ == '__main__':
    test_run()

方案二：直接patch外部库的函数（临时验证用） #

如果方案一没起作用，你可以尝试直接patchexternal_lib里的原函数，这样不管哪个模块引用它，都会指向Mock版本：

from unittest.mock import patch
from bar import Bar

def test_run():
    with patch("external_lib.function_i_want_to_mock") as mock_func:
        mock_func.return_value = "Mocked Result"
        bar = Bar()
        bar.start()
        bar.join()
        
        assert mock_func.called

不过这个方案不够精准，因为它会全局Mock这个函数，可能影响其他测试，所以只适合临时验证问题。

方案三：修改被测代码，明确引用模块级变量（不推荐，除非必要） #

如果不想调整测试代码，可以修改Foo.py的run_function，让它明确引用模块级的function_i_want_to_mock，确保每次调用都用最新的模块属性：

# Foo.py
import threading
import foo  # 导入自身模块
from external_lib import function_i_want_to_mock

class Foo(threading.Thread):
    def __init__(self):
        threading.Thread.__init__(self)
    def run(self):
        print(self.run_function())
    def run_function(self):
        return foo.function_i_want_to_mock()  # 明确用模块级变量

之后用你原来的测试代码就能生效，但修改被测代码来适配测试不是最佳实践，所以优先选方案一。

内容的提问来源于stack exchange，提问作者Moustafa Shama