Stubby should have an option to send response headers back unmolested

CONTRIBUTERS.md

@@ -24,6 +24,8 @@
 
 [JJ4TH](https://github.com/jj4th)
 
+[Aparna Sathyanarayan](https://github.com/aparna-sath)
+
 # Special Thanks
 
 Alexander Zagniotov

README.md

@@ -53,7 +53,7 @@ Some systems require you to `sudo` before running services on certain ports (lik
 
 ```
 stubby [-a <port>] [-c <file>] [-d <file>] [-h] [-k <file>] [-l <hostname>] [-p <file>] [-q]
-       [-s <port>] [-t <port>] [-v] [-w]
+       [-s <port>] [-t <port>] [-v] [-w] [-q] [-r]
 
 -a, --admin <port>          Port for admin portal. Defaults to 8889.
 -c, --cert <file>           Certificate file. Use with --key.
@@ -67,6 +67,7 @@ stubby [-a <port>] [-c <file>] [-d <file>] [-h] [-k <file>] [-l <hostname>] [-p
 -t, --tls <port>            Port for https stubs portal. Defaults to 7443.
 -v, --version               Prints stubby's version number.
 -w, --watch                 Auto-reload data file when edits are made.
+-r, --dittoResponse         Ditto the response back to the client, without any modifications.
 ```
 
 When used from the command-line, `stubby` responds to the `SIGHUP` signal to reload its configuration.
@@ -671,6 +672,7 @@ What can I do with it, you ask? Read on!
    * `quiet`: defaults to `true`. Pass in `false` to have console output (if available)
    * `_httpsOptions`: additional options to pass to the [underlying tls
      server](http://nodejs.org/api/tls.html#tls_tls_createserver_options_secureconnectionlistener).
+   * `dittoResponse`: tells stubby to send the response back without lower-casing the headers
 * `callback`: takes one parameter: the error message (if there is one), undefined otherwise
 
 #### start([callback])

src/console/cli.js

@@ -69,6 +69,11 @@ var options = [{
   name: 'watch',
   flag: 'w',
   description: 'Auto-reload data file when edits are made.'
+}, {
+  name: 'dittoResponse',
+  flag: 'r',
+  default: false,
+  description: 'Maintain casing for response headers.'
 }];
 
 function help (go) {

src/main.js

@@ -99,6 +99,7 @@ Stubby.prototype.start = function (o, cb) {
 
     if (errors) { return callback(errors); }
     if (options.datadir != null) { self.endpoints.datadir = options.datadir; }
+    if (options.dittoResponse != null) { self.endpoints.dittoResponse = options.dittoResponse; }
 
     self.endpoints.create(options.data, onEndpointLoaded);
 

src/models/endpoint.js

@@ -6,14 +6,15 @@ var http = require('http');
 var q = require('querystring');
 var out = require('../console/out');
 
-function Endpoint (endpoint, datadir) {
+function Endpoint (endpoint, datadir, dittoResponse) {
   if (endpoint == null) { endpoint = {}; }
   if (datadir == null) { datadir = process.cwd(); }
+  if (dittoResponse == null) { dittoResponse = false; }
 
   Object.defineProperty(this, 'datadir', { value: datadir });
 
   this.request = purifyRequest(endpoint.request);
-  this.response = purifyResponse(this, endpoint.response);
+  this.response = purifyResponse(this, endpoint.response, dittoResponse);
   this.hits = 0;
 }
 
@@ -117,7 +118,7 @@ function purifyRequest (incoming) {
   outgoing = {
     url: incoming.url,
     method: incoming.method == null ? 'GET' : incoming.method,
-    headers: purifyHeaders(incoming.headers),
+    headers: purifyHeaders(incoming.headers, true),
     query: incoming.query,
     file: incoming.file,
     post: incoming.post,
@@ -133,7 +134,7 @@ function purifyRequest (incoming) {
   return outgoing;
 }
 
-function purifyResponse (me, incoming) {
+function purifyResponse (me, incoming, dittoResponse) {
   var outgoing = [];
 
   if (incoming == null) { incoming = []; }
@@ -145,7 +146,7 @@ function purifyResponse (me, incoming) {
       outgoing.push(record(me, response));
     } else {
       outgoing.push(pruneUndefined({
-        headers: purifyHeaders(response.headers),
+        headers: purifyHeaders(response.headers, !dittoResponse),
         status: parseInt(response.status, 10) || 200,
         latency: parseInt(response.latency, 10) || null,
         file: response.file,
@@ -157,13 +158,17 @@ function purifyResponse (me, incoming) {
   return outgoing;
 }
 
-function purifyHeaders (incoming) {
+function purifyHeaders (incoming, shouldLowerCase) {
   var prop;
   var outgoing = {};
 
   for (prop in incoming) {
     if (Object.prototype.hasOwnProperty.call(incoming, prop)) {
-      outgoing[prop.toLowerCase()] = incoming[prop];
+      if (shouldLowerCase) {
+        outgoing[prop.toLowerCase()] = incoming[prop];
+      } else {
+        outgoing[prop] = incoming[prop];
+      }
     }
   }
 

src/models/endpoints.js

@@ -15,6 +15,7 @@ function Endpoints (data, callback, datadir) {
   if (callback == null) { callback = noop; }
   if (datadir == null) { datadir = process.cwd(); }
 
+  this.dittoResponse = false;
   this.datadir = datadir;
   this.db = {};
   this.lastId = 0;
@@ -27,7 +28,7 @@ Endpoints.prototype.create = function (data, callback) {
   if (callback == null) { callback = noop; }
 
   function insert (item) {
-    item = new Endpoint(item, self.datadir);
+    item = new Endpoint(item, self.datadir, self.dittoResponse);
     item.id = ++self.lastId;
     self.db[item.id] = item;
     callback(null, clone(item));

test/cli.js

@@ -231,6 +231,26 @@ describe('CLI', function () {
     });
   });
 
+  describe('-r, --dittoResponse', function () {
+    it('should return default if no flag provided', function () {
+      const expected = false;
+      const actual = sut.getArgs([]);
+      assert.strictEqual(actual.dittoResponse, expected);
+    });
+
+    it('should return supplied value when provided', function () {
+      const expected = true;
+      const actual = sut.getArgs(['-r', expected]);
+      assert.strictEqual(actual.dittoResponse, expected);
+    });
+
+    it('should return supplied value when provided with full flag', function () {
+      const expected = true;
+      const actual = sut.getArgs(['--dittoResponse', expected]);
+      assert.strictEqual(actual.dittoResponse, expected);
+    });
+  });
+
   describe('getArgs', function () {
     it('should gather all arguments', function () {
       var actual;
@@ -247,6 +267,7 @@ describe('CLI', function () {
         quiet: true,
         watch: filename,
         datadir: process.cwd(),
+        dittoResponse: false,
         help: undefined, // eslint-disable-line no-undefined
         version: (require('../package.json')).version
       };

test/endpoint.js

@@ -269,6 +269,33 @@ describe('Endpoint', function () {
       assert.deepStrictEqual(actual.request.headers, expected.request);
     });
 
+    it('should not lower case response headers properties if dittoResponse is true', function () {
+      var actual, expected;
+      this.data.request = {
+        headers: {
+          'Content-Type': 'application/json'
+        }
+      };
+      this.data.response = {
+        headers: {
+          'Content-Type': 'application/json'
+        }
+      };
+      expected = {
+        request: {
+          'content-type': 'application/json'
+        },
+        response: {
+          'Content-Type': 'application/json'
+        }
+      };
+
+      actual = new Endpoint(this.data, null, true);
+
+      assert.deepStrictEqual(actual.response[0].headers, expected.response);
+      assert.deepStrictEqual(actual.request.headers, expected.request);
+    });
+
     it('should define multiple headers with same name', function () {
       var actual, expected;
       this.data.request = {