如何在Django测试中验证Settings里的环境变量赋值逻辑？

完全可以通过Django Test来测试这段逻辑，核心是在测试过程中临时修改环境变量，验证MY_SETTING的取值是否符合预期。

具体实现步骤 #

• 编写继承自django.test.TestCase的测试用例
• 在测试前后保存并恢复原始环境变量，避免影响其他测试
• 修改环境变量后重新加载settings模块（规避模块缓存），再校验取值

测试代码示例 #

import os
from importlib import reload
from django.test import TestCase
from django.conf import settings

class TestMySetting(TestCase):
    def setUp(self):
        # 保存原始环境变量状态
        self.original_my_setting = os.environ.get("MY_SETTING")

    def tearDown(self):
        # 恢复环境变量至初始状态
        if self.original_my_setting:
            os.environ["MY_SETTING"] = self.original_my_setting
        else:
            os.environ.pop("MY_SETTING", None)

    def test_setting_with_env_value(self):
        # 模拟存在环境变量的场景
        os.environ["MY_SETTING"] = "MY_CUSTOM_VALUE"
        reload(settings)
        self.assertEqual(settings.MY_SETTING, "MY_CUSTOM_VALUE")

    def test_setting_with_default_value(self):
        # 模拟不存在环境变量的场景
        os.environ.pop("MY_SETTING", None)
        reload(settings)
        self.assertEqual(settings.MY_SETTING, "NOT MINE")

关键注意事项 #

• 必须恢复环境变量：如果不还原，会导致后续测试用例的环境变量被污染
• 强制重新加载settings：Django启动时会缓存settings模块，修改环境变量后必须用reload()才能获取最新配置

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