work on #742 withLocalConfig - basic tests & implementation

yeoman · Jan 23, 2015 · f8b7f3e · f8b7f3e
1 parent 461a181
commit f8b7f3e
Show file tree

Hide file tree

Showing 3 changed files with 38 additions and 0 deletions.
diff --git a/lib/test/helpers.js b/lib/test/helpers.js
@@ -212,6 +212,15 @@ exports.restorePrompt = function (generator) {
   generator.env.adapter.prompt.restoreDefaultPrompts();
 };
 
+/**
+ * Provide mocked values to the config
+ * @param  {Generator} generator - a Yeoman generator
+ * @param  {Ojbect} localConfig - localConfig - should look just like if called config.getAll()
+ */
+exports.mockLocalConfig = function (generator, localConfig) {
+  generator.config.defaults(localConfig);
+};
+
 /**
  * Create a simple, dummy generator
  */

diff --git a/lib/test/run-context.js b/lib/test/run-context.js
@@ -29,6 +29,7 @@ var RunContext = module.exports = function RunContext(Generator, settings) {
   this.args = [];
   this.options = {};
   this.answers = {};
+  this.localConfig = null;
   this.dependencies = [];
   this.Generator = Generator;
   this.settings = _.extend({ tmpdir: true }, settings);
@@ -86,6 +87,10 @@ RunContext.prototype._run = function () {
 
   helpers.mockPrompt(this.generator, this.answers);
 
+  if (this.localConfig) { // only mock local config when withLocalConfig was called
+    helpers.mockLocalConfig(this.generator, this.localConfig);
+  }
+
   this.generator.on('error', this.emit.bind(this, 'error'));
   this.generator.once('end', function () {
     helpers.restorePrompt(this.generator);
@@ -196,6 +201,17 @@ RunContext.prototype.withGenerators = function (dependencies) {
   return this;
 };
 
+/**
+ * mock the local configuration with the provided config
+ * @param  {Object} localConfig - should look just like if called config.getAll()
+ * @return {this}
+ */
+RunContext.prototype.withLocalConfig = function (localConfig) {
+  assert(_.isObject(localConfig), 'config should be an object');
+  this.localConfig = localConfig;
+  return this;
+};
+
 /**
  * Add a callback to be called after the generator has ran
  * @deprecated `onEnd` is deprecated, use .on('end', onEndHandler) instead.

diff --git a/test/run-context.js b/test/run-context.js
@@ -383,4 +383,17 @@ describe('RunContext', function () {
       }.bind(this));
     });
   });
+
+  describe('#withLocalConfig()', function () {
+    it('provides config to the generator', function (done) {
+      this.ctx.withLocalConfig({
+        some: true,
+        data: 'here'
+      }).on('ready', function () {
+        assert.equal(this.ctx.generator.config.get('some'), true);
+        assert.equal(this.ctx.generator.config.get('data'), 'here');
+        done();
+      }.bind(this));
+    });
+  });
 });