Updated lambda test

gruntwork-io · May 23, 2024 · b18a248 · b18a248
1 parent b8a4f98
commit b18a248
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 3 deletions.
diff --git a/examples/terraform-aws-lambda-example/main.tf b/examples/terraform-aws-lambda-example/main.tf
@@ -28,7 +28,7 @@ resource "aws_lambda_function" "lambda" {
   source_code_hash = data.archive_file.zip.output_base64sha256
   function_name    = var.function_name
   role             = aws_iam_role.lambda.arn
-  handler          = "lambda"
+  handler          = "bootstrap"
   runtime          = "provided.al2023"
 }
 

diff --git a/examples/terraform-aws-lambda-example/src/bootstrap b/examples/terraform-aws-lambda-example/src/bootstrap
diff --git a/examples/terraform-aws-lambda-example/src/bootstrap.go b/examples/terraform-aws-lambda-example/src/bootstrap.go
@@ -13,7 +13,10 @@ type Event struct {
 }
 
 // Fails if ShouldFail is `true`, otherwise echos the input.
-func HandleRequest(ctx context.Context, evnt Event) (string, error) {
+func HandleRequest(ctx context.Context, evnt *Event) (string, error) {
+	if evnt == nil {
+		return "", fmt.Errorf("received nil event")
+	}
 	if evnt.ShouldFail {
 		return "", fmt.Errorf("Failed to handle %#v", evnt)
 	}

diff --git a/test/terraform_aws_lambda_example_test.go b/test/terraform_aws_lambda_example_test.go
@@ -63,7 +63,7 @@ func TestTerraformAwsLambdaExample(t *testing.T) {
 	assert.Contains(t, string(functionError.Payload), "Failed to handle")
 }
 
-// Annother example of how to test the Terraform module in
+// Another example of how to test the Terraform module in
 // examples/terraform-aws-lambda-example using Terratest, this time with
 // the aws.InvokeFunctionWithParams.
 func TestTerraformAwsLambdaWithParamsExample(t *testing.T) {