jest-wake-lock-mock/src/WakeLock.ts

Mock Screen Wake Lock API `navigator.wakeLock` with ease and run your tests using Jest

const RELEASE_EVENT_NAME = "release";
type ReleaseEventName = typeof RELEASE_EVENT_NAME;

/**
 * Create a `navigator.wakeLock.request`
 * @class
 * @classdesc Aims to reproduce Screen Wake Lock API in Jest environment
 * @link https://w3c.github.io/screen-wake-lock/
 */
export class WakeLockSentinel {
  #released: boolean;
  #type: WakeLockType;
  #onrelease: EventListener | undefined;
  #releaseEvent: Event;
  #wakeLockEventTarget: EventTarget;

  constructor(type: WakeLockType) {
    this.#released = false;
    this.#type = type;
    this.#onrelease = undefined;
    this.#releaseEvent = new Event(RELEASE_EVENT_NAME);
    this.#wakeLockEventTarget = new EventTarget();
  }

  addEventListener(eventName: ReleaseEventName, listener: EventListenerOrEventListenerObject | null, options?: boolean | AddEventListenerOptions) {
    this.#wakeLockEventTarget.addEventListener(eventName, listener, options);
  }

  removeEventListener(eventName: ReleaseEventName, callback: EventListenerOrEventListenerObject | null, options?: EventListenerOptions | boolean) {
    this.#wakeLockEventTarget.removeEventListener(eventName, callback, options);
  }

  async release() {
    this.#released = true;
    this.#wakeLockEventTarget.dispatchEvent(this.#releaseEvent);
  }

  set onrelease(listener: EventListener) {
    if (this.#onrelease) this.removeEventListener('release', this.#onrelease);
    this.addEventListener('release', listener);
    this.#onrelease = listener;
  }

  get released() {
    return this.#released;
  }

  get type() {
    return this.#type;
  }
}