Stanko K.R.
95 lines
3.0 KiB
Ruby
95 lines
3.0 KiB
Ruby
require "test_helper"
|
|
|
|
class Ai::ListCommentsToolTest < ActiveSupport::TestCase
|
|
include McpHelper
|
|
|
|
setup do
|
|
@tool = Ai::ListCommentsTool.new(user: users(:kevin))
|
|
end
|
|
|
|
test "execute" do
|
|
response = @tool.execute
|
|
page = parse_paginated_response(response)
|
|
|
|
assert page[:records].is_a?(Array)
|
|
end
|
|
|
|
test "execute when ordering the result" do
|
|
response = @tool.execute(ordered_by: "id ASC")
|
|
page = parse_paginated_response(response)
|
|
ids = page[:records].map { |comment| comment["id"] }
|
|
|
|
assert_equal ids.sort, ids, "The IDs are sorted in ascending order"
|
|
|
|
response = @tool.execute(ordered_by: "id DESC")
|
|
page = parse_paginated_response(response)
|
|
ids = page[:records].map { |comment| comment["id"] }
|
|
|
|
assert_equal ids.sort.reverse, ids, "The IDs are sorted in descending order"
|
|
|
|
assert_raises(ArgumentError) do
|
|
@tool.execute(ordered_by: "created_at foobar")
|
|
end
|
|
end
|
|
|
|
test "execute when filtering by ids" do
|
|
comments = comments(:logo_1, :logo_3)
|
|
comment_ids = comments.pluck(:id)
|
|
|
|
response = @tool.execute(ids: comment_ids.join(", "))
|
|
page = parse_paginated_response(response)
|
|
record_ids = page[:records].map { |comment| comment["id"].to_i }
|
|
|
|
assert_equal 2, record_ids.count
|
|
assert_equal comment_ids.sort, record_ids.sort
|
|
end
|
|
|
|
test "execute when filtering by card_ids" do
|
|
card = cards(:logo)
|
|
|
|
response = @tool.execute(card_ids: card.id.to_s)
|
|
page = parse_paginated_response(response)
|
|
|
|
assert page[:records].all? { |comment| card.id == comment["card_id"] }
|
|
end
|
|
|
|
test "execute when filtering by type" do
|
|
response = @tool.execute(type: "system")
|
|
page = parse_paginated_response(response)
|
|
|
|
assert page[:records].all? { |comment| comment["system"] == true }
|
|
|
|
response = @tool.execute(type: "user")
|
|
page = parse_paginated_response(response)
|
|
|
|
assert page[:records].all? { |comment| comment["system"] == false }
|
|
end
|
|
|
|
test "execute when filtering by created_at" do
|
|
response = @tool.execute(created_after: 8.days.ago.to_s)
|
|
page = parse_paginated_response(response)
|
|
|
|
assert_not_empty page[:records], "There are comments created in the last 8 days"
|
|
|
|
response = @tool.execute(created_after: 3.days.ago.to_s)
|
|
page = parse_paginated_response(response)
|
|
|
|
assert_not_empty page[:records], "There are comments created in the last 3 days"
|
|
|
|
response = @tool.execute(created_before: 3.days.ago.to_s)
|
|
page = parse_paginated_response(response)
|
|
|
|
assert_not_empty page[:records], "There are comments created more than 3 days ago"
|
|
|
|
response = @tool.execute(created_before: 8.days.ago.to_s)
|
|
page = parse_paginated_response(response)
|
|
|
|
assert_empty page[:records], "There are no comments created more than 8 days ago"
|
|
|
|
response = @tool.execute(created_before: 3.days.ago.to_s, created_after: 8.days.ago.to_s)
|
|
page = parse_paginated_response(response)
|
|
|
|
assert_not_empty page[:records], "There are comments created between 3 and 8 days ago"
|
|
end
|
|
end
|