33 lines
908 B
Ruby
33 lines
908 B
Ruby
module IqdbTestHelper
|
|
def mock_iqdb_service!
|
|
mock_sqs_service = Class.new do
|
|
def initialize
|
|
@commands = []
|
|
end
|
|
|
|
def commands
|
|
@commands
|
|
end
|
|
|
|
def send_message(msg)
|
|
@commands << msg.split(/\n/).first
|
|
end
|
|
end
|
|
|
|
service = mock_sqs_service.new
|
|
Post.stubs(:iqdb_sqs_service).returns(service)
|
|
Post.stubs(:iqdb_enabled?).returns(true)
|
|
|
|
Danbooru.config.stubs(:iqdbs_auth_key).returns("hunter2")
|
|
Danbooru.config.stubs(:iqdbs_server).returns("http://localhost:3004")
|
|
end
|
|
|
|
def mock_iqdb_matches!(post_or_source, matches)
|
|
source = post_or_source.is_a?(Post) ? post_or_source.complete_preview_file_url : post_or_source
|
|
url = "http://localhost:3004/similar?key=hunter2&url=#{CGI.escape source}&ref"
|
|
body = matches.map { |post| { post_id: post.id } }.to_json
|
|
|
|
stub_request(:get, url).to_return(body: body)
|
|
end
|
|
end
|