BrowserSkill/apps/extension/src/lib/__tests__/help-bridge.test.ts

import { describe, expect, it } from "vitest";
import {
  HELP_ACK,
  HELP_CANCEL,
  HELP_FINISH,
  HELP_QUERY,
  HELP_REQUEST,
  HELP_RESPONSE,
  isHelpAckMessage,
  isHelpCancelMessage,
  isHelpFinishMessage,
  isHelpQueryMessage,
  isHelpRequestMessage,
  isHelpResponseMessage,
} from "../help-bridge";

describe("help-bridge", () => {
  it("exposes stable message type constants", () => {
    expect(HELP_REQUEST).toBe("bsk-help-request");
    expect(HELP_RESPONSE).toBe("bsk-help-response");
    expect(HELP_CANCEL).toBe("bsk-help-cancel");
    expect(HELP_ACK).toBe("bsk-help-ack");
    expect(HELP_QUERY).toBe("bsk-help-query");
    expect(HELP_FINISH).toBe("bsk-help-finish");
  });

  it("type-guards a help-request message", () => {
    expect(
      isHelpRequestMessage({
        type: HELP_REQUEST,
        requestId: "r1",
        prompt: "log in",
        displayMode: "compact",
        selectors: ["#login"],
        rects: [{ top: 10, left: 20, width: 100, height: 40 }],
        timeoutMs: 1000,
      }),
    ).toBe(true);
    expect(isHelpRequestMessage({ type: "borrow-request" })).toBe(false);
    expect(isHelpRequestMessage(null)).toBe(false);
  });

  it("rejects a help-request message with missing/wrong fields", () => {
    expect(isHelpRequestMessage({ type: HELP_REQUEST })).toBe(false);
    expect(
      isHelpRequestMessage({
        type: HELP_REQUEST,
        requestId: "r1",
        prompt: "p",
        selectors: "nope",
        timeoutMs: 1,
      }),
    ).toBe(false);
    expect(
      isHelpRequestMessage({
        type: HELP_REQUEST,
        requestId: "r1",
        prompt: "p",
        selectors: [123],
        timeoutMs: 1,
      }),
    ).toBe(false);
  });

  it("type-guards a help-cancel message", () => {
    expect(isHelpCancelMessage({ type: HELP_CANCEL, requestId: "r1" })).toBe(true);
    expect(isHelpCancelMessage({ type: HELP_REQUEST, requestId: "r1" })).toBe(false);
    expect(isHelpCancelMessage(null)).toBe(false);
  });

  it("type-guards lifecycle messages", () => {
    expect(isHelpAckMessage({ type: HELP_ACK, ok: true })).toBe(true);
    expect(isHelpQueryMessage({ type: HELP_QUERY })).toBe(true);
    expect(isHelpFinishMessage({ type: HELP_FINISH, requestId: "r1", outcome: "continued" })).toBe(
      true,
    );
    expect(isHelpResponseMessage({ type: HELP_RESPONSE, outcome: "cancelled", note: "no" })).toBe(
      true,
    );
    expect(isHelpFinishMessage({ type: HELP_FINISH, requestId: "r1", outcome: "weird" })).toBe(
      false,
    );
  });
});