如何Mock含相等运算符的环境变量？优化常量定义以支持单元测试

你的判断完全正确：USE_THING作为模块级变量，在thing.py被导入时就已经基于当时的os.environ计算出了值，后续测试中patch环境变量不会影响这个已经初始化好的变量。下面提供几种既能复用配置、又能支持测试patch的方案：

方案1：直接Patch模块中的USE_THING变量 #

既然环境变量的最终结果已经存在模块的USE_THING里，测试时直接patch这个变量比patch环境变量更直接，不需要修改业务代码：

修改测试代码：

# test.py
from unittest import TestCase, mock
import thing

class TestThing(TestCase):
    @mock.patch("thing.USE_THING", True)
    def test_do_thing_when_enabled(self):
        # 此时do_thing会执行内部逻辑
        thing.do_thing()
        # 验证doing_the_thing是否被调用（如果需要的话）
        ...

    @mock.patch("thing.USE_THING", False)
    def test_do_thing_when_disabled(self):
        # 此时do_thing会直接返回，不执行内部逻辑
        thing.do_thing()
        ...

这种方案不需要改动业务代码，测试成本最低，适合快速解决问题。

方案2：用函数封装配置获取逻辑（支持缓存） #

如果希望配置能在运行时响应环境变量变化（不止测试场景），可以把USE_THING的计算逻辑封装成带缓存的函数，这样既复用逻辑，又能在测试时清空缓存或patch函数：

修改thing.py：

import os
from functools import lru_cache

@lru_cache(maxsize=None)
def get_use_thing():
    return os.environ.get("USE_THING", "False").upper() == "TRUE"

def do_thing():
    if not get_use_thing():
        return
    try:
        doing_the_thing()
    except Exception as e:
        logger.error(e)

def do_another_thing():
    if not get_use_thing():
        return
    try:
        doing_the_other_thing()
    except Exception as e:
        logger.error(e)

测试时可以选择patch环境变量后清空缓存，或者直接patchget_use_thing函数：

# test.py
from unittest import TestCase, mock
import thing

class TestThing(TestCase):
    @mock.patch.dict(os.environ, {"USE_THING": "True"}, clear=True)
    def test_do_thing_with_env_patch(self):
        # 清空缓存，让函数重新读取环境变量
        thing.get_use_thing.cache_clear()
        thing.do_thing()
        ...

    @mock.patch("thing.get_use_thing", return_value=False)
    def test_do_thing_disabled(self, mock_get):
        thing.do_thing()
        ...

方案3：使用配置类管理（适合复杂配置场景） #

如果项目中有多个类似的配置项，可以用类来统一管理，通过延迟加载的方式确保每次获取时都能读取最新的环境变量（或允许测试patch）：

修改thing.py：

import os

class Config:
    @staticmethod
    def get_use_thing():
        return os.environ.get("USE_THING", "False").upper() == "TRUE"

def do_thing():
    if not Config.get_use_thing():
        return
    try:
        doing_the_thing()
    except Exception as e:
        logger.error(e)

def do_another_thing():
    if not Config.get_use_thing():
        return
    try:
        doing_the_other_thing()
    except Exception as e:
        logger.error(e)

测试时可以直接patchConfig.get_use_thing方法：

# test.py
from unittest import TestCase, mock
import thing

class TestThing(TestCase):
    @mock.patch("thing.Config.get_use_thing", return_value=True)
    def test_do_thing_enabled(self):
        thing.do_thing()
        ...

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