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

import { PromptQueue } from "./prompt-queue"

const prompt = (content: string) => ({ content, options: {} })

describe("PromptQueue", () => {
  test("returns the latest follow-up for editing", () => {
    const queue = new PromptQueue()
    queue.enqueue(prompt("next one"))
    queue.enqueue(prompt("steer now"))

    expect(queue.takeLast()?.content).toBe("steer now")
    expect(queue.snapshot().map(({ content }) => content)).toEqual(["next one"])
    expect(queue.takeFollowUp()?.content).toBe("next one")
  })

  test("keeps follow-ups FIFO and restores unsent drafts", () => {
    const queue = new PromptQueue()
    queue.enqueue(prompt("first"))
    queue.enqueue(prompt("second"))

    expect(queue.takeFollowUp()?.content).toBe("first")
    expect(queue.restore().map(({ content }) => content)).toEqual(["second"])
    expect(queue.length).toBe(0)
  })
})