要测试AWS SAM(Serverless Application Model)中的Lambda函数是否被正确调用以及Kinesis事件是否被正确创建,您可以使用以下代码示例来解决问题:
import boto3
def test_lambda_handler():
# 创建Lambda客户端
lambda_client = boto3.client('lambda', region_name='us-west-2')
# 调用Lambda函数
response = lambda_client.invoke(
FunctionName='YourLambdaFunctionName',
InvocationType='Event', # 使用Event类型,异步调用Lambda函数并立即返回
Payload='{}' # 可选,传递给Lambda函数的事件数据
)
# 检查Lambda函数调用的响应
assert response['StatusCode'] == 202 # 202表示请求已接受
# 执行Lambda函数测试
test_lambda_handler()
import boto3
def test_kinesis_event():
# 创建Kinesis客户端
kinesis_client = boto3.client('kinesis', region_name='us-west-2')
# 发布Kinesis事件
response = kinesis_client.put_record(
StreamName='YourKinesisStreamName',
Data='{"key1": "value1", "key2": "value2"}', # 事件数据
PartitionKey='partitionkey' # Kinesis分区键
)
# 检查Kinesis事件发布的响应
assert response['ResponseMetadata']['HTTPStatusCode'] == 200 # 200表示请求成功
# 执行Kinesis事件测试
test_kinesis_event()
请注意,您需要根据您的具体情况替换代码中的以下值:
'YourLambdaFunctionName'
:替换为要测试的Lambda函数的名称。'YourKinesisStreamName'
:替换为要测试的Kinesis数据流的名称。'us-west-2'
:替换为您的AWS区域代码。另外,确保您的测试环境中已配置有适当的AWS凭证,以便代码能够访问相关的AWS资源。
上一篇:AWS SAM - API Gateway执行失败,原因是Lambda函数的权限配置错误:无效的权限。
下一篇:AWS SAM - 获取错误消息: pywintypes.error: (5, 'CreateFile', '访问被拒绝。')