uninstall sourcemaps after test results are processed

jestjs · May 18, 2018 · 02558bd · 02558bd
1 parent 6a1e78d
commit 02558bd
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 3 deletions.
diff --git a/packages/jest-runner/src/index.js b/packages/jest-runner/src/index.js
@@ -23,6 +23,7 @@ import exit from 'exit';
 import runTest from './run_test';
 import throat from 'throat';
 import Worker from 'jest-worker';
+import sourcemapSupport from 'source-map-support';
 
 const TEST_WORKER_PATH = require.resolve('./test_worker');
 
@@ -81,7 +82,12 @@ class TestRunner {
               );
             })
             .then(result => onResult(test, result))
-            .catch(err => onFailure(test, err)),
+            .catch(err => onFailure(test, err))
+            .then(res => {
+              sourcemapSupport.resetRetrieveHandlers();
+
+              return res;
+            }),
         ),
       Promise.resolve(),
     );

diff --git a/packages/jest-runner/src/run_test.js b/packages/jest-runner/src/run_test.js
@@ -184,8 +184,6 @@ async function runTestInternal(
     });
   } finally {
     await environment.teardown();
-
-    sourcemapSupport.resetRetrieveHandlers();
   }
 }
 

diff --git a/packages/jest-runner/src/test_worker.js b/packages/jest-runner/src/test_worker.js
@@ -13,6 +13,7 @@ import type {RawModuleMap} from 'types/HasteMap';
 import type {ErrorWithCode} from 'types/Errors';
 
 import exit from 'exit';
+import sourcemapSupport from 'source-map-support';
 import HasteMap from 'jest-haste-map';
 import {separateMessageFromStack} from 'jest-message-util';
 import Runtime from 'jest-runtime';
@@ -84,5 +85,7 @@ export async function worker({
     );
   } catch (error) {
     throw formatError(error);
+  } finally {
+    sourcemapSupport.resetRetrieveHandlers();
   }
 }