Include headers field in the output object

Author: janaz

src/fixResponseHeaders.ts

@@ -38,11 +38,14 @@ const fixResponseHeaders = (headers: OutgoingHttpHeaders, supportMultiHeaders: b
       delete multiValueHeaders['transfer-encoding']
     }
   }
+  const output: LambdaResponseHeaders = { headers: singleValueHeaders }
   if (supportMultiHeaders) {
-    return { multiValueHeaders };
-  } else {
-    return { headers: singleValueHeaders, cookies }
+    output.multiValueHeaders = multiValueHeaders
   }
+  if (supportCookies) {
+    output.cookies = cookies
+  }
+  return output
 };
 
 export default fixResponseHeaders;

test/integration.test.ts

@@ -144,7 +144,8 @@ describe('integration', () => {
     expect(response.isBase64Encoded).toEqual(false);
     expect(response.multiValueHeaders!["content-type"][0]).toEqual('application/json; charset=utf-8');
     expect(response.multiValueHeaders!["x-powered-by"][0]).toEqual('Express');
-    expect(response.headers).toBeUndefined()
+    expect(response.headers!["content-type"]).toEqual('application/json; charset=utf-8');
+    expect(response.headers!["x-powered-by"]).toEqual('Express');
     const json = JSON.parse(response.body);
     expect(json).toEqual({
       baseUrl: "",
@@ -187,7 +188,8 @@ describe('integration', () => {
     expect(response.isBase64Encoded).toEqual(false);
     expect(response.multiValueHeaders!["content-type"][0]).toEqual('application/json; charset=utf-8');
     expect(response.multiValueHeaders!["x-powered-by"][0]).toEqual('Express');
-    expect(response.headers).toBeUndefined()
+    expect(response.headers!["content-type"]).toEqual('application/json; charset=utf-8');
+    expect(response.headers!["x-powered-by"]).toEqual('Express');
     const json = JSON.parse(response.body);
     expect(json).toEqual({
       baseUrl: "",