Merge pull request dherault#402 from njriordan/binary-support-with-lambda-integration

Add binary support when using lambda integration

Author: dherault

package.json

@@ -72,6 +72,7 @@
     "Matt Jonker (https://github.com/msjonker)",
     "Michael MacDonald (https://github.com/mjmac)",
     "Miso (Mike) Zmiric (https://github.com/mzmiric5)",
+    "Niall Riordan (https://github.com/njriordan)",
     "Norimitsu Yamashita (https://github.com/nori3tsu)",
     "Oliv (https://github.com/obearn)",
     "Paul Esson (https://github.com/thepont)",

src/index.js

@@ -584,6 +584,7 @@ class Offline {
               let result = data;
               let responseName = 'default';
               const responseContentType = endpoint.responseContentType;
+              const contentHandling = endpoint.contentHandling;
 
               /* RESPONSE SELECTION (among endpoint's possible responses) */
 
@@ -731,7 +732,14 @@ class Offline {
                   override: false, // Maybe a responseParameter set it already. See #34
                 });
                 response.statusCode = statusCode;
-                response.source = result;
+                if (contentHandling === 'CONVERT_TO_BINARY') {
+                  response.encoding = 'binary';
+                  response.source = Buffer.from(result, 'base64');
+                  response.variety = 'buffer';
+                }
+                else {
+                  response.source = result;
+                }
               }
               else if (integration === 'lambda-proxy') {
                 response.statusCode = statusCode = result.statusCode || 200;
@@ -742,7 +750,7 @@ class Offline {
                 if (!_.isUndefined(result.body)) {
                   if (result.isBase64Encoded) {
                     response.encoding = 'binary';
-                    response.source = new Buffer(result.body, 'base64');
+                    response.source = Buffer.from(result.body, 'base64');
                     response.variety = 'buffer';
                   }
                   else {