Allow for custom Access-Control-Allow-Headers

package.json

@@ -1,6 +1,6 @@
 {
   "name": "api-mock",
-  "version": "0.1.0",
+  "version": "0.1.1",
   "description": "A mock server generated from your API Blueprint.",
   "author": "Evan Cordell <cordell.evan@gmail.com>",
   "main": "lib/api-mock.js",

src/api-mock.coffee

@@ -28,7 +28,10 @@ class ApiMock
       )
 
     if !@configuration.options['cors-disable']
-      corsSupport = new CorsSupport @app
+      corsSupport = new CorsSupport(
+        @app,
+            headers: @configuration.options['custom-headers']
+      )
 
   run: () ->
     app = @app

src/cors-support.coffee

@@ -1,13 +1,16 @@
 winston = require 'winston'
 
 class CorsSupport
-  constructor: (app) ->
+  constructor: (app, configuration) ->
 
     options =
       origin: '*'
       methods: 'GET, PUT, POST, PATCH, DELETE, TRACE, OPTIONS'
       headers: 'Origin, X-Requested-With, Content-Type, Accept, Authorization, Referer, Prefer'
 
+    if configuration.headers
+      options.headers = options.headers + ', ' + configuration.headers
+
     app.all '*', (req, res, next) ->
       unless req.get('Origin')
         next()

test/unit/api-mock-test.coffee

@@ -73,6 +73,21 @@ describe 'ApiMock class', () ->
           assert.ok api_mock.configuration.options.port == 3005
           assert.notOk CorsSupportStub.called
 
+      describe 'with custom Access-Control-Allow-Headers', () ->
+
+        beforeEach ()->
+          configuration =
+            blueprintPath: './test/fixtures/single-get.apib',
+            options:
+              port: 3005
+              'custom-headers': 'x-custom-header'
+              'cors-disable': false
+
+        it 'should add custom header to configuration', () ->
+          api_mock = new ApiMock(configuration)
+          assert.ok api_mock.configuration.options['custom-headers'] == 'x-custom-header'
+          assert.ok CorsSupportStub.called
+
     describe 'with invalid configuration', () ->
 
       beforeEach ()->