如何装饰unittest的TestCase类并让测试框架识别？

为什么使用装饰器后测试不被识别？ #

你的装饰器wrapper_for_test犯了核心错误：它接收测试类，但返回的是函数而非类。

unittest框架的测试发现机制只会扫描模块中继承自unittest.TestCase的类，不会将函数当作测试类处理。同时functools.wraps仅能保留函数的元信息，无法传递原测试类的继承关系和测试方法，导致框架找不到任何测试用例。

如何正确包装unittest TestCase类？ #

装饰器必须返回类（要么是原测试类的子类，要么是修改后的原类），同时保留原类的关键元信息，让测试框架能正常识别。

方案1：创建测试类子类添加全局逻辑 #

适合统一添加前置/后置操作等全局修改场景：

import unittest
import functools

def wrapper_for_test(test_case_cls):
    # 继承原测试类，添加自定义逻辑
    class WrappedTestCase(test_case_cls):
        def setUp(self):
            # 自定义前置操作
            print("执行测试前置逻辑")
            super().setUp()
        
        def tearDown(self):
            super().tearDown()
            # 自定义后置操作
            print("执行测试后置逻辑")
    
    # 保留原类的类名、文档等元信息
    functools.update_wrapper(WrappedTestCase, test_case_cls)
    return WrappedTestCase

@wrapper_for_test
class TestClass(unittest.TestCase):
    def test_example(self):
        self.assertTrue(False)

执行测试后，框架会识别到继承自TestCase的WrappedTestCase，正常执行测试用例并触发自定义逻辑。

方案2：直接修改原类的测试方法 #

适合给单个测试方法添加装饰（如日志、异常捕获）的场景：

import unittest
import functools

def test_method_decorator(func):
    @functools.wraps(func)
    def wrapper(self, *args, **kwargs):
        print(f"开始执行测试：{func.__name__}")
        try:
            result = func(self, *args, **kwargs)
            print(f"测试{func.__name__}执行成功")
            return result
        except Exception as e:
            print(f"测试{func.__name__}执行失败：{str(e)}")
            raise
    return wrapper

def wrapper_for_test(test_case_cls):
    # 遍历类中所有test_开头的方法，添加装饰器
    for attr_name in dir(test_case_cls):
        attr = getattr(test_case_cls, attr_name)
        if callable(attr) and attr_name.startswith("test_"):
            setattr(test_case_cls, attr_name, test_method_decorator(attr))
    return test_case_cls

@wrapper_for_test
class TestClass(unittest.TestCase):
    def test_example(self):
        self.assertTrue(False)

这种方式直接修改原测试类的方法并返回原类，测试框架能正常识别并执行装饰后的测试方法。

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