Stanko K.R.
29 lines
748 B
Ruby
29 lines
748 B
Ruby
require "test_helper"
|
|
|
|
class Ai::ListCollectionsToolTest < ActiveSupport::TestCase
|
|
include McpHelper
|
|
|
|
setup do
|
|
@tool = Ai::ListCollectionsTool.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 filtering by ids" do
|
|
collections = collections(:writebook, :private)
|
|
collection_ids = collections.pluck(:id)
|
|
|
|
response = @tool.execute(ids: collection_ids.join(", "))
|
|
page = parse_paginated_response(response)
|
|
record_ids = page[:records].map { |collection| collection["id"].to_i }
|
|
|
|
assert_equal 2, record_ids.count
|
|
assert_equal collection_ids.sort, record_ids.sort
|
|
end
|
|
end
|