Adding lambda test

Author: yhakbar

README.md

@@ -7,7 +7,7 @@
 Use this URL for the source of the module. See the usage examples below for more details.
 
 ```hcl
-github.com/pbs/terraform-aws-elasticache-redis-standalone-module?ref=0.0.1
+github.com/pbs/terraform-aws-elasticache-redis-standalone-module?ref=x.y.z
 ```
 
 ### Alternative Installation Methods
@@ -26,7 +26,7 @@ Integrate this module like so:
 
 ```hcl
 module "elasticache-redis-standalone" {
-  source = "github.com/pbs/terraform-aws-elasticache-redis-standalone-module?ref=0.0.1"
+  source = "github.com/pbs/terraform-aws-elasticache-redis-standalone-module?ref=x.y.z"
 
   # Tagging Parameters
   organization = var.organization
@@ -42,7 +42,7 @@ module "elasticache-redis-standalone" {
 
 If this repo is added as a subtree, then the version of the module should be close to the version shown here:
 
-`0.0.1`
+`x.y.z`
 
 Note, however that subtrees can be altered as desired within repositories.
 

examples/lambda/outputs.tf

@@ -37,3 +37,8 @@ output "tags" {
   description = "The tags"
   value       = module.redis.tags
 }
+
+output "lambda_name" {
+  description = "The name of the lambda function"
+  value       = module.lambda.name
+}

scripts/test.sh

@@ -9,4 +9,4 @@ GIT_ROOT=$(git rev-parse --show-toplevel)
 "$GIT_ROOT"/scripts/package.sh
 
 pushd "$GIT_ROOT"/tests >/dev/null
-go test -timeout 30m -count=1 -parallel 10 ./...
+go test -timeout 45m -count=1 -parallel 10 ./...

tests/basic_test.go

@@ -5,5 +5,5 @@ import (
 )
 
 func TestBasicExample(t *testing.T) {
-	testTemplate(t, "basic")
+	testRedis(t, "basic")
 }

tests/lambda_test.go

@@ -0,0 +1,9 @@
+package test
+
+import (
+	"testing"
+)
+
+func TestLambdaExample(t *testing.T) {
+	testRedis(t, "lambda")
+}

tests/utilities_redis.go

@@ -1,15 +1,18 @@
 package test
 
 import (
+	"encoding/json"
 	"fmt"
 	"strconv"
 	"testing"
 
+	"github.com/aws/aws-sdk-go/aws/session"
+	"github.com/aws/aws-sdk-go/service/lambda"
 	"github.com/gruntwork-io/terratest/modules/terraform"
 	"github.com/stretchr/testify/assert"
 )
 
-func testTemplate(t *testing.T, variant string) {
+func testRedis(t *testing.T, variant string) {
 	t.Parallel()
 
 	terraformDir := fmt.Sprintf("../examples/%s", variant)
@@ -50,4 +53,46 @@ func testTemplate(t *testing.T, variant string) {
 	assert.Equal(t, expectedMemberClusters, memberClusters)
 	assert.Contains(t, primaryEndpointAddress, expectedPartialPrimaryEndpointAddress)
 	assert.Contains(t, readerEndpointAddress, expectedPartialReaderEndpointAddress)
+
+	if variant == "lambda" {
+		expectedLambdaName := expectedName
+		lambdaName := terraform.Output(t, terraformOptions, "lambda_name")
+		assert.Equal(t, expectedLambdaName, lambdaName)
+
+		session, err := session.NewSession()
+		if err != nil {
+			t.Fatalf("Failed to create AWS session: %v", err)
+		}
+
+		lambdaSvc := lambda.New(session)
+
+		invokeOutput, err := lambdaSvc.Invoke(&lambda.InvokeInput{
+			FunctionName: &lambdaName,
+		})
+
+		if err != nil {
+			t.Fatal(err)
+		}
+
+		var lambdaResp struct {
+			Result string `json:"result"`
+			Error  string `json:"error"`
+		}
+
+		err = json.Unmarshal(invokeOutput.Payload, &lambdaResp)
+
+		if err != nil {
+			t.Fatal(err)
+		}
+
+		if *invokeOutput.StatusCode != 200 {
+			t.Logf("lambda response: %v", lambdaResp)
+			t.Fatalf("Expected status code 200, got %d", *invokeOutput.StatusCode)
+		}
+
+		if lambdaResp.Result == "failure" {
+			t.Logf("lambda response: %v", lambdaResp)
+			t.Fatalf("Expected result 'success', got 'failure'")
+		}
+	}
 }