Add registerCompletionHandler to main.d.cts

entrypoints/main.cjs

@@ -1,2 +1,6 @@
 'use strict';
 module.exports = require('../lib/worker/main.cjs');
+
+const {registerCompletionHandler} = require('../lib/worker/completion-handlers.cjs');
+
+module.exports.registerCompletionHandler = registerCompletionHandler;

entrypoints/main.d.cts

@@ -10,3 +10,11 @@ declare const test: TestFn;
 
 /** Call to declare a test, or chain to declare hooks or test modifiers */
 export default test;
+
+/**
+ * Register a function to be called when AVA has completed a test run without uncaught exceptions or unhandled rejections.
+ *
+ * Completion handlers are invoked in order of registration. Results are not awaited.
+ */
+declare const registerCompletionHandler: (handler: () => void) => void;
+export {registerCompletionHandler};

entrypoints/main.mjs

@@ -1,2 +1,2 @@
 export {default} from '../lib/worker/main.cjs';
-export {registerCompletionHandler} from '../lib/worker/completion-handlers.js';
+export {registerCompletionHandler} from '../lib/worker/completion-handlers.cjs';

lib/worker/base.js

@@ -15,7 +15,7 @@ import Runner from '../runner.js';
 import serializeError from '../serialize-error.js';
 
 import channel from './channel.cjs';
-import {runCompletionHandlers} from './completion-handlers.js';
+import {runCompletionHandlers} from './completion-handlers.cjs';
 import lineNumberSelection from './line-numbers.js';
 import {set as setOptions} from './options.cjs';
 import {flags, refs, sharedWorkerTeardowns} from './state.cjs';

lib/worker/completion-handlers.cjs

@@ -0,0 +1,15 @@
+'use strict';
+
+const process = require('node:process');
+
+const state = require('./state.cjs');
+
+exports.runCompletionHandlers = () => {
+	for (const handler of state.completionHandlers) {
+		process.nextTick(() => handler());
+	}
+};
+
+exports.registerCompletionHandler = handler => {
+	state.completionHandlers.push(handler);
+};