cover RequestFromContext

aws · Apr 23, 2023 · 1650763 · 1650763
1 parent 787a3c2
commit 1650763
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/lambdaurl/http_handler_test.go b/lambdaurl/http_handler_test.go
@@ -142,3 +142,16 @@ func TestWrap(t *testing.T) {
 		})
 	}
 }
+
+func TestRequestContext(t *testing.T) {
+	var req *events.LambdaFunctionURLRequest
+	require.NoError(t, json.Unmarshal(helloRequest, &req))
+	handler := Wrap(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		reqFromContext, exists := RequestFromContext(r.Context())
+		require.True(t, exists)
+		require.NotNil(t, reqFromContext)
+		assert.Equal(t, req, reqFromContext)
+	}))
+	_, err := handler(context.Background(), req)
+	require.NoError(t, err)
+}