@flrande/browserctl 0.5.0-dev.22.1 → 0.6.0

package/apps/browserd/src/chrome-relay-extension-bridge.test.ts

@@ -1,250 +0,0 @@
-import { afterEach, describe, expect, it } from "vitest";
-import { WebSocket } from "ws";
-
-import { createChromeRelayExtensionBridge } from "./chrome-relay-extension-bridge";
-import { reserveLoopbackPort } from "./test-port";
-
-async function waitForCondition(predicate: () => boolean, timeoutMs = 1000): Promise<void> {
-  const start = Date.now();
-  while (!predicate()) {
-    if (Date.now() - start > timeoutMs) {
-      throw new Error("Timed out waiting for condition.");
-    }
-
-    await new Promise<void>((resolve) => {
-      setTimeout(resolve, 10);
-    });
-  }
-}
-
-type Closable = {
-  close(): Promise<void> | void;
-};
-
-const closables: Closable[] = [];
-
-afterEach(async () => {
-  while (closables.length > 0) {
-    const item = closables.pop();
-    if (item === undefined) {
-      continue;
-    }
-
-    await item.close();
-  }
-});
-
-describe("createChromeRelayExtensionBridge", () => {
-  it("requires token configuration for extension bridge startup", async () => {
-    const port = await reserveLoopbackPort();
-
-    expect(() =>
-      createChromeRelayExtensionBridge({
-        relayUrl: `http://127.0.0.1:${port}`
-      })
-    ).toThrow("BROWSERD_CHROME_RELAY_EXTENSION_TOKEN");
-  });
-
-  it("sends requests to extension client and receives responses", async () => {
-    const port = await reserveLoopbackPort();
-    const bridge = createChromeRelayExtensionBridge({
-      relayUrl: `http://127.0.0.1:${port}`,
-      token: "secret-token",
-      requestTimeoutMs: 500
-    });
-    closables.push(bridge);
-
-    const socket = new WebSocket(`ws://127.0.0.1:${port}/bridge`);
-    closables.push({
-      async close() {
-        if (socket.readyState === WebSocket.OPEN || socket.readyState === WebSocket.CONNECTING) {
-          await new Promise<void>((resolve) => {
-            socket.once("close", () => resolve());
-            socket.close();
-          });
-        }
-      }
-    });
-
-    await new Promise<void>((resolve, reject) => {
-      socket.once("open", () => {
-        socket.send(
-          JSON.stringify({
-            type: "hello",
-            role: "extension",
-            extensionId: "ext:test",
-            token: "secret-token"
-          })
-        );
-        resolve();
-      });
-      socket.once("error", reject);
-    });
-    await waitForCondition(() => bridge.isConnected());
-
-    socket.on("message", (message) => {
-      const parsed = JSON.parse(String(message)) as Record<string, unknown>;
-      if (parsed.type !== "request") {
-        return;
-      }
-
-      socket.send(
-        JSON.stringify({
-          type: "response",
-          id: parsed.id,
-          ok: true,
-          result: {
-            accepted: true,
-            method: parsed.method
-          }
-        })
-      );
-    });
-
-    await expect(bridge.invoke("tab.open", { url: "https://example.com" })).resolves.toEqual({
-      accepted: true,
-      method: "tab.open"
-    });
-    expect(bridge.isConnected()).toBe(true);
-  });
-
-  it("rejects invoke when extension is not connected", async () => {
-    const port = await reserveLoopbackPort();
-    const bridge = createChromeRelayExtensionBridge({
-      relayUrl: `http://127.0.0.1:${port}`,
-      token: "secret-token",
-      requestTimeoutMs: 200
-    });
-    closables.push(bridge);
-
-    await expect(bridge.invoke("tab.info", { tabId: 1 })).rejects.toThrowError(
-      "Chrome relay extension is not connected."
-    );
-  });
-
-  it("enforces token validation for extension websocket connection", async () => {
-    const port = await reserveLoopbackPort();
-    const bridge = createChromeRelayExtensionBridge({
-      relayUrl: `http://127.0.0.1:${port}`,
-      token: "secret-token",
-      requestTimeoutMs: 500
-    });
-    closables.push(bridge);
-
-    const socket = new WebSocket(`ws://127.0.0.1:${port}/bridge`);
-    closables.push({
-      async close() {
-        if (socket.readyState === WebSocket.OPEN || socket.readyState === WebSocket.CONNECTING) {
-          await new Promise<void>((resolve) => {
-            socket.once("close", () => resolve());
-            socket.close();
-          });
-        }
-      }
-    });
-
-    await new Promise<void>((resolve, reject) => {
-      socket.once("open", () => {
-        socket.send(
-          JSON.stringify({
-            type: "hello",
-            role: "extension",
-            extensionId: "ext:test",
-            token: "wrong-token"
-          })
-        );
-      });
-      socket.once("close", () => resolve());
-      socket.once("error", reject);
-    });
-
-    expect(socket.readyState).not.toBe(WebSocket.OPEN);
-    expect(bridge.isConnected()).toBe(false);
-  });
-
-  it("forwards extension event envelopes to bridge listeners", async () => {
-    const port = await reserveLoopbackPort();
-    const bridge = createChromeRelayExtensionBridge({
-      relayUrl: `http://127.0.0.1:${port}`,
-      token: "secret-token",
-      requestTimeoutMs: 500
-    });
-    closables.push(bridge);
-
-    const socket = new WebSocket(`ws://127.0.0.1:${port}/bridge`);
-    closables.push({
-      async close() {
-        if (socket.readyState === WebSocket.OPEN || socket.readyState === WebSocket.CONNECTING) {
-          await new Promise<void>((resolve) => {
-            socket.once("close", () => resolve());
-            socket.close();
-          });
-        }
-      }
-    });
-
-    await new Promise<void>((resolve, reject) => {
-      socket.once("open", () => {
-        socket.send(
-          JSON.stringify({
-            type: "hello",
-            role: "extension",
-            extensionId: "ext:test",
-            token: "secret-token"
-          })
-        );
-        resolve();
-      });
-      socket.once("error", reject);
-    });
-    await waitForCondition(() => bridge.isConnected());
-
-    const receivedEvents: Array<Record<string, unknown>> = [];
-    const detach = bridge.onEvent((event) => {
-      receivedEvents.push(event as unknown as Record<string, unknown>);
-    });
-    closables.push({
-      close() {
-        detach();
-      }
-    });
-
-    socket.send(
-      JSON.stringify({
-        type: "event",
-        event: {
-          kind: "console",
-          tabId: 7,
-          entry: {
-            type: "log",
-            text: "hello-event"
-          }
-        }
-      })
-    );
-
-    await new Promise<void>((resolve) => {
-      const start = Date.now();
-      const poll = () => {
-        if (receivedEvents.length >= 1 || Date.now() - start > 1000) {
-          resolve();
-          return;
-        }
-
-        setTimeout(poll, 10);
-      };
-
-      poll();
-    });
-
-    expect(receivedEvents).toHaveLength(1);
-    expect(receivedEvents[0]).toMatchObject({
-      kind: "console",
-      tabId: 7,
-      entry: {
-        type: "log",
-        text: "hello-event"
-      }
-    });
-  });
-});