paseo/packages/app/src/runtime/websocket-transport.browser.test.ts

import { inject } from "vitest";
import { describe, expect, test } from "vitest";
import { createAppWebSocketFactory } from "./websocket-factory";

declare module "vitest" {
  interface ProvidedContext {
    passwordlessRelayUrl: string;
  }
}

function openPasswordlessRelayConnection(url: string): Promise<WebSocket> {
  const socket = createAppWebSocketFactory()(url, {
    headers: {},
    protocols: undefined,
  });

  return new Promise((resolve, reject) => {
    const timeout = setTimeout(() => {
      socket.close();
      reject(new Error("Timed out waiting for the passwordless relay connection"));
    }, 2_000);

    socket.addEventListener?.("open", () => {
      clearTimeout(timeout);
      resolve(socket as WebSocket);
    });
    socket.addEventListener?.("error", () => {
      clearTimeout(timeout);
      reject(new Error("Passwordless relay connection failed before open"));
    });
  });
}

describe("browser WebSocket transport", () => {
  test("opens a passwordless relay-style connection when the server selects no subprotocol", async () => {
    const socket = await openPasswordlessRelayConnection(inject("passwordlessRelayUrl"));

    try {
      expect(socket.protocol).toBe("");
    } finally {
      socket.close();
    }
  });
});