(url: string)
| 105 | } |
| 106 | |
| 107 | function openConn(url: string): Promise<Conn> { |
| 108 | return new Promise((resolve, reject) => { |
| 109 | const ws = new WebSocket(url, ['kimi-code.bearer.test-token']); |
| 110 | const queue: WsFrame[] = []; |
| 111 | const waiters: Array<(frame: WsFrame) => void> = []; |
| 112 | ws.on('message', (data) => { |
| 113 | let parsed: WsFrame; |
| 114 | try { |
| 115 | parsed = JSON.parse(rawDataToString(data)) as WsFrame; |
| 116 | } catch { |
| 117 | return; |
| 118 | } |
| 119 | if (waiters.length > 0) { |
| 120 | const w = waiters.shift(); |
| 121 | w?.(parsed); |
| 122 | } else { |
| 123 | queue.push(parsed); |
| 124 | } |
| 125 | }); |
| 126 | ws.once('open', () => resolve({ ws, queue, waiters })); |
| 127 | ws.once('error', (err) => reject(err)); |
| 128 | }); |
| 129 | } |
| 130 | |
| 131 | function receive(conn: Conn, timeoutMs: number): Promise<WsFrame> { |
| 132 | return new Promise((resolve, reject) => { |
no test coverage detected