From aa11c82e1c51debc21c72e40e7be533b7ac37fe4 Mon Sep 17 00:00:00 2001
From: "Stanko K.R." <stanko@stanko.io>
Date: Wed, 13 Aug 2025 13:43:32 +0200
Subject: [PATCH] Add test for the list comments tool

---
 test/models/ai/list_comments_tool_test.rb | 94 +++++++++++++++++++++++
 1 file changed, 94 insertions(+)
 create mode 100644 test/models/ai/list_comments_tool_test.rb

diff --git a/test/models/ai/list_comments_tool_test.rb b/test/models/ai/list_comments_tool_test.rb
new file mode 100644
index 000000000..f3a4556fe
--- /dev/null
+++ b/test/models/ai/list_comments_tool_test.rb
@@ -0,0 +1,94 @@
+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