Jorge Manrubia
86 lines
3.4 KiB
Ruby
86 lines
3.4 KiB
Ruby
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
|