如何Mock AWS Lambda模块初始化阶段执行的外部初始化逻辑

在Lambda场景中，模块级的单次初始化是常见写法，但这会导致测试导入待处理函数时，初始化逻辑自动执行。要Mock这类导入阶段运行的函数，核心思路是在模块导入前完成Mock操作，或者重新加载模块时应用Mock，以下是两种可行方案：

方案一：用pytest monkeypatch + 模块缓存清理 #

直接在测试用例中先清理模块缓存、Mock目标函数，再执行导入：

import pytest
import sys

def test_lambda_handler(monkeypatch):
    # 定义Mock的初始化函数，返回自定义值
    def mock_foo_init():
        return 2.0

    # 移除已缓存的xyz模块（避免之前的导入影响当前测试）
    if 'xyz' in sys.modules:
        del sys.modules['xyz']
    
    # 提前Mock xyz模块中的foo_init
    monkeypatch.setattr('xyz.foo_init', mock_foo_init)
    
    # 此时导入lambda_handler，模块级代码会使用Mock后的foo_init初始化x
    from xyz import lambda_handler
    
    # 执行测试断言
    assert lambda_handler() == 预期结果

方案二：用unittest.mock.patch + 模块重载 #

如果已经提前导入了模块，可以通过重载模块的方式，在Mock生效时重新执行模块级初始化：

from unittest.mock import patch
import importlib
import xyz

def test_lambda_handler():
    # Mock foo_init并指定返回值
    with patch('xyz.foo_init', return_value=2.0):
        # 重载xyz模块，触发模块级代码重新执行
        importlib.reload(xyz)
        # 使用重载后模块中的lambda_handler
        result = xyz.lambda_handler()
        assert result == 预期结果

关键注意事项 #

• 每次测试前确保模块缓存被清理或重载，避免不同测试用例之间的状态污染。
• Mock的目标路径要准确，必须是模块实际引用的foo_init路径（这里就是xyz.foo_init）。

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