From b72d24011f1f539ce398719a3a81a2d33e7048ee Mon Sep 17 00:00:00 2001
From: Daniel Nalborczyk <dnalborczyk@gmail.com>
Date: Fri, 2 Aug 2019 09:32:19 -0400
Subject: [PATCH] Add tests for #758, #759

---
 .../integration/uncategorized/serverless.yml    | 17 +++++++++++++++++
 .../integration/uncategorized/uncategorized.js  |  6 ++++++
 .../uncategorized/uncategorized.test.js         |  9 +++++++++
 3 files changed, 32 insertions(+)

diff --git a/src/__tests__/integration/uncategorized/serverless.yml b/src/__tests__/integration/uncategorized/serverless.yml
index 47912526c..3cd93952b 100644
--- a/src/__tests__/integration/uncategorized/serverless.yml
+++ b/src/__tests__/integration/uncategorized/serverless.yml
@@ -32,3 +32,20 @@ functions:
                 application/json;charset=UTF-8: ''
               statusCode: 200
     handler: uncategorized.uncategorizedHandler1
+
+  uncategorizedHandler2:
+    events:
+      - http:
+          integration: lambda
+          method: get
+          path: uncategorized-2
+          responses:
+            default:
+              responseModels:
+                application/json;charset=UTF-8: Empty
+              responseParameters:
+                method.response.header.x-foo: 'integration.response.body.more-foo.property-of-undefined'
+              responseTemplates:
+                application/json;charset=UTF-8: ''
+              statusCode: 200
+    handler: uncategorized.uncategorizedHandler2
diff --git a/src/__tests__/integration/uncategorized/uncategorized.js b/src/__tests__/integration/uncategorized/uncategorized.js
index c951ff425..f4fbef911 100644
--- a/src/__tests__/integration/uncategorized/uncategorized.js
+++ b/src/__tests__/integration/uncategorized/uncategorized.js
@@ -5,3 +5,9 @@ module.exports.uncategorizedHandler1 = async () => {
     foo: 'bar',
   };
 };
+
+module.exports.uncategorizedHandler2 = async () => {
+  return {
+    foo: 'bar',
+  };
+};
diff --git a/src/__tests__/integration/uncategorized/uncategorized.test.js b/src/__tests__/integration/uncategorized/uncategorized.test.js
index 1bbcf5d5f..11ed37d7c 100644
--- a/src/__tests__/integration/uncategorized/uncategorized.test.js
+++ b/src/__tests__/integration/uncategorized/uncategorized.test.js
@@ -36,4 +36,13 @@ describe('handler payload tests', () => {
     const json = await response.json();
     expect(json).toEqual({ foo: 'bar' });
   });
+
+  // issue: https://github.com/dherault/serverless-offline/issues/758
+  // PR: https://github.com/dherault/serverless-offline/pull/759
+  test('Uncategorized 2', async () => {
+    url.pathname = 'uncategorized-2';
+    const response = await fetch(url);
+    const json = await response.json();
+    expect(json).toEqual({ foo: 'bar' });
+  });
 });