Django REST Framework单元测试误用生产数据库问题咨询

这问题我之前帮同事排查过类似的，核心原因几乎都是web应用的测试没有触发Django自带的临时测试数据库机制，下面分几个步骤排查和解决：

1. 检查测试类的继承关系 #

确保web应用的所有测试类都继承自 django.test.TestCase，而不是Python原生的 unittest.TestCase。

• django.test.TestCase 会自动初始化临时测试数据库，跑完测试后自动销毁；
• 原生的 unittest.TestCase 不会触发Django的测试环境，会直接使用当前配置的生产数据库。

举个正确的例子：

# web/tests.py 正确写法
from django.test import TestCase

class WebViewTests(TestCase):
    def test_home_page(self):
        # 测试逻辑
        response = self.client.get('/')
        self.assertEqual(response.status_code, 200)

2. 确认测试环境的数据库配置 #

检查你的项目配置，确保测试时自动切换到临时数据库：

• 可以在 settings.py 中添加测试环境的判断逻辑，强制使用内存型SQLite（适合快速测试）：

# settings.py
import sys

# 其他配置...

if 'test' in sys.argv:
    DATABASES['default'] = {
        'ENGINE': 'django.db.backends.sqlite3',
        'NAME': ':memory:'
    }

这样只要你运行 python manage.py test 命令，就会自动使用内存中的临时数据库，完全隔离生产数据。

• 如果你有单独的测试配置文件（比如 settings_test.py），要确保运行测试时指定该配置：

python manage.py test web --settings=myproject.settings_test

3. 排查测试代码中的硬编码连接 #

检查web应用的测试文件，有没有手动指定生产数据库的连接，或者错误导入了生产环境的配置。比如有没有类似这样的代码：

# 错误示例：手动连接生产库
import psycopg2
conn = psycopg2.connect(host='生产地址', dbname='生产库名')

这种硬编码会绕过Django的测试数据库机制，直接操作生产数据，一定要删掉这类代码，改用Django的ORM或测试客户端。

4. 确保用Django的测试命令运行测试 #

不要直接用 unittest 或其他第三方测试 runner 单独运行web的测试文件，比如：

# 错误方式：不会触发Django测试环境
python -m unittest web.tests

正确的做法是用Django自带的测试命令：

# 正确方式：初始化测试环境，使用临时数据库
python manage.py test web

5. 检查@override_settings装饰器的误用 #

如果你的测试中使用了 @override_settings 装饰器，要确认没有错误地覆盖了数据库配置：

# 错误示例：不小心设置成生产库
from django.test import override_settings

@override_settings(DATABASES={'default': 生产库配置})
class BadTest(TestCase):
    # ...

这种情况会强制测试使用生产数据库，要删掉或修正装饰器里的配置。

验证方法 #

运行web应用的测试前，先删掉项目根目录下的 db.sqlite3（如果用SQLite），然后执行测试命令。如果测试后没有重新生成这个文件，说明临时数据库已经生效了；如果重新生成了，那就要再检查上面的配置步骤。

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