Add routeKey and stage and others to Lambda event and update tests

Author: daido1976

main.go

@@ -9,6 +9,7 @@ import (
 	"net/http"
 	"os"
 	"strings"
+	"time"
 
 	"github.com/aws/aws-lambda-go/events"
 )
@@ -53,19 +54,26 @@ func handler(w http.ResponseWriter, r *http.Request) {
 	}
 }
 
+// See. https://docs.aws.amazon.com/lambda/latest/dg/urls-invocation.html#urls-payloads
 func buildLambdaEvent(r *http.Request) (*events.APIGatewayV2HTTPRequest, error) {
 	bodyBytes, _ := io.ReadAll(r.Body)
 	body := string(bodyBytes)
 
 	return &events.APIGatewayV2HTTPRequest{
 		Version:               "2.0",
+		RouteKey:              "$default",
 		RawPath:               r.URL.Path,
 		RawQueryString:        r.URL.RawQuery,
 		Headers:               joinValuesWithComma(r.Header),
 		QueryStringParameters: joinValuesWithComma(r.URL.Query()),
 		Body:                  body,
 		IsBase64Encoded:       false,
 		RequestContext: events.APIGatewayV2HTTPRequestContext{
+			RouteKey:   "$default",
+			Stage:      "$default",
+			Time:       time.Now().Format("02/Jan/2006:15:04:05 -0700"),
+			TimeEpoch:  time.Now().Unix(),
+			DomainName: r.Host,
 			HTTP: events.APIGatewayV2HTTPRequestContextHTTPDescription{
 				Method:    r.Method,
 				Path:      r.URL.Path,

main_test.go

@@ -84,6 +84,10 @@ func TestHandler(t *testing.T) {
 		t.Errorf("expected version to be '2.0', got %v", requestedBodyJSON["version"])
 	}
 
+	if requestedBodyJSON["routeKey"] != "$default" {
+		t.Errorf("expected routeKey to be '$default', got %v", requestedBodyJSON["routeKey"])
+	}
+
 	if requestedBodyJSON["rawPath"] != "/foo/bar" {
 		t.Errorf("expected rawPath to be '/foo/bar', got %v", requestedBodyJSON["rawPath"])
 	}
@@ -114,6 +118,26 @@ func TestHandler(t *testing.T) {
 	}
 
 	requestContext := requestedBodyJSON["requestContext"].(map[string]interface{})
+	if requestContext["routeKey"] != "$default" {
+		t.Errorf("expected routeKey to be '$default', got %v", requestContext["routeKey"])
+	}
+
+	if requestContext["stage"] != "$default" {
+		t.Errorf("expected stage to be '$default', got %v", requestContext["stage"])
+	}
+
+	if _, ok := requestContext["time"].(string); !ok {
+		t.Errorf("expected time to be a string, got %v", requestContext["time"])
+	}
+
+	if _, ok := requestContext["timeEpoch"].(float64); !ok {
+		t.Errorf("expected timeEpoch to be a float64, got %v", requestContext["timeEpoch"])
+	}
+
+	if requestContext["domainName"] != req.Host {
+		t.Errorf("expected domainName to be '%s', got %v", req.Host, requestContext["domainName"])
+	}
+
 	httpContext := requestContext["http"].(map[string]interface{})
 
 	if httpContext["method"] != "POST" {
@@ -131,4 +155,5 @@ func TestHandler(t *testing.T) {
 	if httpContext["userAgent"] != "Go-http-client/1.1" {
 		t.Errorf("expected userAgent to be 'Go-http-client/1.1', got %v", httpContext["userAgent"])
 	}
+
 }

test/e2e.test.mjs

@@ -12,7 +12,7 @@ test("E2E Test for GET Request", async () => {
     message: "Hello from Lambda!",
     event: {
       version: "2.0",
-      routeKey: "",
+      routeKey: "$default",
       rawPath: "/foo/bar",
       rawQueryString: "testkey=testvalue",
       headers: expect.objectContaining({
@@ -23,14 +23,12 @@ test("E2E Test for GET Request", async () => {
         testkey: "testvalue",
       },
       requestContext: {
-        routeKey: "",
-        accountId: "",
-        stage: "",
-        requestId: expect.any(String),
-        apiId: "",
-        domainName: "",
-        domainPrefix: "",
-        time: expect.any(String),
+        routeKey: "$default",
+        stage: "$default",
+        domainName: "localhost:8080",
+        time: expect.stringMatching(
+          /^\d{2}\/[A-Za-z]{3}\/\d{4}:\d{2}:\d{2}:\d{2} [-+]\d{4}$/
+        ),
         timeEpoch: expect.any(Number),
         http: expect.objectContaining({
           method: "GET",
@@ -78,7 +76,7 @@ test("E2E Test for POST Request", async () => {
     message: "Hello from Lambda!",
     event: {
       version: "2.0",
-      routeKey: "",
+      routeKey: "$default",
       rawPath: "/foo/bar",
       rawQueryString: "",
       headers: expect.objectContaining({
@@ -88,14 +86,12 @@ test("E2E Test for POST Request", async () => {
         "User-Agent": expect.any(String),
       }),
       requestContext: {
-        routeKey: "",
-        accountId: "",
-        stage: "",
-        requestId: expect.any(String),
-        apiId: "",
-        domainName: "",
-        domainPrefix: "",
-        time: expect.any(String),
+        routeKey: "$default",
+        stage: "$default",
+        domainName: "localhost:8080",
+        time: expect.stringMatching(
+          /^\d{2}\/[A-Za-z]{3}\/\d{4}:\d{2}:\d{2}:\d{2} [-+]\d{4}$/
+        ),
         timeEpoch: expect.any(Number),
         http: expect.objectContaining({
           method: "POST",