如何在Python测试中Mock环境变量？常量引用断言返回None

问题根源 #

你业务代码里的PERSONAL_ID和PATH_NAME属于模块加载阶段就执行os.getenv赋值的常量，而@mock.patch.dict(os.environ)是在测试方法运行时才修改环境变量。等测试代码执行patch时，模块早就加载完毕，这两个常量已经被初始化为None，所以后续的环境变量修改不会同步到常量上。

另外注意你原装饰器里的key多了个空格："PERSONAL_ID "，应该改成"PERSONAL_ID"，不过这不是导致测试失败的核心原因。

解决办法 #

1. 延迟加载环境变量（推荐，优化业务代码）

把环境变量的获取改成延迟执行的方式，每次使用时才读取环境变量，这样patch的环境变量就能生效：

# 业务代码修改
def get_personal_id():
    return os.getenv("PERSONAL_ID")

def get_path_name():
    return os.getenv("PATH_NAME")

# 业务方法里调用函数获取值
def your_business_method():
    pid = get_personal_id()
    path = get_path_name()
    # 后续逻辑

如果是类中的逻辑，可以用@property实现延迟加载：

class YourBusinessClass:
    @property
    def personal_id(self):
        return os.getenv("PERSONAL_ID")
    
    @property
    def path_name(self):
        return os.getenv("PATH_NAME")
    
    def do_business(self):
        # 使用self.personal_id和self.path_name

2. 测试时重新加载模块

在测试方法中先patch环境变量，再重新加载业务模块，让常量重新读取环境变量：

import importlib
import your_business_module
from unittest import mock
import os

@mock.patch.dict(os.environ, {"PERSONAL_ID": "123", "PATH_NAME": "test_path"})
def test_business_method():
    # 重新加载模块，触发常量重新赋值
    importlib.reload(your_business_module)
    # 调用业务方法并断言
    result = your_business_module.your_business_method()
    assert result is not None

⚠️ 注意：重新加载模块可能影响其他测试用例，建议在测试前后做模块清理，或者单独隔离这类测试。

3. 直接patch模块中的常量（无需修改业务代码）

跳过环境变量，直接mock业务模块里的常量本身，这是最直接的测试方式：

from unittest import mock
import your_business_module

@mock.patch.object(your_business_module, "PERSONAL_ID", "123")
@mock.patch.object(your_business_module, "PATH_NAME", "test_path")
def test_business_method():
    result = your_business_module.your_business_method()
    assert result is not None

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