django - pytest 和 Django 事务数据库

Question

我使用生产数据库进行测试(实际上它是 docker 中的测试数据库)。问题是:如何让测试在针对该数据库的事务中运行。我需要与 @pytest.mark.django_db(transaction=True) 相同的行为，但需要使用生产数据库。

当前设置:

conftest.py

@pytest.fixture(scope='session')
def django_db_setup():
    """Avoid creating/setting up the test database"""
    pass


@pytest.fixture
def db(request, django_db_setup, django_db_blocker):
    django_db_blocker.unblock()


@pytest.fixture
def myfixture(db):
    ...
    return SomeObject

测试示例.py

def test_something(db, myfixture):
    assert ...

Answer 1

我终于找到了解决方案。

添加 fixture 加载代码到db fixture :

conftest.py

from django.core.management import call_command

@pytest.fixture
def db(request, django_db_setup, django_db_blocker):
    django_db_blocker.unblock()
    call_command('loaddata', 'fixture.json')

并使用 @pytest.mark.django_db(transaction=True) 进行测试:

测试示例.py

@pytest.mark.django_db(transaction=True)
def test_something(db, myfixture):
    assert ...

每次测试后，pytest 将刷新您的数据库并用固定数据填充它。