refactor iqdb endpoints

test/functional/iqdb_queries_controller_test.rb

@@ -3,6 +3,7 @@ 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)
@@ -10,10 +11,52 @@ class IqdbQueriesControllerTest < ActionDispatch::IntegrationTest
     end
 
     context "show action" do
-      should "render with matches" do
-        json = @posts.map {|x| [x.id, 1]}.to_json
-        get_auth iqdb_queries_path, @user, params: { matches: json }
-        assert_response :success
+      context "with a url parameter" do
+        setup do
+          @url = "https://google.com"
+          @params = { url: @url }
+          @mocked_response = [{
+            "post" => @posts[0],
+            "id" => @posts[0].id,
+            "score" => 1
+          }]
+        end
+
+        should "render a response" do
+          IqdbProxy.expects(:query).with(@url).returns(@mocked_response)
+          get_auth iqdb_queries_path(variant: "xhr"), @user, params: @params
+          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],
+            "id" => @posts[0].id,
+            "score" => 1
+          }]
+        end
+
+        should "redirect to iqdbs" do
+          IqdbProxy.expects(:query).with(@posts[0].preview_file_url).returns(@mocked_response)
+          get_auth iqdb_queries_path, @user, params: @params
+          assert_select("#post_#{@posts[0].id}")
+        end
+      end
+
+      context "with matches" do
+        setup do
+          json = @posts.map {|x| {"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