如何编写可批量应用多个patch装饰器的函数/装饰器？

方法1：自定义组合装饰器 #

直接把多个patch实例打包成一个装饰器，不用在每个测试函数上重复写一堆@patch。示例代码：

from unittest import mock
from functools import wraps

def combine_patches(*patches):
    def decorator(test_func):
        # 倒序应用patch，和直接写多个@patch的执行顺序保持一致
        for patcher in reversed(patches):
            test_func = patcher(test_func)
        @wraps(test_func)
        def wrapper(*args, **kwargs):
            return test_func(*args, **kwargs)
        return wrapper
    return decorator

# 提前定义好所有需要的patch
COMMON_PATCHES = [
    mock.patch('your_module.TargetClass'),
    mock.patch('your_module.target_function'),
    mock.patch('your_module.TARGET_CONST', 456)
]

# 直接用组合后的装饰器
@combine_patches(*COMMON_PATCHES)
def test_my_case(mock_cls, mock_func, mock_const):
    # 测试逻辑
    mock_cls.assert_called_once_with('test_arg')
    assert mock_const == 456

这里必须倒序遍历patches，因为Python装饰器是从下往上执行的，和你手动写多个@patch的顺序逻辑一致，避免mock参数顺序混乱。

方法2：类装饰器封装 #

如果需要动态调整patch集合，用类装饰器更灵活：

from unittest import mock

class MultiPatch:
    def __init__(self, *patches):
        self.patches = patches

    def __call__(self, test_func):
        for patcher in reversed(self.patches):
            test_func = patcher(test_func)
        return test_func

# 用法示例
@MultiPatch(
    mock.patch('your_module.TargetClass'),
    mock.patch('your_module.target_function'),
    mock.patch('your_module.TARGET_CONST', 456)
)
def test_another_case(mock_cls, mock_func, mock_const):
    # 测试逻辑
    pass

方法3：pytest fixture批量patch（适用于pytest环境） #

如果用pytest做测试，用fixture管理批量patch更省心，不用写装饰器：

import pytest
from unittest import mock

@pytest.fixture
def common_patches():
    # 用上下文管理器批量启动patch
    with mock.patch('your_module.TargetClass') as mock_cls, \
         mock.patch('your_module.target_function') as mock_func, \
         mock.patch('your_module.TARGET_CONST', 456) as mock_const:
        yield mock_cls, mock_func, mock_const

# 测试函数直接引用fixture
def test_with_fixture(common_patches):
    mock_cls, mock_func, mock_const = common_patches
    # 测试逻辑
    mock_func.assert_called()

你之前踩坑的原因分析 #

1. 方案1测试未执行：大概率是装饰器写法错误，比如没返回可调用的测试函数、漏掉@wraps导致测试框架识别不出测试用例。
2. 方案3patch未生效：要么是装饰器顺序搞反了（没倒序应用patch），要么是patch的目标路径写错了（比如用了相对路径而非绝对导入路径）。

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