时间戳

首页 > 解决方案 > django和pytest，多个数据库，只用一个数据库

django - django和pytest，多个数据库，只用一个数据库

问题描述

我的项目中有 2 个数据库：

DATABASES = {
    'default': env.db('DEFAULT_DATABASE_URL'),
    'second': env.db('SECOND_DATABASE_URL'),
}

当我只有一个数据库时，所有需要访问数据库的测试都顺利通过，但是当我连接第二个数据库时，所有测试都开始失败并出现错误 - django.db.utils.IntegrityError: duplicate key value violates unique constraint。

据我了解，测试为每个数据库运行两次并因此崩溃。有没有办法告诉 pytest 只使用一个数据库进行测试？

一项测试的示例：

@pytest.mark.django_db
def test_complaint_reasons(client: APIClient):
    reasons_count = 12
    Profile.create_batch(reasons_count)

    client.force_login(Factory())
    response = client.get(reverse('url'))
    assert response.status_code == status.HTTP_200_OK
    assert len(response.data) == reasons_count
    jsonschema.validate(response.data, ProfileComplaintReasonsSchema)

标签： djangodjango-rest-frameworkpytestpytest-django

解决方案

请改用此标记： @pytest.mark.django_db(databases=['default', 'second'])

但是要小心这个，因为它仍然是实验性的。

推荐阅读