test_runner: cleanup global event listeners after run

EddieAbbondanzio · marco-ippolito · commit 367f9e77f375 · 2024-08-19T10:32:46.000+02:00
PR-URL: #53878 Fixes: #53868 Reviewed-By: Matteo Collina <matteo.collina@gmail.com> Reviewed-By: James M Snell <jasnell@gmail.com> Reviewed-By: Colin Ihrig <cjihrig@gmail.com>
diff --git a/lib/internal/test_runner/harness.js b/lib/internal/test_runner/harness.js
@@ -170,8 +170,13 @@ function setup(root) {
       kCancelledByParent));
 
     hook.disable();
-    process.removeListener('unhandledRejection', rejectionHandler);
     process.removeListener('uncaughtException', exceptionHandler);
+    process.removeListener('unhandledRejection', rejectionHandler);
+    process.removeListener('beforeExit', exitHandler);
+    if (globalOptions.isTestRunner) {
+      process.removeListener('SIGINT', terminationHandler);
+      process.removeListener('SIGTERM', terminationHandler);
+    }
   };
 
   const terminationHandler = () => {
diff --git a/lib/internal/test_runner/runner.js b/lib/internal/test_runner/runner.js
@@ -591,6 +591,7 @@ function run(options) {
   }
 
   let postRun = () => root.postRun();
+  let teardown = () => root.harness.teardown();
   let filesWatcher;
   const opts = {
     __proto__: null,
@@ -604,6 +605,7 @@ function run(options) {
   if (watch) {
     filesWatcher = watchFiles(testFiles, opts);
     postRun = undefined;
+    teardown = undefined;
   }
   const runFiles = () => {
     root.harness.bootstrapComplete = true;
@@ -615,7 +617,8 @@ function run(options) {
     });
   };
 
-  PromisePrototypeThen(PromisePrototypeThen(PromiseResolve(setup?.(root.reporter)), runFiles), postRun);
+  const setupPromise = PromiseResolve(setup?.(root.reporter));
+  PromisePrototypeThen(PromisePrototypeThen(PromisePrototypeThen(setupPromise, runFiles), postRun), teardown);
 
   return root.reporter;
 }
diff --git a/test/parallel/test-runner-run.mjs b/test/parallel/test-runner-run.mjs
@@ -542,3 +542,13 @@ describe('forceExit', () => {
     });
   });
 });
+
+
+// exitHandler doesn't run until after the tests / after hooks finish.
+process.on('exit', () => {
+  assert.strictEqual(process.listeners('uncaughtException').length, 0);
+  assert.strictEqual(process.listeners('unhandledRejection').length, 0);
+  assert.strictEqual(process.listeners('beforeExit').length, 0);
+  assert.strictEqual(process.listeners('SIGINT').length, 0);
+  assert.strictEqual(process.listeners('SIGTERM').length, 0);
+});
