Class TestBridge

vscode/src/webview/createVscodeOrpcLink.test.ts:7–40

import type { VscodeBridge } from "./vscodeBridge";

class TestBridge implements VscodeBridge {
  traceId = "test";
  startedAtMs = 0;

  readonly sent: WebviewToExtensionMessage[] = [];
  throwOnPostMessage = false;

  private readonly listeners = new Set<(data: unknown) => void>();

  postMessage(payload: WebviewToExtensionMessage): void {
    this.sent.push(payload);

    if (this.throwOnPostMessage) {
      throw new Error("postMessage failed");
    }
  }

  onMessage(handler: (data: unknown) => void): () => void {
    this.listeners.add(handler);
    return () => {
      this.listeners.delete(handler);
    };
  }

  debugLog(_message: string, _data?: unknown): void {
    // no-op for tests
  }

  emit(data: unknown): void {
    for (const listener of this.listeners) {
      listener(data);
    }
  }
}

function getLastSent(bridge: TestBridge, type: WebviewToExtensionMessage["type"]): WebviewToExtensionMessage {
  const msg = [...bridge.sent].reverse().find((m) => m.type === type);