fix: fix aws api gateway v2 event source cookies handling (#389)

CodeGenieApp · May 11, 2021 · 6e0868c · 6e0868c
1 parent 909c514
commit 6e0868c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/jest-helpers/api-gateway-v2-event.js b/jest-helpers/api-gateway-v2-event.js
@@ -102,6 +102,7 @@ function makeApiGatewayV2Response (values = {}, {
   }
   values.headers = convertMultiValueHeadersToHeaders({ multiValueHeaders: values.multiValueHeaders })
   delete values.multiValueHeaders
+  delete values.headers['set-cookie']
 
   if (shouldConvertContentLengthToInt) {
     // APIGWV2 returns content-length as a number instead of a string under certain conditions:

diff --git a/src/event-sources/aws/api-gateway-v2.js b/src/event-sources/aws/api-gateway-v2.js
@@ -62,8 +62,10 @@ function getResponseToApiGateway ({
     isBase64Encoded
   }
 
-  if (headers['set-cookie']) {
-    responseToApiGateway.cookies = headers['set-cookie']
+  const cookies = headers['set-cookie']
+  if (cookies) {
+    responseToApiGateway.cookies = Array.isArray(cookies) ? cookies : [cookies]
+    delete headers['set-cookie']
   }
 
   responseToApiGateway.headers = getCommaDelimitedHeaders({ headersMap: headers })