jaculabilis/danbooru
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

searchable: fix bug with searching multiple association attributes.

Browse Source 
Fix a bug with searches like the following not working correctly:

* https://danbooru.donmai.us/comments.json?search[creator][level]=20&search[creator_id]=1234
* https://danbooru.donmai.us/comments.json?search[creator][level]=20&search[creator_name]=abcd
* https://danbooru.donmai.us/comments.json?search[post][rating]=s&search[post_tags_match]=touhou

It wasn't possible to search for both `creator` and `creator_id` at the
same time (or `post` and `post_tags_match`, etc). Only the `creator_id`
param would be recognized.

Also refactor some internals:

* `search_includes` was renamed to `search_associated_attribute`.
* `search_attribute` was split up into `search_basic_attribute` and
  `search_associated_attribute`.
This commit is contained in:
evazion
2021-01-07 15:59:56 -06:00
parent e356fd0fd5
commit da3e8e4726
2 changed files with 121 additions and 38 deletions
Download Patch File Download Diff File Expand all files Collapse all files

85
test/unit/concerns/searchable.rb
View File

@@ -1,8 +1,10 @@
require 'test_helper'
class SearchableTest < ActiveSupport::TestCase
def assert_search_equals(results, **params)
assert_equal(Array(results).map(&:id), subject.search(**params).ids)
def assert_search_equals(results, current_user: User.anonymous, **params)
as(current_user) do
assert_equal(Array(results).map(&:id), subject.search(**params).ids)
end
end
context "#search method" do
@@ -14,6 +16,14 @@ class SearchableTest < ActiveSupport::TestCase
@p3 = create(:post, source: "c3", score: 3, is_deleted: false)
end
context "for a nonexistent attribute" do
should "raise an error" do
assert_raises(ArgumentError) do
Post.search_attribute(:answer, 42, User.anonymous)
end
end
end
context "for a numeric attribute" do
should "support basic operators" do
assert_search_equals(@p1, score_eq: 1)
@@ -120,5 +130,76 @@ class SearchableTest < ActiveSupport::TestCase
assert_search_equals(@wp, other_name_count: 2)
end
end
context "for a belongs_to association" do
context "for a user association" do
should "work" do
assert_search_equals(@p1, uploader_id: @p1.uploader_id)
assert_search_equals(@p1, uploader_name: @p1.uploader.name)
assert_search_equals(@p1, uploader: { id: @p1.uploader_id })
assert_search_equals(@p1, uploader: { name: @p1.uploader.name })
assert_search_equals(@p1, uploader: { name: @p1.uploader.name }, uploader_id: @p1.uploader.id)
assert_search_equals([], uploader: { name: @p1.uploader.name }, uploader_id: @p2.uploader.id)
end
end
context "for a post association" do
should "work" do
@p1.update!(parent: @p2)
assert_search_equals(@p1, parent_id: @p2.id)
assert_search_equals(@p1, parent: { id: @p2.id })
assert_search_equals(@p1, parent_tags_match: "id:#{@p2.id}")
assert_search_equals([], parent_tags_match: "id:0")
assert_search_equals(@p2, children_tags_match: "id:#{@p1.id}")
assert_search_equals([], children_tags_match: "id:0")
assert_search_equals(@p1, has_parent: true)
assert_search_equals([@p3, @p2], has_parent: false)
end
end
context "for a polymorphic association" do
subject { ModerationReport }
should "work" do
as(create(:user)) do
@mr1 = create(:moderation_report, model: create(:comment))
@mr2 = create(:moderation_report, model: create(:forum_post))
@mr3 = create(:moderation_report, model: create(:dmail))
end
assert_search_equals(@mr1, model_type: "Comment")
assert_search_equals(@mr2, model_type: "ForumPost")
assert_search_equals(@mr3, model_type: "Dmail")
assert_search_equals(@mr1, model_type: "Comment", model_id: @mr1.model.id)
assert_search_equals(@mr2, model_type: "ForumPost", model_id: @mr2.model.id)
assert_search_equals(@mr3, model_type: "Dmail", model_id: @mr3.model.id)
assert_search_equals(@mr1, Comment: { body: @mr1.model.body })
assert_search_equals(@mr2, ForumPost: { body: @mr2.model.body })
assert_search_equals([], Dmail: { body: @mr3.model.body }, current_user: User.anonymous)
assert_search_equals(@mr3, Dmail: { body: @mr3.model.body }, current_user: @mr3.model.owner)
end
end
end
context "for a has_many association" do
should "work" do
as(@p1.uploader) { create(:comment, post: @p1) }
assert_search_equals(@p1, has_comments: true)
assert_search_equals([@p3, @p2], has_comments: false)
assert_search_equals(@p1, comments: { id: @p1.comments.first.id })
assert_search_equals(@p1, has_comments: true, comments: { id: @p1.comments.first.id })
end
end
end
end