Implement getSnapshot (#143)

* Use temp-dir for real tmp directory. * Implement getSnapshot
yeoman · May 1, 2021 · 479fcde · 479fcde
1 parent 1a0b1d0
commit 479fcde
Show file tree

Hide file tree

Showing 6 changed files with 254 additions and 123 deletions.
diff --git a/lib/run-context.js b/lib/run-context.js
@@ -2,11 +2,12 @@
 const crypto = require('crypto');
 const fs = require('fs');
 const path = require('path');
-const os = require('os');
 const assert = require('assert');
 const _ = require('lodash/string');
 const util = require('util');
 const {EventEmitter} = require('events');
+const tempDirectory = require('temp-dir');
+
 const RunResult = require('./run-result');
 
 /**
@@ -344,7 +345,7 @@ RunContext.prototype.cd = function (dirPath) {
  */
 RunContext.prototype.inTmpDir = function (cb) {
   return this.inDir(
-    path.join(os.tmpdir(), crypto.randomBytes(20).toString('hex')),
+    path.join(tempDirectory, crypto.randomBytes(20).toString('hex')),
     cb
   );
 };

diff --git a/lib/run-result.js b/lib/run-result.js
@@ -49,6 +49,10 @@ module.exports = class RunResult {
     );
   }
 
+  getSnapshot() {
+    return this.fs.dump(this.cwd);
+  }
+
   /**
    * Prints files names and contents from mem-fs
    * @param {...string} files - Files to print or empty for entire mem-fs