Class MockWebSocket

packages/client/test/mock-ws.ts:10–67

type Listener = (ev: { data?: string | undefined; code?: number | undefined; reason?: string | undefined }) => void;

export class MockWebSocket implements WsLike {
  static OPEN = 1;
  static CLOSED = 3;

  readyState: number = 0;
  sent: string[] = [];
  url: string;
  private listeners: Record<EventType, Listener[]> = {
    open: [],
    close: [],
    error: [],
    message: [],
  };

  constructor(url: string) {
    this.url = url;
  }

  addEventListener(type: EventType, fn: Listener): void {
    this.listeners[type].push(fn);
  }

  removeEventListener(type: EventType, fn: Listener): void {
    this.listeners[type] = this.listeners[type].filter((l) => l !== fn);
  }

  send(data: string): void {
    this.sent.push(data);
  }

  close(code?: number, reason?: string): void {
    this.readyState = MockWebSocket.CLOSED;
    const ev: { code?: number; reason?: string } = {};
    if (code !== undefined) ev.code = code;
    if (reason !== undefined) ev.reason = reason;
    this.dispatch('close', ev);
  }

  // Test helpers.
  acceptOpen(): void {
    this.readyState = MockWebSocket.OPEN;
    this.dispatch('open', {});
  }
  acceptMessage(data: string): void {
    this.dispatch('message', { data });
  }
  acceptError(): void {
    this.dispatch('error', {});
  }
  acceptClose(code = 1000, reason = ''): void {
    this.readyState = MockWebSocket.CLOSED;
    this.dispatch('close', { code, reason });
  }

  private dispatch(type: EventType, ev: { data?: string | undefined; code?: number | undefined; reason?: string | undefined }): void {
    for (const l of this.listeners[type].slice()) l(ev);
  }
}