jestjs#15215 Reduce memory leak in node env by fully uninstalling sou…

…rce-map-support
eyalroth · Jul 27, 2024 · 7423314 · 7423314
1 parent 9ff3d3e
commit 7423314
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/packages/jest-environment-node/src/index.ts b/packages/jest-environment-node/src/index.ts
@@ -216,6 +216,9 @@ export default class NodeEnvironment implements JestEnvironment<Timer> {
     }
 
     if (this.context) {
+      // source-map-support keeps memory leftovers in `Error.prepareStackTrace`
+      runInContext("Error.prepareStackTrace = () => '';", this.context);
+
       // remove any leftover listeners that may hold references to sizable memory
       this.context.process.removeAllListeners();
       const cluster = runInContext(

diff --git a/packages/jest-runner/src/runTest.ts b/packages/jest-runner/src/runTest.ts
@@ -312,8 +312,12 @@ async function runTestInternal(
         sendMessageToJest,
       );
     } catch (error: any) {
-      // Access stack before uninstalling sourcemaps
-      error.stack;
+      // Access all stacks before uninstalling sourcemaps
+      let e = error;
+      while (typeof e === 'object' && 'stack' in e) {
+        e.stack;
+        e = e?.cause ?? {};
+      }
 
       throw error;
     } finally {