require "test_helper"

class ConversationTest < ActiveSupport::TestCase
  include VcrTestHelper

  test "asking questions" do
    conversation = users(:kevin).conversation

    # You can't respond to a conversation while it's in the thinking state
    assert_raises(Conversation::InvalidStateError) do
      conversation.respond("Ok")
    end

    assert_raises(ActiveRecord::RecordInvalid) do
      conversation.ask("")
    end

    conversation.reload

    assert conversation.ready?, "The conversation should be ready before a question is asked"

    message = nil
    assert_turbo_stream_broadcasts [ conversation.user, :conversation ], count: +2 do
      message = conversation.ask("What is the meaning of life, the Universe, and everything else?", client_message_id: "deep-thought")
    end

    assert_not_nil message, "A message should be created when a question is asked"
    assert message.persisted?, "The message should be saved to the database"
    assert_equal "What is the meaning of life, the Universe, and everything else?", message.content.to_plain_text.chomp, "The message content should match the question asked"
    assert message.user?, "The message role should be 'user' for a question"
    assert_equal "deep-thought", message.client_message_id, "Additional attributes should be set correctly"
    assert conversation.thinking?, "The conversation should switch to thinking after a question is asked"
  end

  test "responding to questions" do
    conversation = users(:david).conversation

    # You can't ask a question in a conversation that isn't ready
    assert_raises(Conversation::InvalidStateError) do
      conversation.ask("hi!")
    end

    assert_raises(ActiveRecord::RecordInvalid) do
      conversation.respond("")
    end

    conversation.reload

    assert conversation.thinking?, "The conversation should be thinking before a response is made"

    message = nil
    assert_turbo_stream_broadcasts [ conversation.user, :conversation ], count: +2 do
      message = conversation.respond("42", client_message_id: "deep-thought-response")
    end

    assert_not_nil message, "A message should be created when a response is made"
    assert message.persisted?, "The message should be saved to the database"
    assert_equal "42", message.content.to_plain_text.chomp, "The message content should match the response given"
    assert message.assistant?, "The message role should be 'assistant' for a response"
    assert_equal "deep-thought-response", message.client_message_id, "Asdditional attributes should be set correctly"
    assert conversation.ready?, "The conversation should switch back to ready after a response is made"
  end

  test "cost limits" do
    conversation = conversations(:kevin)

    conversation.ask("Where does the planning office keep demolition notices?")
    conversation.respond(
      "In a locked filing cabinet in a disused lavatory",
      cost_in_microcents: Ai::Quota::Money.wrap("$3").in_microcents
    )

    conversation.ask("What's the meaning of life?")
    conversation.respond("42", cost_in_microcents: Ai::Quota::Money.wrap("$120").in_microcents)

    assert_raises Ai::Quota::UsageExceedsQuotaError do
      conversation.ask("Should you leave a house without a towel?")
    end

    travel 1.month

    conversation.ask("Should you leave a house without a towel?")
    conversation.respond("Never", cost_in_microcents: Ai::Quota::Money.wrap("$0.01").in_microcents)
  end
end