如何使用两个数据库(mysql和mongo)为django项目编写单元测试

DATABASES = {
    'default': {
        'ENGINE': 'django.db.backends.mysql',
        'NAME': 'mysql_database',
        'USER': 'mysql_user',
        'PASSWORD': 'mysql_password',
        'HOST': 'mysql_host',
        'PORT': 'mysql_port',
    },
    'mongo': {
        'ENGINE': 'djongo',
        'NAME': 'mongo_database',
        'CLIENT': {
            'host': 'mongo_host',
            'port': 'mongo_port',
            'username': 'mongo_user',
            'password': 'mongo_password',
            'authSource': 'admin',
        },
    },
}

from django.test import TestCase
from django.db import connections

class MyTestCase(TestCase):
    def test_mysql(self):
        # 使用MySQL数据库进行测试
        with connections['default'].cursor() as cursor:
            cursor.execute("SELECT * FROM my_table")
            result = cursor.fetchall()
            self.assertEqual(len(result), 10)  # 断言结果是否符合预期

    def test_mongo(self):
        # 使用MongoDB数据库进行测试
        from pymongo import MongoClient
        client = MongoClient('mongodb://mongo_user:mongo_password@mongo_host:mongo_port/admin')
        db = client['mongo_database']
        collection = db['my_collection']
        result = collection.find()
        self.assertEqual(result.count(), 10)  # 断言结果是否符合预期

python manage.py test