Function createMockWSServer

e2e/vscode-connect.spec.ts:25–97

启动一个临时 WebSocket 服务器，返回 URL 和清理函数

()

/** 启动一个临时 WebSocket 服务器，返回 URL 和清理函数 */
function createMockWSServer(): Promise<{
  url: string;
  connections: WebSocket[];
  close: () => Promise<void>;
  /** 向所有已连接客户端发送消息 */
  broadcast: (data: unknown) => void;
  /** 等待收到指定 action 的消息 */
  waitForAction: (action: string, timeout?: number) => Promise<unknown>;
}> {
  return new Promise((resolve, reject) => {
    const connections: WebSocket[] = [];
    const messageListeners: Array<(msg: unknown) => void> = [];

    const wss = new WebSocketServer({ host: "127.0.0.1", port: 0 }, () => {
      const addr = wss.address();
      if (typeof addr === "string") {
        reject(new Error("Unexpected address type"));
        return;
      }
      const url = `ws://127.0.0.1:${addr.port}`;

      wss.on("connection", (ws) => {
        connections.push(ws);
        ws.on("message", (raw) => {
          try {
            const msg = JSON.parse(raw.toString());
            for (const listener of messageListeners) {
              listener(msg);
            }
          } catch {
            // 忽略非 JSON 消息
          }
        });
      });

      resolve({
        url,
        connections,
        close: () =>
          new Promise<void>((res) => {
            for (const ws of connections) ws.close();
            wss.close(() => res());
          }),
        broadcast: (data: unknown) => {
          const payload = JSON.stringify(data);
          for (const ws of connections) {
            if (ws.readyState === ws.OPEN) {
              ws.send(payload);
            }
          }
        },
        waitForAction: (action: string, timeout = 10_000) =>
          new Promise<unknown>((resolve, reject) => {
            const timer = setTimeout(() => {
              const idx = messageListeners.indexOf(handler);
              if (idx >= 0) messageListeners.splice(idx, 1);
              reject(new Error(`Timeout waiting for action: ${action}`));
            }, timeout);