import { describe, expect, test } from "vitest";
import type { UIMessage } from "ai";

import { shouldAutomaticallyContinueAssistant } from "./assistant-auto-submit";

const messagesWithToolPart = (part: UIMessage["parts"][number]): UIMessage[] => [
  {
    id: "assistant-message",
    role: "assistant",
    parts: [{ type: "step-start" }, part],
  },
];

describe("shouldAutomaticallyContinueAssistant", () => {
  test("continues after a human or frontend tool supplies its result", () => {
    expect(
      shouldAutomaticallyContinueAssistant({
        messages: messagesWithToolPart({
          type: "dynamic-tool",
          toolName: "ask_user",
          toolCallId: "ask-1",
          input: { question: "Continue?" },
          state: "output-available",
          output: { answer: "Yes" },
        }),
      }),
    ).toBe(true);
  });

  test("continues after a mutation approval response", () => {
    expect(
      shouldAutomaticallyContinueAssistant({
        messages: messagesWithToolPart({
          type: "dynamic-tool",
          toolName: "app_create",
          toolCallId: "create-1",
          input: { name: "YouTube" },
          state: "approval-responded",
          approval: { id: "approval-1", approved: true },
        }),
      }),
    ).toBe(true);
  });

  test("waits while a human tool has no result", () => {
    expect(
      shouldAutomaticallyContinueAssistant({
        messages: messagesWithToolPart({
          type: "dynamic-tool",
          toolName: "ask_user",
          toolCallId: "ask-1",
          input: { question: "Continue?" },
          state: "input-available",
        }),
      }),
    ).toBe(false);
  });
});