Django测试时如何指定要使用的数据库？

嘿，这个需求我之前也碰到过，环境变量确实能解决，但还有几个更顺手的方案，给你参考：

方案1：自定义扩展test命令，直接支持命令行参数 #

这应该是最贴合你需求的方式——直接给Django原生的test命令加一个-db参数，不用额外改配置文件，一步到位。

步骤很简单：

1. 在你的Django项目里创建一个management/commands目录（如果没有的话），记得加__init__.py文件。
2. 在这个目录下新建test.py文件，内容如下：

from django.core.management.commands.test import Command as TestCommand
from django.conf import settings

class Command(TestCommand):
    def add_arguments(self, parser):
        # 先继承原test命令的所有参数
        super().add_arguments(parser)
        # 新增-db/--database参数
        parser.add_argument(
            '-db', '--database',
            type=str,
            help='Specify which configured database to use for running tests'
        )

    def handle(self, *args, **options):
        db_choice = options.get('database')
        if db_choice and db_choice in settings.DATABASES:
            # 把指定的数据库配置替换成default
            settings.DATABASES['default'] = settings.DATABASES[db_choice].copy()
        # 调用原test命令的逻辑
        super().handle(*args, **options)

之后你就可以直接用你想要的命令格式运行测试了：

python manage.py test -db=mysql

这个方案的好处是完全贴合你的使用习惯，不用额外维护多个配置文件，所有逻辑都集中在一个自定义命令里。

方案2：用--settings参数切换测试专用配置文件 #

如果你的测试环境针对不同数据库还有其他差异化配置（比如缓存、日志），这个方案会更清晰。

1. 在项目根目录创建测试专用的配置文件，比如settings_test_mysql.py：

# 导入基础配置
from .settings import *

# 替换默认数据库为mysql配置
DATABASES['default'] = DATABASES['mysql'].copy()
# 这里还可以加其他测试专属配置，比如DEBUG=False，调整日志级别等

2. 运行测试时指定这个配置文件：

python manage.py test --settings=myproject.settings_test_mysql

这个方案的优势是配置隔离清晰，不同数据库的测试配置可以分开维护，适合复杂的测试场景。

方案3：数据库路由配合环境变量（适合多数据库共存的测试场景） #

如果你不想替换default数据库，而是希望测试时所有读写操作都定向到指定数据库，可以用Django的数据库路由功能。

1. 创建一个db_router.py文件：

import os
from django.conf import settings

# 从环境变量获取要测试的数据库，默认用default
TEST_TARGET_DB = os.environ.get('TEST_TARGET_DB', 'default')

class TestDatabaseRouter:
    def db_for_read(self, model, **hints):
        return TEST_TARGET_DB

    def db_for_write(self, model, **hints):
        return TEST_TARGET_DB

    def allow_migrate(self, db, app_label, model_name=None, **hints):
        # 只在目标数据库执行迁移
        return db == TEST_TARGET_DB

2. 在settings.py里配置路由：

DATABASE_ROUTERS = ['myproject.db_router.TestDatabaseRouter']

3. 运行测试时通过环境变量指定数据库：

TEST_TARGET_DB=mysql python manage.py test

这个方案适合你需要保留多数据库结构，但测试时统一用某个库的场景，比单纯的环境变量切换更灵活。

内容的提问来源于stack exchange，提问作者amalsalim