createTimeoutController.ts

import { WaitOptions } from '../types'
import { jestFakeTimersAreEnabled } from './jestFakeTimersAreEnabled'
const DEFAULT_TIMEOUT = 1000

function createTimeoutController(
  timeout: WaitOptions['timeout'] = DEFAULT_TIMEOUT,
  allowFakeTimers: boolean
) {
  let timeoutId: NodeJS.Timeout
  const timeoutCallbacks: Array<() => void> = []
  let finished: boolean = false

  const advanceTime = async (currentMs: number) => {
    if (currentMs <= timeout) {
      jest.advanceTimersByTime(1)

      await Promise.resolve()

      if (finished) {
        return
      }
      // eslint-disable-next-line @typescript-eslint/no-unused-vars
      await advanceTime(currentMs + 1)
    }
  }
  const timeoutController = {
    onTimeout(callback: () => void) {
      timeoutCallbacks.push(callback)
    },
    wrap(promise: Promise<void>) {
      return new Promise<void>((resolve, reject) => {
        timeoutController.timedOut = false
        timeoutController.onTimeout(resolve)

        if (timeout) {
          timeoutId = setTimeout(() => {
            timeoutController.timedOut = true
            timeoutCallbacks.forEach((callback) => callback())
            resolve()
          }, timeout)

          if (jestFakeTimersAreEnabled() && allowFakeTimers) {
            advanceTime(0)
          }
        }

        promise
          .then(resolve)
          .catch(reject)
          .finally(() => {
            finished = true
            timeoutController.cancel()
          })
      })
    },
    cancel() {
      finished = true
      clearTimeout(timeoutId)
    },
    timedOut: false
  }

  return timeoutController
}

export { createTimeoutController, DEFAULT_TIMEOUT }