Restore MockAdapter in spec/javascripts/diffs/store/actions_spec.js

1 files changed, 13 insertions, 4 deletions

diff --git a/spec/javascripts/diffs/store/actions_spec.js b/spec/javascripts/diffs/store/actions_spec.js
index 73f83d31e12..e61780c9928 100644
--- a/spec/javascripts/diffs/store/actions_spec.js
+++ b/spec/javascripts/diffs/store/actions_spec.js
@@ -61,7 +61,10 @@ describe('DiffsStoreActions', () => {
           { type: types.SET_DIFF_DATA, payload: res },
         ],
         [],
-        done,
+        () => {
+          mock.restore();
+          done();
+        },
       );
     });
   });
@@ -154,7 +157,10 @@ describe('DiffsStoreActions', () => {
           },
         ],
         [],
-        done,
+        () => {
+          mock.restore();
+          done();
+        },
       );
     });
   });
@@ -177,7 +183,10 @@ describe('DiffsStoreActions', () => {
           },
         ],
         [],
-        done,
+        () => {
+          mock.restore();
+          done();
+        },
       );
     });
   });
@@ -187,7 +196,7 @@ describe('DiffsStoreActions', () => {
       testAction(
         actions.expandAllFiles,
         null,
-        { },
+        {},
         [
           {
             type: types.EXPAND_ALL_FILES,