Update alb test data

events/alb.go

@@ -4,10 +4,10 @@ package events
 type ALBTargetGroupRequest struct {
 	HTTPMethod                      string                       `json:"httpMethod"`
 	Path                            string                       `json:"path"`
-	QueryStringParameters           map[string]string            `json:"queryStringParameters"`
-	MultiValueQueryStringParameters map[string][]string          `json:"multiValueQueryStringParameters"`
-	Headers                         map[string]string            `json:"headers"`
-	MultiValueHeaders               map[string][]string          `json:"multiValueHeaders"`
+	QueryStringParameters           map[string]string            `json:"queryStringParameters,omitempty"`
+	MultiValueQueryStringParameters map[string][]string          `json:"multiValueQueryStringParameters,omitempty"`
+	Headers                         map[string]string            `json:"headers,omitempty"`
+	MultiValueHeaders               map[string][]string          `json:"multiValueHeaders,omitempty"`
 	RequestContext                  ALBTargetGroupRequestContext `json:"requestContext"`
 	IsBase64Encoded                 bool                         `json:"isBase64Encoded"`
 	Body                            string                       `json:"body"`

events/alb_test.go

@@ -10,25 +10,30 @@ import (
 )
 
 func TestALBTargetRequestMarshaling(t *testing.T) {
-	// read json from file
-	inputJSON, err := ioutil.ReadFile("./testdata/alb-lambda-target-request.json")
-	if err != nil {
-		t.Errorf("could not open test file. details: %v", err)
-	}
+	inputFiles := []string{
+		"./testdata/alb-lambda-target-request-headers-only.json",
+		"./testdata/alb-lambda-target-request-multivalue-headers.json"}
 
-	// de-serialize into Go object
-	var inputEvent ALBTargetGroupRequest
-	if err := json.Unmarshal(inputJSON, &inputEvent); err != nil {
-		t.Errorf("could not unmarshal event. details: %v", err)
-	}
+	for _, filename := range inputFiles {
+		// read json from file
+		inputJSON, err := ioutil.ReadFile(filename)
+		if err != nil {
+			t.Errorf("could not open test file. details: %v", err)
+		}
 
-	// serialize to json
-	outputJSON, err := json.Marshal(inputEvent)
-	if err != nil {
-		t.Errorf("could not marshal event. details: %v", err)
-	}
+		// de-serialize into Go object
+		var inputEvent ALBTargetGroupRequest
+		if err := json.Unmarshal(inputJSON, &inputEvent); err != nil {
+			t.Errorf("could not unmarshal event. details: %v", err)
+		}
 
-	assert.JSONEq(t, string(inputJSON), string(outputJSON))
+		// serialize to json
+		outputJSON, err := json.Marshal(inputEvent)
+		if err != nil {
+			t.Errorf("could not marshal event. details: %v", err)
+		}
+		assert.JSONEq(t, string(inputJSON), string(outputJSON))
+	}
 }
 
 func TestALBTargetRequestMalformedJson(t *testing.T) {

events/testdata/alb-lambda-target-request-headers-only.json

@@ -0,0 +1,26 @@
+{
+  "requestContext": {
+    "elb": {
+      "targetGroupArn": "arn:aws:elasticloadbalancing:us-east-1:123456789012:targetgroup/lambda-target/abcdefg"
+    }
+  },
+  "httpMethod": "GET",
+  "path": "/",
+  "queryStringParameters": {
+    "key": "hello"
+  },
+  "headers": {
+    "accept": "*/*",
+    "connection": "keep-alive",
+    "host": "lambda-test-alb-1334523864.us-east-1.elb.amazonaws.com",
+    "user-agent": "curl/7.54.0",
+    "x-amzn-trace-id": "Root=1-5c34e93e-4dea0086f9763ac0667b115a",
+    "x-forwarded-for": "25.12.198.67",
+    "x-forwarded-port": "80",
+    "x-forwarded-proto": "http",
+    "x-imforwards": "20",
+    "x-myheader": "123"
+  },
+  "body": "",
+  "isBase64Encoded": false
+}

events/testdata/alb-lambda-target-request-multivalue-headers.json

@@ -0,0 +1,48 @@
+{
+  "requestContext": {
+    "elb": {
+      "targetGroupArn": "arn:aws:elasticloadbalancing:us-east-1:123456789012:targetgroup/lambda-target/abcdefgh"
+    }
+  },
+  "httpMethod": "GET",
+  "path": "/",
+  "multiValueQueryStringParameters": {
+    "key": [
+      "hello"
+    ]
+  },
+  "multiValueHeaders": {
+    "accept": [
+      "*/*"
+    ],
+    "connection": [
+      "keep-alive"
+    ],
+    "host": [
+      "lambda-test-alb-1234567.us-east-1.elb.amazonaws.com"
+    ],
+    "user-agent": [
+      "curl/7.54.0"
+    ],
+    "x-amzn-trace-id": [
+      "Root=1-5c34e7d4-00ca239424b68028d4c56d68"
+    ],
+    "x-forwarded-for": [
+      "72.21.198.67"
+    ],
+    "x-forwarded-port": [
+      "80"
+    ],
+    "x-forwarded-proto": [
+      "http"
+    ],
+    "x-imforwards": [
+      "20"
+    ],
+    "x-myheader": [
+      "123"
+    ]
+  },
+  "body": "",
+  "isBase64Encoded": false
+}