如何在Python单元测试中递归Mock整个external模块？

针对你不想逐个mock.patch外部模块类的需求，这里有两种实用的批量Mock方法，适配不同场景：

方法1：直接Mock整个外部模块 #

如果你的类A是通过import external然后以external.B()的方式实例化外部类，直接Mock整个external模块是最省事的方式，模块内所有属性（包括类）都会返回Mock实例：

from unittest import mock
from my_module import A  # 你的类A所在模块

def test_A_init():
    with mock.patch('external', new=mock.Mock()) as mock_external:
        # 初始化A，此时所有external.B/C/D的实例化都会调用Mock对象
        a = A()
        
        # 可选：验证外部类是否被正确实例化
        mock_external.B.assert_called_once()
        mock_external.C.assert_called_once()

优缺点：

• 优势：零配置批量Mock，无需枚举类名
• 劣势：会Mock模块内所有内容（包括非类属性、函数），如果类A依赖模块内其他非类逻辑，需要额外手动恢复

方法2：精准Mock模块内所有类 #

如果只想Mock模块内的类，不影响其他属性，可以结合inspect模块筛选类，再用mock.patch.multiple批量处理：

场景1：类A通过import external调用外部类 #

from unittest import mock
import inspect
import external
from my_module import A

def test_A_init():
    # 筛选external模块内的所有类
    external_classes = {name: mock.Mock() for name, obj in inspect.getmembers(external, inspect.isclass)}
    
    with mock.patch.multiple('external', **external_classes):
        a = A()
        
        # 验证实例化调用
        external.B.assert_called_once()
        external.C.assert_called_once()

场景2：类A通过from external import B,C,D直接导入类 #

此时要遵循Mock的**"使用位置优先"**原则，Mock类A所在模块中的导入引用：

from unittest import mock
import inspect
from my_module import A
import external

def test_A_init():
    # 获取my_module中来自external模块的类
    target_module = __import__('my_module')
    external_imports = {}
    for name, cls in inspect.getmembers(target_module, inspect.isclass):
        if cls.__module__ == 'external':
            external_imports[name] = mock.Mock()
    
    with mock.patch.multiple('my_module', **external_imports):
        a = A()
        
        # 验证调用
        target_module.B.assert_called_once()

优缺点：

• 优势：仅Mock类，不干扰模块内其他逻辑，精准度高
• 劣势：需要少量代码筛选类，适合模块结构清晰的场景

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