Coder-Spirit / ts-insistent

'use strict'

// eslint-disable-next-line @typescript-eslint/no-var-requires

const bluebirdModule = require('bluebird')

const globalPromise = global.Promise

const globalSetTimeout = global.setTimeout

exports.__esModule = true

exports.mockSetTimeout = expectedPause => {

  global.Promise = bluebirdModule.Promise

  const mockState = {

    expectedPause: expectedPause,

    passedTimeout: null,

    numTimeoutCalls: 0,

    // eslint-disable-next-line @typescript-eslint/no-empty-function

    pendingCallback: () => {},

    pendingCallbackArgs: [],

  }

  global.setTimeout = (callback, ms, ...args) => {

    if (ms !== mockState.expectedPause) {

      // We use a "magic number" to allow all the other cases work as expected

      return globalSetTimeout(callback, ms, ...args)

    }

    mockState.passedTimeout = ms

    mockState.numTimeoutCalls += 1

    mockState.pendingCallback = callback

    mockState.pendingCallbackArgs = args

    return null

  }

  return mockState

}

exports.restoreSetTimeout = () => {

  global.setTimeout = globalSetTimeout

  global.Promise = globalPromise

}