Function openConn

packages/server/test/host-origin.e2e.test.ts:120–151

(url: string, opts?: ConnectOptions)

}

function openConn(url: string, opts?: ConnectOptions): Promise<Conn> {
  return new Promise((resolve, reject) => {
    // Offer the fixed bearer token so the M5.1 WS auth passes; the Host/Origin
    // cases that expect rejection use `expectRejected` (no token) instead.
    const protocols = [...(opts?.protocols ?? []), 'kimi-code.bearer.test-token'];
    const ws = new WebSocket(url, protocols, { headers: opts?.headers });
    const queue: WsFrame[] = [];
    const waiters: Array<(frame: WsFrame) => void> = [];
    let closedResolve: (v: { code: number; reason: string }) => void;
    const closed = new Promise<{ code: number; reason: string }>((res) => {
      closedResolve = res;
    });
    ws.on('message', (data) => {
      let parsed: WsFrame;
      try {
        parsed = JSON.parse(rawDataToString(data)) as WsFrame;
      } catch {
        return;
      }
      if (waiters.length > 0) {
        waiters.shift()?.(parsed);
      } else {
        queue.push(parsed);
      }
    });
    ws.on('close', (code, reason) => {
      closedResolve({ code, reason: String(reason) });
    });
    ws.once('open', () => resolve({ ws, queue, waiters, closed }));
    ws.once('error', (err) => reject(err));
  });
}

function receive(conn: Conn, timeoutMs: number): Promise<WsFrame> {
  return new Promise((resolve, reject) => {