不同Model实例场景下复用相同接口测试的最优实现方式咨询

你提到的基类继承+子类实现不同前置数据的方案完全可行，而且是这类场景下的最优解——既能最大化复用测试逻辑，又能清晰区分不同测试场景，符合DRY（Don't Repeat Yourself）原则。

具体实现示例（以Django TestCase为例） #

首先封装公共测试逻辑到基类，只留空前置数据准备方法由子类实现：

from django.test import TestCase
import requests

class BaseDeleteEndpointTest(TestCase):
    @classmethod
    def setUpTestData(cls):
        # 基类不做具体数据创建，交给子类实现
        pass

    def test_delete_endpoint_returns_204(self):
        # 所有场景共享的断言逻辑
        response = requests.delete("endpoint?pk=a")
        self.assertEqual(response.status_code, 204)

然后创建三个子类，分别实现不同的前置数据：

# 场景1：仅创建Model1对象
class Model1OnlyDeleteTest(BaseDeleteEndpointTest):
    @classmethod
    def setUpTestData(cls):
        super().setUpTestData()
        Model1.objects.create(name="a")

# 场景2：仅创建Model2对象
class Model2OnlyDeleteTest(BaseDeleteEndpointTest):
    @classmethod
    def setUpTestData(cls):
        super().setUpTestData()
        Model2.objects.create(name="a")

# 场景3：同时创建Model1和Model2对象
class BothModelsDeleteTest(BaseDeleteEndpointTest):
    @classmethod
    def setUpTestData(cls):
        super().setUpTestData()
        Model1.objects.create(name="a")
        Model2.objects.create(name="a")

方案优势 #

1. 逻辑复用：核心断言代码只写一次，避免重复维护
2. 场景清晰：每个子类对应一个测试场景，代码可读性强
3. 扩展灵活：后续如果某个场景需要额外测试步骤，直接在子类中新增方法即可，不影响其他场景

替代方案：参数化测试 #

如果使用pytest测试框架，也可以用参数化测试实现，代码更紧凑：

import pytest
from django.test import TestCase
import requests

@pytest.mark.parametrize("setup_func", [
    lambda: Model1.objects.create(name="a"),
    lambda: Model2.objects.create(name="a"),
    lambda: (Model1.objects.create(name="a"), Model2.objects.create(name="a")),
])
def test_delete_endpoint_returns_204(setup_func):
    setup_func()
    response = requests.delete("endpoint?pk=a")
    assert response.status_code == 204

但对比下来，基类继承的方式更贴合Django原生测试框架的使用习惯，且当测试逻辑复杂时，代码结构更清晰。

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