如何全局检测代码是否在测试环境运行？测试时怎样设置debug=True？

一、让测试时DEBUG == True #

有两种简单可行的实现方式：

1. 使用override_settings装饰器 #

直接给测试类或单个测试方法加上装饰器，强制将DEBUG设置为True：

from django.test import TestCase, override_settings

@override_settings(DEBUG=True)
class MyTestCase(TestCase):
    def test_debug_behavior(self):
        from django.conf import settings
        self.assertEqual(settings.DEBUG, True)
        # 这里编写需要在DEBUG=True下运行的测试逻辑

2. 在测试类的初始化/清理方法中临时修改 #

如果需要更灵活的开关控制，可以在测试启动时修改配置，结束时恢复：

from django.test import TestCase
from django.conf import settings

class MyTestCase(TestCase):
    def setUp(self):
        super().setUp()
        # 保存原始DEBUG值
        self.original_debug = settings.DEBUG
        settings.DEBUG = True

    def tearDown(self):
        # 测试结束后恢复原始配置
        settings.DEBUG = self.original_debug
        super().tearDown()

    def test_debug_behavior(self):
        self.assertEqual(settings.DEBUG, True)

二、检测代码是否处于测试运行状态 #

推荐几种可靠的实现方法：

1. 检查命令行参数 #

直接判断启动命令是否包含测试相关指令：

import sys

def is_running_tests():
    return 'test' in sys.argv or 'pytest' in sys.argv

2. 利用Django的测试配置标记 #

Django运行测试时会加载特定的测试runner，通过配置判断：

from django.conf import settings

def is_running_tests():
    return hasattr(settings, 'TEST_RUNNER') and 'test' in settings.TEST_RUNNER.lower()

3. 用信号设置全局标记 #

在测试启动时通过Django信号设置全局变量，方便任意位置调用：

import django.test.signals
import sys

def mark_test_running(**kwargs):
    sys._is_running_tests = True

# 绑定测试启动的信号
django.test.signals.setting_up.connect(mark_test_running)

def is_running_tests():
    return getattr(sys, '_is_running_tests', False)

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