Merge pull request #655 from ibnc/custom_imposters

custom imposters repository

.gitignore

@@ -3,6 +3,8 @@ node_modules/
 .DS_Store
 mb*.log
 *.pid
+*.swp
+*.swo
 /dist/
 !scripts/dist/
 mountebank.iml

src/cli/cli.js

@@ -60,6 +60,12 @@ const fs = require('fs-extra'),
             type: 'string',
             global: false
         },
+        impostersRepository: {
+            description: 'path to custom imposters repository',
+            nargs: 1,
+            type: 'string',
+            global: false
+        },
         pidfile: {
             default: 'mb.pid',
             description: 'where the pid is stored for the stop command',

src/models/impostersRepository.js

@@ -12,12 +12,35 @@
  * @returns {Object} - the repository
  */
 function create (config, logger) {
-    if (config.datadir) {
+    if (config.impostersRepository) {
+        const fs = require('fs-extra'),
+            path = require('path'),
+            filename = path.resolve(path.relative(process.cwd(), config.impostersRepository));
+
+        if (fs.existsSync(filename)) {
+            try {
+                return require(filename).create(config, logger);
+            }
+            catch (e) {
+                logger.error(`An error occured while creating custom impostersRepository:\n ${e}`);
+                return {};
+            }
+        }
+        else {
+            logger.warn(`Imposters Respository ${filename} does not exist. The default will be used`);
+            return this.inMemory();
+        }
+    }
+    else if (config.datadir) {
         return require('./filesystemBackedImpostersRepository').create(config, logger);
     }
     else {
-        return require('./inMemoryImpostersRepository').create();
+        return this.inMemory();
     }
 }
 
-module.exports = { create };
+function inMemory () {
+    return require('./inMemoryImpostersRepository').create();
+}
+
+module.exports = { create, inMemory };

test/fakes/fakeImpostersRepository.js

@@ -0,0 +1,10 @@
+'use strict';
+
+
+function create () {
+    return { fake: 'fake' };
+}
+
+module.exports = {
+    create
+};

test/models/impostersRepositoryTest.js

@@ -0,0 +1,34 @@
+'use strict';
+
+const assert = require('assert'),
+    FakeLogger = require('../fakes/fakeLogger'),
+    impostersRepository = require('../../src/models/impostersRepository');
+
+describe('ImpostersRepository', function () {
+    describe('#create', function () {
+        describe('custom impostersRepository', function () {
+            let logger;
+
+            beforeEach(function () {
+                logger = FakeLogger.create();
+                impostersRepository.inMemory = require('../mock').mock();
+            });
+
+            it('should return the custom impostersRepository', function () {
+                const config = {
+                    impostersRepository: 'test/fakes/fakeImpostersRepository.js'
+                };
+                assert.deepEqual(impostersRepository.create(config, logger), require('../fakes/fakeImpostersRepository.js').create());
+                assert.ok(!impostersRepository.inMemory.wasCalled());
+            });
+
+            it('should default to inMemory if the custom impostersRepository does not exist', function () {
+                const config = {
+                    impostersRepository: '../doesnotexist'
+                };
+                impostersRepository.create(config, logger);
+                assert.ok(impostersRepository.inMemory.wasCalled());
+            });
+        });
+    });
+});