在 pytest 中导入 Django 模型时如何解决“Apps aren't loaded yet”错误？

python

2024-03-13 08:48:28

在 pytest 中导入 Django 模型时解决“Apps aren't loaded yet”错误

背景

在使用 pytest 测试 Django 模型时，我们经常会遇到“Apps aren't loaded yet”的错误。这是因为 Django 应用还没有被加载，导致无法导入模型。

解决方案

为了解决此错误，我们可以采取以下步骤：

1. 检查应用安装

确保所需的 Django 应用已正确安装到你的 Python 环境中。使用以下命令检查：

pip freeze | grep <app-name>

2. 添加到 INSTALLED_APPS

在 settings.py 文件中，将所需的应用添加到 INSTALLED_APPS 设置中，以便 Django 在启动时加载该应用。

3. 使用 pytest-django 插件

pytest-django 插件自动加载 Django 应用，简化了 Django 测试。安装并启用该插件：

pip install pytest-django

import pytest
pytest_plugins = "pytest_django"

4. 使用 monkeypatch

如果无法加载 Django 应用，可以使用 monkeypatch 模拟应用的存在，从而测试模型：

@pytest.fixture(autouse=True)
def monkeypatch_app(monkeypatch):
    monkeypatch.setattr(apps, 'get_app_config', lambda x: x)

代码示例

以下代码示例演示了如何从 usecase1.models 模块导入 MyModel 模型：

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

class MyModelTests(TestCase):

    def test_model_creation(self):
        model = MyModel(name='Test Model')
        model.save()
        self.assertEqual(MyModel.objects.count(), 1)