Django：TextContextDecorator作为类装饰器时setUp失败不执行__exit__的Bug

问题场景 #

如果你在Django测试中把TextContextDecorator作为类装饰器使用，会遇到这个隐蔽的问题：

• 装饰器的enable()方法会在测试类的setUp()之前执行
• 正常情况下，对应上下文管理器__exit__方法的disable()会在tearDown()之后运行
• 但如果setUp()执行失败（抛出异常），Django测试框架会直接跳过tearDown()方法，导致disable()完全没机会执行，进而引发后续测试的异常状态

这其实是测试场景的共性问题：只要setUp()失败，绑定到tearDown()的清理逻辑都不会触发，依赖这个生命周期的上下文装饰器就会失效。

问题根源 #

Django测试 runner 的生命周期规则很严格：一旦setUp()失败，框架会判定测试环境未初始化完成，直接跳过测试方法和tearDown()，避免错误扩散。而TextContextDecorator作为类装饰器时，是把disable()绑定在tearDown()的后置流程里，所以tearDown()不跑，清理逻辑也就彻底缺位了。

解决方案：使用addCleanup()方法 #

修复方案核心是利用Django测试内置的addCleanup()方法——这个方法能保证无论测试成功、失败还是setUp崩溃，注册的清理逻辑都会执行。

下面是调整后的TextContextDecorator子类实现（针对类装饰器场景）：

from django.test import TestCase
from django.utils.decorators import TextContextDecorator

class 自定义上下文装饰器(TextContextDecorator):
    def enable(self):
        # 这里写你的启用逻辑，比如修改全局配置、初始化mock等
        self.原始配置 = 某个全局变量
        某个全局变量 = "测试用值"

    def disable(self):
        # 这里写你的清理逻辑，恢复原始状态
        某个全局变量 = self.原始配置

    def __call__(self, obj):
        # 专门处理类装饰器的情况
        if isinstance(obj, type):
            # 重写setUp方法，通过addCleanup绑定清理逻辑
            原始setUp = obj.setUp

            def 包装后的setUp(self):
                # 先启用上下文
                self.enable()
                # 注册清理方法，不管setUp成功与否都会执行
                self.addCleanup(self.disable)
                # 调用原来的setUp
                原始setUp(self)

            obj.setUp = 包装后的setUp
            return obj
        # 方法装饰器场景沿用父类逻辑
        return super().__call__(obj)

# 作为类装饰器使用
@自定义上下文装饰器()
class 我的测试类(TestCase):
    def setUp(self):
        # 故意抛出错误测试清理逻辑
        raise ValueError("模拟setUp执行失败")

    def test_测试方法(self):
        # 这个方法不会执行，但清理逻辑依然会跑
        pass

通过addCleanup()，我们把清理逻辑和tearDown()的生命周期解耦了。无论setUp()是否成功完成，测试runner都会执行注册的清理函数，确保测试环境在每次测试后都能恢复到初始状态。

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