如何为导入时加载的环境变量编写Python测试并解决断言失败

你的测试失败的核心原因是：Python模块的顶层代码（比如PREFIX的赋值、router的初始化）只会在首次导入时执行一次，之后再导入都会直接复用缓存里的模块对象。哪怕你在测试里修改了环境变量，只要模块已经被导入过，新的环境变量值也不会生效。

不用改业务代码（比如加get_prefix函数）的前提下，你可以通过清理模块缓存+重新导入的方式解决，具体修改测试代码如下：

# 使用pytest-mock的mocker fixture
import os
import sys  # 新增：用来操作模块缓存

def test_prefix_value_digested_correctly(mocker):
    # 1. 先patch环境变量
    mocker.patch.dict(
        os.environ,
        {"SERVICE_PREFIX": "/services"},
        clear=True,
    )
    
    # 2. 清理已缓存的the_file模块（如果存在）
    if "the_file" in sys.modules:
        del sys.modules["the_file"]
    
    # 3. 重新导入模块，此时会执行顶层代码读取新的环境变量
    from the_file import router
    assert router.prefix == "/services"

补充说明 #

• sys.modules是Python存储已导入模块的字典，删除对应键后，再次导入模块会重新执行整个模块的顶层代码，自然会读取最新的环境变量值。
• 如果你的项目中有其他模块依赖the_file，可能需要一并清理这些依赖模块的缓存，但在这个简单场景下，只清理the_file就足够。
• 这种方式适合做合理性验证，完全符合你不想修改业务代码的需求。

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