Add immediately parameter to composeWith

lib/index.js

@@ -1163,7 +1163,9 @@ class Generator extends Base {
   /**
    * Compose this generator with another one.
    * @param  {String|Object|Array} generator  The path to the generator module or an object (see examples)
+   * @param  {Array}  [args]       Arguments passed to the Generator
    * @param  {Object}  [options]   The options passed to the Generator
+   * @param  {boolean}  [immediately] Boolean whether to queue the Generator immediately
    * @return {Generator}    The composed generator
    *
    * @example <caption>Using a peerDependency generator</caption>
@@ -1175,7 +1177,7 @@ class Generator extends Base {
    * @example <caption>Passing a Generator class</caption>
    * this.composeWith({ Generator: MyGenerator, path: '../generator-bootstrap/app/main.js' }, { sass: true });
    */
-  composeWith(generator, args, options) {
+  composeWith(generator, args, options, immediately = false) {
     if (typeof args === 'boolean') {
       args = [];
     } else if (!Array.isArray(args) && typeof args === 'object') {
@@ -1263,7 +1265,7 @@ this.composeWith({
       return instantiatedGenerator;
     }
 
-    if (this._running) {
+    if (this._running || immediately) {
       this.env.queueGenerator(instantiatedGenerator);
     } else {
       this._composedWith.push(instantiatedGenerator);

test/base.js

@@ -886,10 +886,19 @@ describe('Base', () => {
       this.env.registerStub(this.GenCompose, 'composed:gen');
     });
 
-    it('returns the composed generator when pass true', function () {
-      assert(
-        this.dummy.composeWith('composed:gen', true) instanceof this.GenCompose
-      );
+    it('returns the composed generator', function () {
+      assert(this.dummy.composeWith('composed:gen') instanceof this.GenCompose);
+    });
+
+    it('should add to _composedWith', function () {
+      const generator = this.dummy.composeWith('composed:gen');
+      assert(generator instanceof this.GenCompose);
+      assert(generator === this.dummy._composedWith[0]);
+    });
+
+    it('should not add to _composedWith when immediately is true', function () {
+      this.dummy.composeWith('composed:gen', undefined, undefined, true);
+      assert.strictEqual(this.dummy._composedWith.length, 0);
     });
 
     it('runs the composed generators', function (done) {