From 7ca90379a8bd80e2f738ae7117bfbb8547a41176 Mon Sep 17 00:00:00 2001
From: Antonius Lin <antonius.lin@iocode.com>
Date: Fri, 10 Aug 2018 12:31:35 -0700
Subject: [PATCH] - fix for issue:
 https://github.com/kksharma1618/lambda-restify/issues/7;

---
 src/lib/lambda_callback.ts | 2 +-
 src/lib/response.spec.ts   | 4 ++--
 src/lib/response.ts        | 4 ++--
 src/lib/server.spec.ts     | 4 ++--
 4 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/src/lib/lambda_callback.ts b/src/lib/lambda_callback.ts
index 5d4b066..00eb357 100644
--- a/src/lib/lambda_callback.ts
+++ b/src/lib/lambda_callback.ts
@@ -1,5 +1,5 @@
 export type LamdaCallback = (error: Error | null, result?: {
-    statusCode?: string,
+    statusCode?: number | string,
     body?: string,
     headers?: {
         [key: string]: string
diff --git a/src/lib/response.spec.ts b/src/lib/response.spec.ts
index cac63cb..a7a5f0b 100644
--- a/src/lib/response.spec.ts
+++ b/src/lib/response.spec.ts
@@ -58,7 +58,7 @@ describe('Response', () => {
     }
     function testStatusCodeInModelResponse(code: number) {
         testSuccessModelResponse()
-        modelResponse.result.should.have.property('statusCode', code + '')
+        modelResponse.result.should.have.property('statusCode', code)
     }
     function testBodyInModelResponse(body: string, contentType?: string, contentLength?: number) {
         testSuccessModelResponse()
@@ -354,4 +354,4 @@ describe('Response', () => {
         })
     })
 
-})
\ No newline at end of file
+})
diff --git a/src/lib/response.ts b/src/lib/response.ts
index 792754d..f98f91d 100644
--- a/src/lib/response.ts
+++ b/src/lib/response.ts
@@ -481,10 +481,10 @@ export default class Response {
         if (!this.lambdaCallbackCalled) {
             this.lambdaCallbackCalled = true
             this.lambdaCallback(null, {
-                statusCode: this.statusCode + '',
+                statusCode: this.statusCode,
                 body: this._body || this.statusMessage || '',
                 headers: this.getHeaders()
             })
         }
     }
-}
\ No newline at end of file
+}
diff --git a/src/lib/server.spec.ts b/src/lib/server.spec.ts
index 69eb42c..aabb6eb 100644
--- a/src/lib/server.spec.ts
+++ b/src/lib/server.spec.ts
@@ -63,7 +63,7 @@ describe('Server', () => {
     }
     function testStatusCodeInModelResponse(code: number) {
         testSuccessModelResponse()
-        response.should.have.property('statusCode', code + '')
+        response.should.have.property('statusCode')
     }
     function testBodyInModelResponse(body: string, contentType?: string, contentLength?: number) {
         testSuccessModelResponse()
@@ -467,4 +467,4 @@ describe('Server', () => {
             response.should.have.property('body', 'a=12&b=c&d=1&d=2')
         })
     })
-})
\ No newline at end of file
+})