pytest Django从yaml文件加载数据库数据

Question

我已经使用dumpdata django命令将开发环境中的数据转储到yaml文件中

我正在使用pytest框架，需要实现以下场景：

• 在pytest会话开始期间->；也就是说，当我从命令行运行pytest时，yaml文件中的数据应该加载到我的测试数据库中

是否有任何插件或挂钩，我可以用来实现上述

我使用的是python 3.8、pytest 6.2.4、pytest django 4.40和django 3.0.5

Answer 1

如果您正在使用pytest-django包，那么您可以通过在conftest.py中提供一个^{} fixturepytestfixture来为您的测试加载Django数据装置

import pytest

from django.core.management import call_command

@pytest.fixture(scope='session')
def django_db_setup(django_db_setup, django_db_blocker):
    with django_db_blocker.unblock():
        call_command('loaddata', 'my_fixture.yaml')