fa37b1edcd
* Change cutoffs on upload page to max 5 results, min. 20% similarity. * Change cutoffs on standalone /iqdb_queries page to max 20 results, min. 0% similarity. * /iqdb_queries.json: add `limit` and `similarity` params to change default cutoffs.
68 lines
1.8 KiB
Ruby
68 lines
1.8 KiB
Ruby
require 'test_helper'
|
|
|
|
class IqdbQueriesControllerTest < ActionDispatch::IntegrationTest
|
|
context "The iqdb controller" do
|
|
setup do
|
|
Danbooru.config.stubs(:iqdbs_server).returns("https://karasuma.donmai.us")
|
|
@user = create(:user)
|
|
as_user do
|
|
@posts = FactoryBot.create_list(:post, 2)
|
|
end
|
|
end
|
|
|
|
context "show action" do
|
|
context "with a url parameter" do
|
|
setup do
|
|
@url = "https://google.com"
|
|
@params = { url: @url }
|
|
@mocked_response = [{
|
|
"post" => @posts[0],
|
|
"post_id" => @posts[0].id,
|
|
"score" => 1
|
|
}]
|
|
end
|
|
|
|
should "render a response" do
|
|
IqdbProxy.expects(:query).returns(@mocked_response)
|
|
get_auth iqdb_queries_path, @user, as: :javascript, params: @params
|
|
|
|
assert_response :success
|
|
assert_select("#post_#{@posts[0].id}")
|
|
end
|
|
end
|
|
|
|
context "with a post_id parameter" do
|
|
setup do
|
|
@params = { post_id: @posts[0].id }
|
|
@url = @posts[0].preview_file_url
|
|
@mocked_response = [{
|
|
"post" => @posts[0],
|
|
"post_id" => @posts[0].id,
|
|
"score" => 1
|
|
}]
|
|
end
|
|
|
|
should "redirect to iqdbs" do
|
|
IqdbProxy.expects(:query).returns(@mocked_response)
|
|
get_auth iqdb_queries_path, @user, params: @params
|
|
|
|
assert_response :success
|
|
assert_select("#post_#{@posts[0].id}")
|
|
end
|
|
end
|
|
|
|
context "with matches" do
|
|
setup do
|
|
json = @posts.map {|x| {"post_id" => x.id, "score" => 1}}.to_json
|
|
@params = { matches: json }
|
|
end
|
|
|
|
should "render with matches" do
|
|
get_auth iqdb_queries_path, @user, params: @params
|
|
assert_response :success
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|