import { describe, expect, it } from "bun:test";

import {
  cancelBackendTodos,
  upsertBackendQuestion,
} from "../../src/routes/chatUiState.js";

describe("upsertBackendQuestion", () => {
  it("replaces a tool-call placeholder with the actionable question request", () => {
    const questions = upsertBackendQuestion(
      [
        {
          requestId: "call-1",
          sessionId: "session-1",
          questions: [
            {
              header: "测试问题",
              question: "你觉得这个 question 工具好用吗？",
              options: [{ label: "非常好用", description: "交互清晰，选项方便" }],
            },
          ],
          tool: { messageID: "message-1", callID: "call-1" },
          createdAt: 123,
          status: "pending",
        },
      ],
      {
        session_id: "session-1",
        request_id: "question-1",
        questions: [
          {
            header: "测试问题",
            question: "你觉得这个 question 工具好用吗？",
            options: [{ label: "非常好用", description: "交互清晰，选项方便" }],
          },
        ],
        tool: { messageID: "message-1", callID: "call-1" },
        created_at: 456,
      },
    );

    expect(questions).toHaveLength(1);
    expect(questions[0]).toMatchObject({
      requestId: "question-1",
      tool: { callID: "call-1" },
      status: "pending",
    });
  });

  it("does not replace an actionable question request with a later tool-call placeholder", () => {
    const questions = upsertBackendQuestion(
      [
        {
          requestId: "question-1",
          sessionId: "session-1",
          questions: [
            {
              header: "测试问题",
              question: "你觉得这个 question 工具好用吗？",
              options: [{ label: "非常好用", description: "交互清晰，选项方便" }],
            },
          ],
          tool: { messageID: "message-1", callID: "call-1" },
          createdAt: 123,
          status: "pending",
        },
      ],
      {
        session_id: "session-1",
        request_id: "call-1",
        questions: [
          {
            header: "测试问题",
            question: "你觉得这个 question 工具好用吗？",
            options: [{ label: "非常好用", description: "交互清晰，选项方便" }],
          },
        ],
        tool: { messageID: "message-1", callID: "call-1" },
        created_at: 456,
      },
    );

    expect(questions).toHaveLength(1);
    expect(questions[0]).toMatchObject({
      requestId: "question-1",
      tool: { callID: "call-1" },
      status: "pending",
    });
  });
});

describe("cancelBackendTodos", () => {
  it("marks pending and in-progress todos as cancelled", () => {
    const cancelled = cancelBackendTodos([
      {
        sessionId: "session-1",
        todos: [
          { id: "todo-1", content: "分析水位", status: "in_progress" },
          { id: "todo-2", content: "生成建议", status: "pending" },
          { id: "todo-3", content: "完成报告", status: "completed" },
        ],
        createdAt: 123,
      },
    ]);

    expect(cancelled).toEqual([
      expect.objectContaining({
        todos: [
          expect.objectContaining({
            id: "todo-1",
            status: "cancelled",
            updatedAt: expect.any(Number),
          }),
          expect.objectContaining({
            id: "todo-2",
            status: "cancelled",
            updatedAt: expect.any(Number),
          }),
          expect.objectContaining({
            id: "todo-3",
            status: "completed",
          }),
        ],
      }),
    ]);
  });
});