Allow file writes in any run loop phases

Fix #758
yeoman · Feb 15, 2015 · 077577a · 077577a
1 parent eb229af
commit 077577a
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 3 deletions.
diff --git a/lib/base.js b/lib/base.js
@@ -435,9 +435,14 @@ Base.prototype.run = function run(cb) {
 
   methods.filter(methodIsValid).forEach(addInQueue);
 
-  this.env.runLoop.add('conflicts', this._writeFiles.bind(this), {
-    once: 'write memory fs to disk'
-  });
+  var writeFiles = function () {
+    this.env.runLoop.add('conflicts', this._writeFiles.bind(this), {
+      once: 'write memory fs to disk'
+    });
+  }.bind(this);
+
+  this.env.sharedFs.on('change', writeFiles);
+  writeFiles();
 
   // Add the default conflicts handling
   this.env.runLoop.add('conflicts', function (done) {

diff --git a/test/base.js b/test/base.js
@@ -395,6 +395,17 @@ describe('generators.Base', function () {
       };
       this.testGen.run(done);
     });
+
+    it('allows file writes in any priorities', function (done) {
+      this.TestGenerator.prototype.end = function () {
+        this.fs.write(this.destinationPath('foo.txt'), 'test');
+      };
+
+      this.testGen.run(function () {
+        assert(fs.existsSync(this.testGen.destinationPath('foo.txt')));
+        done();
+      }.bind(this));
+    });
   });
 
   describe('#_', function () {