pytest-django测试中Model子类的__init_subclass__属性检查失效问题

嗨，我来帮你解决这个头疼的问题！你遇到的情况其实是pytest-django在测试时会动态生成模型的fake子类，这些子类没有继承原类的SOME_ATTR属性，而父类A的__init_subclass__检查会在动态类创建时触发，导致测试出现意外的报错。下面给你几个实用的解决方案：

方案一：给__init_subclass__打补丁，适配fake类 #

既然你试过session级的autouse fixture没生效（因为类创建时机比fixture更早），我们可以直接给A类的__init_subclass__方法打补丁，让它在识别到fake类时自动从原模型同步SOME_ATTR属性。

具体代码如下：

import pytest
from django.apps import apps
from your_app.models import A  # 替换成你的实际导入路径

@pytest.fixture(scope="session", autouse=True)
def patch_init_subclass_for_fake_models():
    # 先保存原方法，方便测试后恢复
    original_init_subclass = A.__init_subclass__

    @classmethod
    def patched_init_subclass(cls, **kwargs):
        # 判断当前类是否是pytest-django生成的fake类
        if cls.__module__ == "fake":
            try:
                # 从Django的应用注册表中找到原模型类
                original_model = apps.get_model(cls._meta.app_label, cls.__name__)
                # 给fake类补上缺失的SOME_ATTR
                setattr(cls, "SOME_ATTR", original_model.SOME_ATTR)
            except LookupError:
                # 找不到原模型类的话，就按原逻辑执行，避免测试崩溃
                pass
        # 执行原本的属性检查逻辑
        original_init_subclass(cls, **kwargs)

    # 替换原方法
    A.__init_subclass__ = patched_init_subclass

    yield

    # 测试结束后恢复原方法，避免影响其他逻辑
    A.__init_subclass__ = original_init_subclass

这个fixture会在所有测试启动前自动生效，给fake类补上缺失的属性，让你的检查逻辑正常通过。

方案二：改用元类实现属性检查 #

如果你想从根源上解决这个问题，可以把属性检查从__init_subclass__移到元类中。因为Django模型的创建是由元类控制的，元类的__new__方法会在类创建阶段执行，能覆盖到pytest-django动态生成的fake类。

代码示例：

from django.db import models

class RequiredAttrMeta(models.base.ModelBase):
    def __new__(cls, name, bases, attrs):
        # 检查当前类是否直接定义了SOME_ATTR（排除父类继承的情况）
        if "SOME_ATTR" not in attrs or not attrs["SOME_ATTR"]:
            raise TypeError(f"Class '{name}' must define 'SOME_ATTR'")
        return super().__new__(cls, name, bases, attrs)

class B(models.Model, metaclass=RequiredAttrMeta):
    SOME_ATTR = 0
    # 你的其他模型字段...

这种方式的优势是检查逻辑会在类创建的更早阶段执行，能确保所有继承该元类的子类（包括测试中的动态类）都满足属性要求。

方案三：测试环境临时跳过检查（应急用，不推荐） #

如果你只是想先让测试跑起来，之后再优化，可以在__init_subclass__中判断测试环境，临时跳过检查：

import os

class A:
    SOME_ATTR: int

    def __init_subclass__(cls, **kwargs):
        super().__init_subclass__(**kwargs)
        # 判断是否处于pytest测试环境
        is_pytest = os.environ.get("PYTEST_CURRENT_TEST") is not None
        if is_pytest:
            # 测试环境下跳过检查
            return
        # 生产环境执行原检查逻辑
        attr = getattr(cls, "SOME_ATTR", None)
        if not attr:
            raise TypeError(f"Class '{cls.__name__}' must define 'SOME_ATTR'")

这个方案不推荐长期使用，因为会导致测试环境和生产环境的逻辑不一致，可能漏掉潜在问题。

备注：内容来源于stack exchange，提问作者Damian Sowiński