Avoid comparing domains when looking for an exact match of a local account (#3336)

signup-info-prompt
happycoloredbanana 2017-05-27 01:55:08 +03:00 committed by Eugen Rochko
parent 3836d293a1
commit 9642601126
2 changed files with 32 additions and 7 deletions

View File

@ -59,7 +59,13 @@ class AccountSearchService < BaseService
end end
def exact_match def exact_match
@_exact_match ||= Account.find_remote(query_username, query_domain) @_exact_match ||= begin
if domain_is_local?
Account.find_local(query_username)
else
Account.find_remote(query_username, query_domain)
end
end
end end
def search_results def search_results

View File

@ -33,25 +33,25 @@ describe AccountSearchService do
describe 'searching local and remote users' do describe 'searching local and remote users' do
describe "when only '@'" do describe "when only '@'" do
before do before do
allow(Account).to receive(:find_remote) allow(Account).to receive(:find_local)
allow(Account).to receive(:search_for) allow(Account).to receive(:search_for)
subject.call('@', 10) subject.call('@', 10)
end end
it 'uses find_remote with empty query to look for local accounts' do it 'uses find_local with empty query to look for local accounts' do
expect(Account).to have_received(:find_remote).with('', nil) expect(Account).to have_received(:find_local).with('')
end end
end end
describe 'when no domain' do describe 'when no domain' do
before do before do
allow(Account).to receive(:find_remote) allow(Account).to receive(:find_local)
allow(Account).to receive(:search_for) allow(Account).to receive(:search_for)
subject.call('one', 10) subject.call('one', 10)
end end
it 'uses find_remote with nil domain to look for local accounts' do it 'uses find_local to look for local accounts' do
expect(Account).to have_received(:find_remote).with('one', nil) expect(Account).to have_received(:find_local).with('one')
end end
it 'uses search_for to find matches' do it 'uses search_for to find matches' do
@ -101,6 +101,25 @@ describe AccountSearchService do
end end
end end
describe 'when there is a local domain' do
around do |example|
before = Rails.configuration.x.local_domain
example.run
Rails.configuration.x.local_domain = before
end
it 'returns exact match first' do
remote = Fabricate(:account, username: 'a', domain: 'remote', display_name: 'e')
remote_too = Fabricate(:account, username: 'b', domain: 'remote', display_name: 'e')
exact = Fabricate(:account, username: 'e')
Rails.configuration.x.local_domain = 'example.com'
results = subject.call('e@example.com', 2)
expect(results.size).to eq 2
expect(results).to eq([exact, remote]).or eq([exact, remote_too])
end
end
describe 'when there is a domain but no exact match' do describe 'when there is a domain but no exact match' do
it 'follows the remote account when resolve is true' do it 'follows the remote account when resolve is true' do
service = double(call: nil) service = double(call: nil)