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

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',
    }
}

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

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'