Stanko K.R.
82 lines
3.2 KiB
Ruby
82 lines
3.2 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 "clearing conversation messages" do
|
|
conversation = conversations(:kevin)
|
|
|
|
assert conversation.messages.any?, "The conversation should have messages before clearing"
|
|
|
|
original_updated_at = conversation.updated_at
|
|
conversation.clear
|
|
assert conversation.updated_at > original_updated_at, "The conversation's updated_at timestamp should change after clearing messages"
|
|
|
|
assert conversation.messages.empty?, "All messages should be deleted when clearing the conversation"
|
|
end
|
|
|
|
test "cost calculation" do
|
|
conversation = conversations(:kevin)
|
|
|
|
assert_equal "0.01053".to_d, conversation.cost
|
|
end
|
|
end
|