(url: string)
| 149 | } |
| 150 | |
| 151 | function openConn(url: string): Promise<Conn> { |
| 152 | return new Promise((resolve, reject) => { |
| 153 | const ws = new WebSocket(url, ['kimi-code.bearer.test-token']); |
| 154 | const queue: WsFrame[] = []; |
| 155 | const waiters: Array<(frame: WsFrame) => void> = []; |
| 156 | ws.on('message', (data) => { |
| 157 | let parsed: WsFrame; |
| 158 | try { |
| 159 | parsed = JSON.parse(rawDataToString(data)) as WsFrame; |
| 160 | } catch { |
| 161 | return; |
| 162 | } |
| 163 | if (waiters.length > 0) { |
| 164 | const w = waiters.shift(); |
| 165 | w?.(parsed); |
| 166 | } else { |
| 167 | queue.push(parsed); |
| 168 | } |
| 169 | }); |
| 170 | ws.once('open', () => resolve({ ws, queue, waiters })); |
| 171 | ws.once('error', (err) => reject(err)); |
| 172 | }); |
| 173 | } |
| 174 | |
| 175 | function receive(conn: Conn, timeoutMs: number): Promise<WsFrame> { |
| 176 | return new Promise((resolve, reject) => { |
no test coverage detected