fix: Handled assigning headers in LambdaProxyWebRequest when there ar…

…e no headers present (#2293)
newrelic · Jun 25, 2024 · e4d22f3 · e4d22f3
1 parent dbca830
commit e4d22f3
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/lib/serverless/api-gateway.js b/lib/serverless/api-gateway.js
@@ -80,7 +80,7 @@ function normalizeHeaders(event, lowerCaseKey = false) {
   const headers = event.multiValueHeaders ?? event.headers
 
   if (!headers) {
-    return
+    return {}
   }
 
   return Object.fromEntries(

diff --git a/test/unit/serverless/api-gateway-v2.test.js b/test/unit/serverless/api-gateway-v2.test.js
@@ -272,3 +272,17 @@ tap.test('should capture request headers', (t) => {
     t.end()
   }
 })
+
+tap.test('should not crash when headers are non-existent', (t) => {
+  const { lambda, event, functionContext, responseBody } = t.context
+  delete event.headers
+
+  const wrappedHandler = lambda.patchLambdaHandler((event, context, callback) => {
+    callback(null, responseBody)
+  })
+
+  t.doesNotThrow(() => {
+    wrappedHandler(event, functionContext, () => {})
+  })
+  t.end()
+})