Django单元测试：如何避免每个测试用例执行回滚？

Django默认的TestCase类会在每个测试用例前后开启事务，测试结束后自动回滚，以此保证测试之间的隔离性。如果要跳过回滚让数据在测试用例间复用，有以下几种可行方式：

1. 改用TransactionTestCase #

TransactionTestCase不依赖事务隔离，而是在每个测试用例执行完成后截断所有数据库表。如果想让多个测试共享初始化数据，可以在类的setUpClass()方法中创建数据——这个方法只会在所有测试方法执行前运行一次，且创建的数据不会被回滚（注意：如果某个测试修改了这些共享数据，后续测试会直接读到修改后的值，且所有测试结束后表会被截断）。

示例代码：

from django.test import TransactionTestCase
from myapp.models import MyModel

class SharedDataTest(TransactionTestCase):
    @classmethod
    def setUpClass(cls):
        super().setUpClass()
        # 创建共享数据，仅执行一次
        cls.shared_obj = MyModel.objects.create(name="test_shared_data")

    def test_first_access(self):
        self.assertEqual(MyModel.objects.count(), 1)

    def test_second_access(self):
        obj = MyModel.objects.get(id=self.shared_obj.id)
        self.assertEqual(obj.name, "test_shared_data")

2. 自定义测试类禁用自动回滚 #

如果想继续使用TestCase但跳过回滚逻辑，可以重写测试类的_post_teardown方法，跳过父类的事务回滚步骤。不过这种方式需要你手动处理测试后的清理工作，避免残留数据污染后续测试。

示例代码：

from django.test import TestCase
from myapp.models import MyModel

class NoRollbackTestCase(TestCase):
    def _post_teardown(self):
        # 跳过父类的事务回滚逻辑
        pass

    def tearDown(self):
        # 可选：手动清理数据，避免影响其他测试用例
        # MyModel.objects.all().delete()
        pass

注意事项 #

• 禁用自动回滚会打破测试隔离性，一个测试用例对数据的修改会直接影响其他测试，容易导致测试结果不稳定，非必要场景不推荐使用。
• 如果只是想复用初始化数据，更稳妥的方式是在setUp()中创建数据（每个测试用例都会重建，保证隔离），或者使用Django的fixtures功能加载固定测试数据。

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