unpatch: Restore the original require('fs').unpatch so multiple virtual fs implementations can layer over each other.

papandreou · papandreou · commit 613d0f16d0ff · 2013-10-06T22:18:53.000+02:00
diff --git a/lib/GitWrappedFs.js b/lib/GitWrappedFs.js
@@ -197,13 +197,17 @@ GitWrappedFs.patchInPlace = function (fs) {
         gitWrappedFs = new GitWrappedFs();
     _.extend(fs, gitWrappedFs);
     fs.unpatch = function () {
+        if ('unpatch' in fsShallowCopy) {
+            fs.unpatch = fsShallowCopy.unpatch;
+        } else {
+            delete fs.unpatch;
+        }
         Object.keys(gitWrappedFs).forEach(function (propertyName) {
             if (propertyName in fsShallowCopy) {
                 fs[propertyName] = fsShallowCopy[propertyName];
             } else {
                 delete fs[propertyName];
             }
         });
-        delete fs.unpatch;
     };
 };
diff --git a/test/GitWrappedFs.js b/test/GitWrappedFs.js
@@ -145,6 +145,7 @@ describe('GitWrappedFs', function () {
 
     describe('patching the built-in fs module "in-place"', function () {
         var originalReadFile = require('fs').readFile;
+        require('fs').unpatch = 123;
         before(function () {
             GitWrappedFs.patchInPlace();
         });
@@ -177,8 +178,8 @@ describe('GitWrappedFs', function () {
 
         after(function () {
             require('fs').unpatch();
-            expect(require('fs'), 'not to have key', 'unpatch');
             expect(require('fs').readFile, 'to be', originalReadFile);
+            expect(require('fs').unpatch, 'to equal', 123);
         });
     });
 });
