Django单元测试：如何测试抽象模型？

我来分享一个既符合DRY原则又高效的解决方案——在测试代码中创建一个仅用于测试的临时具体模型，用它来承载抽象Mixin的测试逻辑，这样只需要写一次测试，就能覆盖Mixin的所有行为，不用在每个业务模型里重复造轮子。

具体实现步骤 #

1. 编写测试专用模型 #

在core/tests/test_behaviors.py文件里，定义一个简单的具体模型，让它继承你要测试的UniversallyUniqueIdentifiableMixin：

import uuid
from django.db import models
from django.test import TestCase
from core.behaviors import UniversallyUniqueIdentifiable

# 仅用于测试的临时模型，只保留必要字段
class TestUUIDModel(UniversallyUniqueIdentifiable):
    name = models.CharField(max_length=100)

    class Meta:
        # 设置managed=False，告诉Django不要为这个模型生成迁移文件
        managed = False
        # 指定临时表名，避免和其他测试表冲突
        db_table = 'test_uuid_model'

2. 编写针对性测试用例 #

有了这个测试模型，就可以直接针对Mixin的特性写测试了：

class UniversallyUniqueIdentifiableTestCase(TestCase):
    def test_uuid_is_auto_generated(self):
        # 验证实例创建时UUID会自动生成
        test_obj = TestUUIDModel(name="Test Item")
        test_obj.save()
        self.assertIsNotNone(test_obj.uuid)
        self.assertIsInstance(test_obj.uuid, uuid.UUID)

    def test_uuid_field_is_non_editable(self):
        # 验证UUID字段无法被修改
        test_obj = TestUUIDModel.objects.create(name="Test Item")
        original_uuid = test_obj.uuid
        # 尝试修改UUID并保存
        test_obj.uuid = uuid.uuid4()
        test_obj.save()
        # 从数据库重新获取，确认UUID未被改变
        refreshed_obj = TestUUIDModel.objects.get(pk=test_obj.pk)
        self.assertEqual(refreshed_obj.uuid, original_uuid)

    def test_uuid_has_db_index(self):
        # 验证UUID字段被设置了数据库索引
        uuid_field = TestUUIDModel._meta.get_field('uuid')
        self.assertTrue(uuid_field.db_index)

3. 关于临时模型的小细节 #

Django的TestCase会在每次测试前创建临时数据库，测试完成后自动销毁，所以即使你把managed设为True也不会污染正式数据库。不过设置managed=False更稳妥，能明确这是个测试专用模型，不会被误加入迁移文件。

为什么这种方法更优？ #

• 完全符合DRY原则：只需要写一组测试，就能覆盖Mixin的所有功能，不用在每个使用该Mixin的业务模型里重复写相同的测试逻辑。
• 测试隔离性强：测试只针对Mixin本身，不会受业务模型其他字段或业务逻辑的干扰，测试结果更准确。
• 维护成本低：如果后续Mixin的逻辑有修改，只需要更新这一组测试即可，不用修改多个业务模型的测试代码。

进阶技巧：用测试基类复用逻辑 #

如果你有多个抽象Mixin需要测试，可以创建一个测试基类，把通用的测试逻辑抽离出来：

class BaseUUIDMixinTestCase(TestCase):
    # 子类需要指定这个属性为对应的测试模型
    test_model = None

    def setUp(self):
        if not self.test_model:
            self.skipTest("请指定test_model属性")

    def test_uuid_auto_generation(self):
        test_obj = self.test_model(name="Test")
        test_obj.save()
        self.assertIsNotNone(test_obj.uuid)

然后针对具体测试模型继承这个基类：

class TestUUIDModelTestCase(BaseUUIDMixinTestCase):
    test_model = TestUUIDModel

这样可以进一步简化测试代码，让测试结构更清晰。

内容的提问来源于stack exchange，提问作者J. Hesters