import { asyncExitHook, gracefulExit } from 'exit-hook'
import logger from '@wdio/logger'
import Runner from '@wdio/runner'
import type { Workers } from '@wdio/types'
import { SHUTDOWN_TIMEOUT } from './constants.js'
const log = logger('@wdio/local-runner')
* ToDo(Christian): remove when @wdio/runner got typed
*/
interface RunnerInterface extends NodeJS.EventEmitter {
sigintWasCalled: boolean
[key: string]: unknown
}
* send ready event to testrunner to start receive command messages
*/
if (typeof process.send === 'function') {
process.send(<Workers.WorkerMessage>{
name: 'ready',
origin: 'worker'
})
}
export const runner = new Runner() as unknown as RunnerInterface
runner.on('exit', (code = 0) => gracefulExit(code))
runner.on('error', ({ name, message, stack }) => process.send!({
origin: 'worker',
name: 'error',
content: { name, message, stack }
}))
process.on('message', (m: Workers.WorkerCommand) => {
if (!m || !m.command || !runner[m.command] || typeof runner[m.command] !== 'function') {
return
}
log.info(`Run worker command: ${m.command}`)
;(runner[m.command] as Function)(m).then(
(result: unknown) => process.send!({
origin: 'worker',
name: 'finishedCommand',
content: {
command: m.command,
result
}
}),
(e: Error) => {
log.error(`Failed launching test session: ${e.stack}`)
setTimeout(() => process.exit(1), 10)
}
)
})
process.once('SIGINT', () => {
runner.sigintWasCalled = true
gracefulExit(130)
})
* registers an asynchronous exit hook to allow graceful shutdown
* if the process is being terminated due to SIGINT, delays exit for SHUTDOWN_TIMEOUT ms to allow cleanup
* otherwise, exits immediately without delay
*/
asyncExitHook(
async () => {
if (runner.sigintWasCalled) {
log.info(`Received SIGINT, giving process ${SHUTDOWN_TIMEOUT}ms to shutdown gracefully`)
return await new Promise(res => setTimeout(res, SHUTDOWN_TIMEOUT))
}
}, {
wait: SHUTDOWN_TIMEOUT
}
)
