如何在django pytest中对单元测试用例使用多个数据库

在Django pytest中对单元测试用例使用多个数据库可以通过以下步骤实现：

配置多个数据库连接：在Django项目的settings.py文件中，可以通过DATABASES配置项来指定多个数据库连接。每个数据库连接都需要指定名称、引擎、名称（或连接字符串）、用户名和密码等信息。例如：

DATABASES = {
    'default': {
        'ENGINE': 'django.db.backends.mysql',
        'NAME': 'default_db',
        'USER': 'user',
        'PASSWORD': 'password',
        'HOST': 'localhost',
        'PORT': '3306',
    },
    'secondary': {
        'ENGINE': 'django.db.backends.mysql',
        'NAME': 'secondary_db',
        'USER': 'user',
        'PASSWORD': 'password',
        'HOST': 'localhost',
        'PORT': '3306',
    }
}

上述配置定义了两个数据库连接，名称分别为'default'和'secondary'。

创建测试用例：在编写测试用例时，可以使用pytest的fixture机制来创建测试所需的数据库连接。可以在conftest.py文件中定义fixture函数，例如：

import pytest
from django.conf import settings
from django.db import connections
from django.test import TestCase

@pytest.fixture(scope='module')
def default_db():
    settings.DATABASES['default'] = {
        'ENGINE': 'django.db.backends.mysql',
        'NAME': 'default_test_db',
        'USER': 'user',
        'PASSWORD': 'password',
        'HOST': 'localhost',
        'PORT': '3306',
    }
    connections.ensure_defaults()
    yield
    # Tear down code if needed

@pytest.fixture(scope='module')
def secondary_db():
    settings.DATABASES['secondary'] = {
        'ENGINE': 'django.db.backends.mysql',
        'NAME': 'secondary_test_db',
        'USER': 'user',
        'PASSWORD': 'password',
        'HOST': 'localhost',
        'PORT': '3306',
    }
    connections.ensure_defaults()
    yield
    # Tear down code if needed

上述代码定义了两个fixture函数，分别为'default_db'和'secondary_db'。这些fixture函数会在每个测试模块执行前创建对应的数据库连接，并在测试完成后清理相关资源。

编写测试用例：使用上述fixture函数创建的数据库连接，可以在测试用例中使用多个数据库。例如：

import pytest
from django.contrib.auth.models import User
from myapp.models import MyModel

@pytest.mark.django_db
def test_default_db(default_db):
    # 使用'default'数据库连接执行测试逻辑
    user = User.objects.create(username='test_user')
    assert user.username == 'test_user'

@pytest.mark.django_db
def test_secondary_db(secondary_db):
    # 使用'secondary'数据库连接执行测试逻辑
    obj = MyModel.objects.create(name='test_object')
    assert obj.name == 'test_object'

上述代码中，通过在测试用例函数上使用@pytest.mark.django_db装饰器，可以自动创建并使用对应的数据库连接。在每个测试用例中，可以根据需要选择使用'default'或'secondary'数据库连接。

通过上述步骤，我们可以在Django pytest中对单元测试用例使用多个数据库，实现更灵活的测试需求。