Django如何基于已填充数据库搭建单元测试环境？

此前大多基于模型的全新实例运行单元测试，但由于重新填充数据库操作难度较高，希望基于已填充数据的模型开展测试。请问该需求是否可行？目前我未找到相关官方文档，当前使用的测试代码如下：

def setUpTestData(cls):
    create test data for model 1
    create test data for model 2

def tests run some tests

当然可行！这是非常实用的测试方案 #

这在实际项目测试中是很常见的做法，虽然官方文档可能没专门单独拎出来讲解，但测试框架完全支持这种方式，而且能帮你省去重复创建测试数据的麻烦。下面是具体的调整思路和注意事项：

• 放弃手动创建数据的逻辑：你可以移除setUpTestData里手动生成数据的代码，直接在测试方法中读取已填充好的数据库数据即可。
• 确保测试隔离性：这是关键！要保证预填充的测试数据是固定不变的，不会被其他测试用例修改导致结果不稳定。建议专门用一个测试专用数据库，每次测试前可以通过快照恢复、备份脚本重置等方式，把数据库还原到预填充的初始状态。
• 精准定位测试数据：测试时不要依赖动态查询（比如Model1.objects.first()），而是用固定的唯一标识来获取实例，比如主键、唯一slug或编码：

  test_obj = Model1.objects.get(id=1)
  # 或者
  test_obj = Model1.objects.get(unique_slug="test-example")
  

调整后的测试代码示例 #

from django.test import TestCase
from myapp.models import Model1, Model2

class MyModelTests(TestCase):
    # 不再在setUpTestData中创建数据，直接使用预填充的数据库
    def test_model1_name_validation(self):
        # 获取预填充的Model1实例
        test_item = Model1.objects.get(id=1)
        # 执行测试逻辑
        self.assertEqual(test_item.name, "预填充的测试名称")
    
    def test_model2_relationship_integrity(self):
        test_item2 = Model2.objects.get(unique_code="TEST_001")
        # 验证关联关系
        self.assertIsNotNone(test_item2.related_model1)

额外小贴士 #

• 如果担心测试过程中不小心修改了预填充数据，可以用事务回滚来保证数据干净：比如给每个测试方法加上@transaction.atomic()装饰器，或者在tearDown方法里执行数据恢复操作。
• 可以把预填充数据的生成脚本（比如框架对应的fixture文件、SQL导入脚本）和测试代码放在一起，方便团队成员快速搭建一致的测试环境。

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