用 Moto 模拟 AWS lambda

Mocking AWS lambda with Moto

我正在尝试模拟 AWS lambda 函数,下面是我的示例代码

def get_lambda_resp(arn: str, input: str) -> str:
    lambda_client = boto3.client("lambda")
    response = lambda_client.invoke(
        FunctionName=arn, LogType="None",
        Payload=json.dumps({"param": input}).encode("utf-8")
    )
    output = json.loads(response["Payload"].read().decode("utf-8"))
    return output["value"]

下面是我的测试用例

import io
import zipfile

import boto3
from moto import mock_lambda

@mock_lambda
def test():
    conn = boto3.client('lambda', 'us-east-1')

    def get_test_zip_file():
        pfunc = '''
        import json
        def lambda_handler(event, context):
            resp = {"value":"input_str"}
            return json.dumps(resp)
        '''
        zip_output = io.BytesIO()
        zip_file = zipfile.ZipFile(zip_output, 'w', zipfile.ZIP_DEFLATED)
        zip_file.writestr('lambda_function.py', pfunc)
        zip_file.close()
        zip_output.seek(0)
        return zip_output.read()

    conn.create_function(
        FunctionName='lambda-function-name',
        Runtime='python3.8',
        Role='test-iam-role',
        Handler='lambda_function.lambda_handler',
        Code={
            'ZipFile': get_test_zip_file(),
        },
        Description='test lambda function',
        Timeout=3,
        MemorySize=128,
        Publish=True
    )

    resp = get_auth("arn", "input_str")
    assert resp is not None

在 运行 测试用例中,出现以下错误

E   ModuleNotFoundError: No module named 'docker'

我已经有了我的 Docker 运行ning,我还应该 运行 做什么?

该消息指的是名为 docker 的 pip 模块。 假设您使用 Moto >=2.x,请确保正确安装它以获取所有必需的依赖项:

pip install moto[awslambda,s3,service1,etc]

或者如果您使用许多服务,安装所有依赖项而不必列出所有服务:

pip install moto[all]

这将安装所有必需的 Pip 模块,包括 Docker。

来源:https://github.com/spulec/moto/issues/3722