如何本地单元测试依赖Lambda Layer的AWS Lambda函数？

遇到ModuleNotFoundError: No module named 'lambda_layer'是因为本地环境没有Lambda Layer对应的模块，下面是两种实用的解决办法：

办法1：Mock整个Lambda Layer模块 #

既然本地没有这个模块，直接用unittest.mock模拟它的导入和方法，完全不需要依赖真实的Layer代码。

测试代码修改如下：

from unittest import TestCase
from unittest import mock
import sys

# 先把lambda_layer模块mock掉，让导入时不报错
with mock.patch.dict(sys.modules, {'lambda_layer': mock.Mock()}):
    from path.to.lambda import lambda_handler as under_test

class TestStuff(TestCase):
    def test_lambda_handler(self):
        # 精准mock lambda_layer里的function方法，指定返回值
        with mock.patch('path.to.lambda.lambda_layer.function') as mock_func:
            mock_func.return_value = "模拟返回结果"
            
            # 构造测试事件并调用函数
            test_event = {"param1": "test_val1", "param2": "test_val2"}
            under_test(test_event, None)
            
            # 验证function是否按预期被调用
            mock_func.assert_called_once_with("test_val1", "test_val2")

原理：用mock.patch.dict修改sys.modules，把不存在的lambda_layer替换成一个Mock对象，这样导入Lambda函数时不会触发模块找不到的错误。之后再mock具体的function方法，就能专注测试lambda_handler的业务逻辑。

办法2：本地挂载Lambda Layer代码（如果能获取到代码） #

如果你能拿到Lambda Layer的源代码，可以把它的目录加入Python的搜索路径，让本地环境能正常识别这个模块。

步骤：

1. 把Lambda Layer的代码放到本地目录，比如./local_lambda_layer/（确保目录下有__init__.py，或者function所在的模块文件在这个目录里）
2. 修改测试代码，添加路径：

import sys
from pathlib import Path
from unittest import TestCase
from unittest import mock

# 将本地Layer目录添加到Python搜索路径
sys.path.append(str(Path(__file__).parent / "local_lambda_layer"))

# 现在可以正常导入了
from path.to.lambda import lambda_handler as under_test

class TestStuff(TestCase):
    def test_lambda_handler(self):
        # 同样可以mock function，或者直接用真实逻辑测试
        with mock.patch('path.to.lambda.lambda_layer.function') as mock_func:
            mock_func.return_value = "真实逻辑返回值"
            under_test({"param1": "a", "param2": "b"}, None)
            mock_func.assert_called_with("a", "b")

这种方法适合需要测试部分真实Layer逻辑的场景，不需要完全mock。

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