Add specs (and refactor) of FetchRemoteResourceService and SearchService (#2812)

* Coverage for fetch remote resource service

* Refactor fetch remote resource service

* Coverage for search service

* Refactor search service
pull/15/head
Matt Jankowski 2017-05-05 11:26:04 -04:00 committed by Eugen Rochko
parent 9501a87704
commit 20c37ed0f9
4 changed files with 213 additions and 19 deletions

View File

@ -1,18 +1,41 @@
# frozen_string_literal: true # frozen_string_literal: true
class FetchRemoteResourceService < BaseService class FetchRemoteResourceService < BaseService
attr_reader :url
def call(url) def call(url)
atom_url, body = FetchAtomService.new.call(url) @url = url
process_url unless atom_url.nil?
end
return nil if atom_url.nil? private
xml = Nokogiri::XML(body) def process_url
xml.encoding = 'utf-8' case xml_root
when 'feed'
if xml.root.name == 'feed'
FetchRemoteAccountService.new.call(atom_url, body) FetchRemoteAccountService.new.call(atom_url, body)
elsif xml.root.name == 'entry' when 'entry'
FetchRemoteStatusService.new.call(atom_url, body) FetchRemoteStatusService.new.call(atom_url, body)
end end
end end
def fetched_atom_feed
@_fetched_atom_feed ||= FetchAtomService.new.call(url)
end
def atom_url
fetched_atom_feed.first
end
def body
fetched_atom_feed.last
end
def xml_root
xml_data.root.name
end
def xml_data
@_xml_data ||= Nokogiri::XML(body, nil, 'utf-8')
end
end end

View File

@ -1,21 +1,38 @@
# frozen_string_literal: true # frozen_string_literal: true
class SearchService < BaseService class SearchService < BaseService
attr_accessor :query
def call(query, limit, resolve = false, account = nil) def call(query, limit, resolve = false, account = nil)
results = { accounts: [], hashtags: [], statuses: [] } @query = query
return results if query.blank? default_results.tap do |results|
if url_query?
if query =~ /\Ahttps?:\/\// results.merge!(remote_resource_results) unless remote_resource.nil?
resource = FetchRemoteResourceService.new.call(query) elsif query.present?
results[:accounts] = AccountSearchService.new.call(query, limit, resolve, account)
results[:accounts] << resource if resource.is_a?(Account) results[:hashtags] = Tag.search_for(query.gsub(/\A#/, ''), limit) unless query.start_with?('@')
results[:statuses] << resource if resource.is_a?(Status) end
else
results[:accounts] = AccountSearchService.new.call(query, limit, resolve, account)
results[:hashtags] = Tag.search_for(query.gsub(/\A#/, ''), limit) unless query.start_with?('@')
end end
end
results def default_results
{ accounts: [], hashtags: [], statuses: [] }
end
def url_query?
query =~ /\Ahttps?:\/\//
end
def remote_resource_results
{ remote_resource_symbol => [remote_resource] }
end
def remote_resource
@_remote_resource ||= FetchRemoteResourceService.new.call(query)
end
def remote_resource_symbol
remote_resource.class.name.downcase.pluralize.to_sym
end end
end end

View File

@ -0,0 +1,53 @@
# frozen_string_literal: true
require 'rails_helper'
describe FetchRemoteResourceService do
subject { described_class.new }
describe '#call' do
it 'returns nil when there is no atom url' do
url = 'http://example.com/missing-atom'
service = double
allow(FetchAtomService).to receive(:new).and_return service
allow(service).to receive(:call).with(url).and_return([nil, 'body'])
result = subject.call(url)
expect(result).to be_nil
end
it 'fetches remote accounts for feed types' do
url = 'http://example.com/atom-feed'
service = double
allow(FetchAtomService).to receive(:new).and_return service
feed_url = 'http://feed-url'
feed_content = '<feed>contents</feed>'
allow(service).to receive(:call).with(url).and_return([feed_url, feed_content])
account_service = double
allow(FetchRemoteAccountService).to receive(:new).and_return(account_service)
allow(account_service).to receive(:call)
_result = subject.call(url)
expect(account_service).to have_received(:call).with(feed_url, feed_content)
end
it 'fetches remote statuses for entry types' do
url = 'http://example.com/atom-entry'
service = double
allow(FetchAtomService).to receive(:new).and_return service
feed_url = 'http://feed-url'
feed_content = '<entry>contents</entry>'
allow(service).to receive(:call).with(url).and_return([feed_url, feed_content])
account_service = double
allow(FetchRemoteStatusService).to receive(:new).and_return(account_service)
allow(account_service).to receive(:call)
_result = subject.call(url)
expect(account_service).to have_received(:call).with(feed_url, feed_content)
end
end
end

View File

@ -0,0 +1,101 @@
# frozen_string_literal: true
require 'rails_helper'
describe SearchService do
subject { described_class.new }
describe '#call' do
describe 'with a blank query' do
it 'returns empty results without searching' do
allow(AccountSearchService).to receive(:new)
allow(Tag).to receive(:search_for)
results = subject.call('', 10)
expect(results).to eq(empty_results)
expect(AccountSearchService).not_to have_received(:new)
expect(Tag).not_to have_received(:search_for)
end
end
describe 'with an url query' do
before do
@query = 'http://test.host/query'
end
context 'that does not find anything' do
it 'returns the empty results' do
service = double(call: nil)
allow(FetchRemoteResourceService).to receive(:new).and_return(service)
results = subject.call(@query, 10)
expect(service).to have_received(:call).with(@query)
expect(results).to eq empty_results
end
end
context 'that finds an account' do
it 'includes the account in the results' do
account = Account.new
service = double(call: account)
allow(FetchRemoteResourceService).to receive(:new).and_return(service)
results = subject.call(@query, 10)
expect(service).to have_received(:call).with(@query)
expect(results).to eq empty_results.merge(accounts: [account])
end
end
context 'that finds a status' do
it 'includes the status in the results' do
status = Status.new
service = double(call: status)
allow(FetchRemoteResourceService).to receive(:new).and_return(service)
results = subject.call(@query, 10)
expect(service).to have_received(:call).with(@query)
expect(results).to eq empty_results.merge(statuses: [status])
end
end
end
describe 'with a non-url query' do
context 'that matches an account' do
it 'includes the account in the results' do
query = 'username'
account = Account.new
service = double(call: [account])
allow(AccountSearchService).to receive(:new).and_return(service)
results = subject.call(query, 10)
expect(service).to have_received(:call).with(query, 10, false, nil)
expect(results).to eq empty_results.merge(accounts: [account])
end
end
context 'that matches a tag' do
it 'includes the tag in the results' do
query = '#tag'
tag = Tag.new
allow(Tag).to receive(:search_for).with('tag', 10).and_return([tag])
results = subject.call(query, 10)
expect(Tag).to have_received(:search_for).with('tag', 10)
expect(results).to eq empty_results.merge(hashtags: [tag])
end
it 'does not include tag when starts with @ character' do
query = '@username'
allow(Tag).to receive(:search_for)
results = subject.call(query, 10)
expect(Tag).not_to have_received(:search_for)
expect(results).to eq empty_results
end
end
end
end
def empty_results
{ accounts: [], hashtags: [], statuses: [] }
end
end