Cancelable await wait()

#wait.ts

// https://gist.github.com/tkrotoff/c6dd1cabf5570906724097c6e3f65a12
// https://stackoverflow.com/a/67911932
export function wait(ms: number, options: { signal?: AbortSignal } = {}) {
  const { signal } = options;

  return new Promise<void>((resolve, reject) => {
    // FIXME Not supported by Jest 29.3.1 + Node.js 19.3.0
    //signal?.throwIfAborted();
    if (signal?.aborted) reject(signal.reason);

    const id = setTimeout(() => {
      resolve();
      // eslint-disable-next-line @typescript-eslint/no-use-before-define
      signal?.removeEventListener('abort', abort);
    }, ms);

    function abort() {
      clearTimeout(id);
      reject(signal!.reason);
    }

    signal?.addEventListener('abort', abort);
  });
}

wait.test.ts

import { assert } from './assert';
import { wait } from './wait';

test('wait without abort controller', async () => {
  const start = performance.now();

  await wait(10);

  const end = performance.now();
  expect(end - start).toBeGreaterThanOrEqual(9);
});

test('abort', () => {
  const start = performance.now();

  const waitController = new AbortController();

  // eslint-disable-next-line jest/valid-expect-in-promise
  wait(1000, { signal: waitController.signal })
    .then(() => {
      // This code must not be reached
      expect(true).toEqual(false);
    })
    .catch(e => {
      assert(e instanceof DOMException);
      expect(e).toBeInstanceOf(DOMException);
      expect(e.name).toEqual('AbortError');
    });

  waitController.abort();

  const end = performance.now();
  expect(end - start).toBeLessThanOrEqual(10);
});

test('reject if controller already aborted', async () => {
  const start = performance.now();

  const waitController = new AbortController();
  waitController.abort();

  try {
    await wait(1000, { signal: waitController.signal });
    // This code must not be reached
    expect(true).toEqual(false);
  } catch (e) {
    assert(e instanceof DOMException);
    expect(e).toBeInstanceOf(DOMException);
    expect(e.name).toEqual('AbortError');
  }

  waitController.abort();

  const end = performance.now();
  expect(end - start).toBeLessThanOrEqual(10);
});