Add the root to the unmock function for the fs module

negativetwelve · Nov 9, 2017 · 3785b8c · 3785b8c
1 parent 64b6067
commit 3785b8c
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/packages/jest-plugin-fs/src/__tests__/fs.spec.js b/packages/jest-plugin-fs/src/__tests__/fs.spec.js
@@ -84,6 +84,8 @@ describe('fs', () => {
       beforeEach(() => fs.unmock([filename]));
 
       it('should exist in the virtual filesystem', () => {
+        console.log(fs.files());
+        console.log(filename, path.resolve(filename));
         expect(fsExtra.readFileSync(filename, 'utf8')).toEqual('this is a test\n');
       });
     });

diff --git a/packages/jest-plugin-fs/src/fs.js b/packages/jest-plugin-fs/src/fs.js
@@ -56,11 +56,11 @@ const mock = (filesystem = {}, fsRoot = root) => {
  * Reads the passed in files from the filesystem and adds them to the
  * virtual mocked filesystem
  */
-const unmock = (files = []) => {
+const unmock = (files = [], fsRoot = root) => {
   const readAll = (all, file) => ({...all, [file]: read(file)});
   const filesystem = files.reduce(readAll, {});
 
-  mock(filesystem, '');
+  mock(filesystem, fsRoot);
 };
 
 /**
@@ -75,12 +75,12 @@ const restore = () => vol.reset();
 const jestFs = {
   root,
   files,
+  read,
+  restore,
 
   // NOTE(mark): This allows the root to be configurable by the fs object.
   mock: (filesystem) => mock(filesystem, jestFs.root),
-  read,
-  restore,
-  unmock,
+  unmock: (files) => unmock(files, jestFs.root),
 };