用Moto嘲笑AWS lambda

Question

我试图模拟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'

我的码头司机已经在跑了，我还能做些什么呢？

Answer 1

该消息指的是称为码头的pip模块。假设您使用Moto >=2.x，请确保正确安装它以获得所有必需的依赖关系：

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

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

pip install moto[all]

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

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