use afterEach to restore mock even if tests fail

boblannon · boblannon · commit 7ffef00918af · 2018-02-28T11:24:18.000-05:00
diff --git a/__tests__/lib/github.js b/__tests__/lib/github.js
@@ -25,6 +25,10 @@ function evaluate(fn) {
   );
 }
 
+afterEach(function() {
+  mock.restore();
+});
+
 test('github', function() {
   mock(mockRepo.master);
 
@@ -42,8 +46,6 @@ test('github', function() {
     path: 'index.js',
     url: 'https://github.com/foo/bar/blob/this_is_the_sha/index.js#L6-L8'
   });
-
-  mock.restore();
 });
 
 test('malformed repository', function() {
@@ -60,8 +62,6 @@ test('malformed repository', function() {
       }
     })[0].context.github
   ).toBe(undefined);
-
-  mock.restore();
 });
 
 test('enterprise repository', function() {
@@ -82,8 +82,6 @@ test('enterprise repository', function() {
     url:
       'https://github.enterprise.com/foo/bar/blob/this_is_the_sha/index.js#L6-L8'
   });
-
-  mock.restore();
 });
 
 test('typedef', function() {
@@ -108,6 +106,4 @@ test('typedef', function() {
     path: 'index.js',
     url: 'https://github.com/foo/bar/blob/this_is_the_sha/index.js#L1-L4'
   });
-
-  mock.restore();
 });
