import { describe, expect, it } from "vitest";
import {
  assertCronJobMatches,
  assertLiveImageProbeReply,
  buildLiveCronProbeMessage,
  createLiveCronProbeSpec,
  normalizeLiveAgentFamily,
} from "./live-agent-probes.js";

describe("live-agent-probes", () => {
  it("normalizes cli backend ids into live agent families", () => {
    expect(normalizeLiveAgentFamily("claude-cli")).toBe("claude");
    expect(normalizeLiveAgentFamily("codex")).toBe("codex");
    expect(normalizeLiveAgentFamily("google-gemini-cli")).toBe("gemini");
  });

  it("accepts only cat for the shared image probe reply", () => {
    expect(() => assertLiveImageProbeReply("cat")).not.toThrow();
    expect(() => assertLiveImageProbeReply("horse")).toThrow("image probe expected 'cat'");
  });

  it("builds a retryable cron prompt with provider-specific fallback wording", () => {
    const spec = createLiveCronProbeSpec();
    expect(
      buildLiveCronProbeMessage({
        agent: "claude-cli",
        argsJson: spec.argsJson,
        attempt: 1,
        exactReply: spec.name,
      }),
    ).toContain("Return only a tool call");
    expect(
      buildLiveCronProbeMessage({
        agent: "codex",
        argsJson: spec.argsJson,
        attempt: 1,
        exactReply: spec.name,
      }),
    ).toContain("No prose before the tool call");
  });

  it("validates cron cli job shape for the shared live probe", () => {
    expect(() =>
      assertCronJobMatches({
        job: {
          name: "live-mcp-abc",
          sessionTarget: "session:agent:dev:test",
          agentId: "dev",
          sessionKey: "agent:dev:test",
          payload: { kind: "agentTurn", message: "probe-abc" },
        },
        expectedName: "live-mcp-abc",
        expectedMessage: "probe-abc",
        expectedSessionKey: "agent:dev:test",
      }),
    ).not.toThrow();
  });
});