Class MockWebSocket

test/unit/synchronizers/synchronizer-ws-server.test.ts:18–47

const {createWsSynchronizer} = WsClient;

class MockWebSocket {
  OPEN = 1;
  readyState = this.OPEN;
  sentPayloads: string[] = [];
  readonly #listeners: {[event: string]: ((event: any) => void)[]} = {};

  addEventListener(event: string, listener: (event: any) => void): void {
    (this.#listeners[event] ??= []).push(listener);
  }

  removeEventListener(event: string, listener: (event: any) => void): void {
    this.#listeners[event] = (this.#listeners[event] ?? []).filter(
      (testListener) => testListener != listener,
    );
  }

  send(payload: string): void {
    this.sentPayloads.push(payload);
  }

  receive(payload: string): void {
    (this.#listeners.message ?? []).forEach((listener) =>
      listener({data: payload}),
    );
  }

  close(): void {
    this.readyState = 3;
  }
}

const getFragmentGroup = (
  payloads: string[],