lib: update test harness.js to use DisposableStack

Author: jasnell

lib/internal/test_runner/harness.js

@@ -8,6 +8,7 @@ const {
   SafeMap,
   SafePromiseAllReturnVoid,
 } = primordials;
+
 const { getCallerLocation } = internalBinding('util');
 const {
   createHook,
@@ -230,6 +231,9 @@ function setupProcessState(root, globalOptions) {
   const rejectionHandler =
     createProcessEventHandler('unhandledRejection', root);
   const coverage = configureCoverage(root, globalOptions);
+
+  const disposableStack = new DisposableStack();  // eslint-disable-line no-undef
+
   const exitHandler = async (kill) => {
     if (root.subtests.length === 0 && (root.hooks.before.length > 0 || root.hooks.after.length > 0)) {
       // Run global before/after hooks in case there are no tests
@@ -254,27 +258,21 @@ function setupProcessState(root, globalOptions) {
     }
 
     hook.disable();
-    process.removeListener('uncaughtException', exceptionHandler);
-    process.removeListener('unhandledRejection', rejectionHandler);
-    process.removeListener('beforeExit', exitHandler);
-    if (globalOptions.isTestRunner) {
-      process.removeListener('SIGINT', terminationHandler);
-      process.removeListener('SIGTERM', terminationHandler);
-    }
+    disposableStack.dispose();
   };
 
   const terminationHandler = async () => {
     await exitHandler(true);
     process.exit();
   };
 
-  process.on('uncaughtException', exceptionHandler);
-  process.on('unhandledRejection', rejectionHandler);
-  process.on('beforeExit', exitHandler);
+  disposableStack.use(process.use('uncaughtException', exceptionHandler));
+  disposableStack.use(process.use('unhandledRejection', rejectionHandler));
+  disposableStack.use(process.use('beforeExit', exitHandler));
   // TODO(MoLow): Make it configurable to hook when isTestRunner === false.
   if (globalOptions.isTestRunner) {
-    process.on('SIGINT', terminationHandler);
-    process.on('SIGTERM', terminationHandler);
+    disposableStack.use(process.use('SIGINT', terminationHandler));
+    disposableStack.use(process.use('SIGTERM', terminationHandler));
   }
 
   root.harness.coverage = FunctionPrototypeBind(collectCoverage, null, root, coverage);