如何让子进程识别被Mock补丁修改的全局变量？

核心原因 #

unittest.mock.patch的作用范围仅限当前进程的内存空间——它只是替换了当前进程中模块对象的引用。而子进程会启动全新的Python解释器实例，重新加载目标模块，自然会读取SOME_GLOBAL_VARIABLE的原始值，和父进程的Mock操作完全无关。

可行解决方案 #

方案一：通过环境变量传递Mock值

子进程会继承父进程的环境变量，我们可以利用这一点，在目标模块中优先读取环境变量的值，没有则使用原始值。

1. 修改目标模块my_module/file_in_module.py：

import os

# 优先读取环境变量，不存在则使用原始默认值
SOME_GLOBAL_VARIABLE = os.getenv("MY_MOCK_VAR", "原始默认值")

2. 单元测试代码调整：

import os
import subprocess

# 设置环境变量
os.environ["MY_MOCK_VAR"] = mocked_var
try:
    subprocess.check_call([
        # 你的CLI命令，例如："python", "-m", "my_module.target_script"
    ])
finally:
    # 清理环境变量，避免影响其他测试用例
    del os.environ["MY_MOCK_VAR"]

方案二：通过命令行参数传递Mock值

如果子进程执行的脚本支持接收命令行参数，可以直接将Mock值作为参数传入，在脚本内覆盖全局变量。

1. 修改子进程执行的目标脚本（如my_module/target_script.py）：

import sys
from my_module.file_in_module import SOME_GLOBAL_VARIABLE

if __name__ == "__main__":
    # 若传入参数，则替换全局变量
    if len(sys.argv) > 1:
        SOME_GLOBAL_VARIABLE = sys.argv[1]
    # 执行原有业务逻辑

2. 单元测试代码调整：

import subprocess

subprocess.check_call([
    "python", "-m", "my_module.target_script", mocked_var
])

方案三：临时修改模块文件（慎用）

直接修改目标模块的源码文件，替换全局变量值，执行完子进程后再恢复。这种方式存在风险（如并发测试冲突、文件修改失败），仅作为备选方案。

import os
import fileinput
import subprocess

module_path = os.path.join(os.path.dirname(__file__), "my_module", "file_in_module.py")
original_content = None

# 替换模块中的全局变量行
with fileinput.FileInput(module_path, inplace=True, backup='.bak') as f:
    for line in f:
        if line.startswith("SOME_GLOBAL_VARIABLE ="):
            original_content = line
            print(f"SOME_GLOBAL_VARIABLE = {repr(mocked_var)}")
        else:
            print(line, end='')

try:
    subprocess.check_call([
        # 你的CLI命令
    ])
finally:
    # 恢复原始文件内容
    if original_content:
        with fileinput.FileInput(module_path, inplace=True) as f:
            for line in f:
                if line.startswith("SOME_GLOBAL_VARIABLE ="):
                    print(original_content, end='')
                else:
                    print(line, end='')
    # 也可以直接恢复备份文件：os.replace(f"{module_path}.bak", module_path)

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