Function waitForWebSocketRejection

src/node/orpc/server.test.ts:56–89

(ws: WebSocket)

}

async function waitForWebSocketRejection(ws: WebSocket): Promise<void> {
  return new Promise<void>((resolve, reject) => {
    const timeout = setTimeout(() => {
      cleanup();
      reject(new Error("Expected WebSocket handshake to be rejected"));
    }, 5_000);

    const onError = () => {
      cleanup();
      resolve();
    };

    const onClose = () => {
      cleanup();
      resolve();
    };

    const onOpen = () => {
      cleanup();
      reject(new Error("Expected WebSocket handshake to be rejected"));
    };

    const cleanup = () => {
      clearTimeout(timeout);
      ws.off("error", onError);
      ws.off("close", onClose);
      ws.off("open", onOpen);
    };

    ws.once("error", onError);
    ws.once("close", onClose);
    ws.once("open", onOpen);
  });
}

async function closeWebSocket(ws: WebSocket): Promise<void> {
  if (ws.readyState === WebSocket.CLOSED) {