Function openSocket

packages/server/test/ws-terminal.e2e.test.ts:123–140

(r: RunningServer)

}

async function openSocket(r: RunningServer): Promise<{
  ws: WebSocket;
  received: Record<string, unknown>[];
}> {
  const received: Record<string, unknown>[] = [];
  const ws = await new Promise<WebSocket>((resolve, reject) => {
    const sock = new WebSocket(r.address.replace('http://', 'ws://') + '/api/v1/ws', [
      'kimi-code.bearer.test-token',
    ]);
    sock.on('message', (data) => {
      received.push(JSON.parse(rawDataToString(data)) as Record<string, unknown>);
    });
    sock.once('open', () => resolve(sock));
    sock.once('error', reject);
  });
  await waitFor(received, (frame) => frame['type'] === 'server_hello');
  return { ws, received };
}

async function waitFor(
  received: Record<string, unknown>[],