pytest中补丁函数返回值未生效，测试为何失败？

问题场景 #

你有一个位于service目录下的file1.py：

def is_development():
    return is_not_other_env() or is_not_other_other_env()

# 模块导入时就会执行is_development()并完成赋值
APP_ENVIRONMENT = 'development' if is_development() else 'production'

测试文件test_file1.py尝试mockis_development函数来验证APP_ENVIRONMENT：

from service import file1
from unittest.mock import patch

@patch('service.file1.is_development', return_value=False)
def test_app_environment_prod():
    # 直接调用mock函数返回False，但APP_ENVIRONMENT不符合预期
    assert file1.APP_ENVIRONMENT == 'production'  # 测试失败

即使尝试用importlib.reload(file1)也无法解决问题，补丁对模块变量不起作用。

失效核心原因 #

问题出在模块级代码的执行时机：

• 当你执行from service import file1时，Python会立即加载file1模块，执行所有顶级代码——包括调用is_development()并给APP_ENVIRONMENT赋值，这个过程发生在测试函数的补丁生效之前。
• @patch装饰器的生效时间是测试函数执行阶段，此时APP_ENVIRONMENT已经是一个固定值，后续的mock无法回溯修改这个已经初始化完成的变量。
• 若importlib.reload操作放在补丁生效前执行，重新加载模块时依然会用未被mock的is_development()计算变量值，自然达不到预期。

解决方法 #

方法1：将模块变量改为延迟计算的函数 #

修改file1.py，把APP_ENVIRONMENT改为函数，每次调用时才计算值，确保每次都能用到最新的is_development()结果：

def is_development():
    return is_not_other_env() or is_not_other_other_env()

def get_app_environment():
    return 'development' if is_development() else 'production'

对应的测试代码：

from service import file1
from unittest.mock import patch

@patch('service.file1.is_development', return_value=False)
def test_app_environment_prod():
    assert file1.get_app_environment() == 'production'  # 测试通过

方法2：在补丁生效后再导入/重新加载模块 #

如果不想修改原模块代码，可以调整测试逻辑，确保模块初始化发生在补丁生效之后：

from unittest.mock import patch
import sys

def test_app_environment_prod():
    # 清理已导入的模块缓存，避免复用之前的初始化结果
    if 'service.file1' in sys.modules:
        del sys.modules['service.file1']
    
    # 在补丁上下文内导入模块，确保初始化时用mock后的函数
    with patch('service.file1.is_development', return_value=False):
        from service import file1
        assert file1.APP_ENVIRONMENT == 'production'  # 测试通过

也可以用装饰器配合内部导入的方式：

from unittest.mock import patch
import sys

@patch('service.file1.is_development', return_value=False)
def test_app_environment_prod(_):
    # 清理缓存后重新导入
    if 'service.file1' in sys.modules:
        del sys.modules['service.file1']
    import service.file1 as file1
    assert file1.APP_ENVIRONMENT == 'production'

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