mirror of
https://github.com/lunaisnotaboy/mastodon.git
synced 2024-11-25 08:13:24 +00:00
Add coverage for ReportFilter and AccountFilter (#3236)
This commit is contained in:
parent
b51398d0dd
commit
e1b42e9aa0
|
@ -89,6 +89,8 @@ class Account < ApplicationRecord
|
||||||
scope :recent, -> { reorder(id: :desc) }
|
scope :recent, -> { reorder(id: :desc) }
|
||||||
scope :alphabetic, -> { order(domain: :asc, username: :asc) }
|
scope :alphabetic, -> { order(domain: :asc, username: :asc) }
|
||||||
scope :by_domain_accounts, -> { group(:domain).select(:domain, 'COUNT(*) AS accounts_count').order('accounts_count desc') }
|
scope :by_domain_accounts, -> { group(:domain).select(:domain, 'COUNT(*) AS accounts_count').order('accounts_count desc') }
|
||||||
|
scope :matches_username, ->(value) { where(arel_table[:username].matches("#{value}%")) }
|
||||||
|
scope :matches_display_name, ->(value) { where(arel_table[:display_name].matches("#{value}%")) }
|
||||||
|
|
||||||
delegate :email,
|
delegate :email,
|
||||||
:current_sign_in_ip,
|
:current_sign_in_ip,
|
||||||
|
|
|
@ -18,8 +18,6 @@ class AccountFilter
|
||||||
private
|
private
|
||||||
|
|
||||||
def scope_for(key, value)
|
def scope_for(key, value)
|
||||||
accounts = Account.arel_table
|
|
||||||
|
|
||||||
case key.to_s
|
case key.to_s
|
||||||
when 'local'
|
when 'local'
|
||||||
Account.local
|
Account.local
|
||||||
|
@ -34,21 +32,26 @@ class AccountFilter
|
||||||
when 'suspended'
|
when 'suspended'
|
||||||
Account.suspended
|
Account.suspended
|
||||||
when 'username'
|
when 'username'
|
||||||
Account.where(accounts[:username].matches("#{value}%"))
|
Account.matches_username(value)
|
||||||
when 'display_name'
|
when 'display_name'
|
||||||
Account.where(accounts[:display_name].matches("#{value}%"))
|
Account.matches_display_name(value)
|
||||||
when 'email'
|
when 'email'
|
||||||
users = User.arel_table
|
accounts_with_users.merge User.matches_email(value)
|
||||||
Account.joins(:user).merge(User.where(users[:email].matches("#{value}%")))
|
|
||||||
when 'ip'
|
when 'ip'
|
||||||
return Account.default_scoped unless valid_ip?(value)
|
if valid_ip?(value)
|
||||||
matches_ip = User.where(current_sign_in_ip: value).or(User.where(last_sign_in_ip: value))
|
accounts_with_users.merge User.with_recent_ip_address(value)
|
||||||
Account.joins(:user).merge(matches_ip)
|
else
|
||||||
|
Account.default_scoped
|
||||||
|
end
|
||||||
else
|
else
|
||||||
raise "Unknown filter: #{key}"
|
raise "Unknown filter: #{key}"
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
|
|
||||||
|
def accounts_with_users
|
||||||
|
Account.joins(:user)
|
||||||
|
end
|
||||||
|
|
||||||
def valid_ip?(value)
|
def valid_ip?(value)
|
||||||
IPAddr.new(value)
|
IPAddr.new(value)
|
||||||
true
|
true
|
||||||
|
|
|
@ -53,6 +53,8 @@ class User < ApplicationRecord
|
||||||
scope :admins, -> { where(admin: true) }
|
scope :admins, -> { where(admin: true) }
|
||||||
scope :confirmed, -> { where.not(confirmed_at: nil) }
|
scope :confirmed, -> { where.not(confirmed_at: nil) }
|
||||||
scope :inactive, -> { where(arel_table[:current_sign_in_at].lt(ACTIVE_DURATION.ago)) }
|
scope :inactive, -> { where(arel_table[:current_sign_in_at].lt(ACTIVE_DURATION.ago)) }
|
||||||
|
scope :matches_email, ->(value) { where(arel_table[:email].matches("#{value}%")) }
|
||||||
|
scope :with_recent_ip_address, ->(value) { where(arel_table[:current_sign_in_ip].eq(value).or(arel_table[:last_sign_in_ip].eq(value))) }
|
||||||
|
|
||||||
before_validation :sanitize_languages
|
before_validation :sanitize_languages
|
||||||
|
|
||||||
|
|
|
@ -17,19 +17,50 @@ describe AccountFilter do
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
|
|
||||||
|
describe 'when an IP address is provided' do
|
||||||
|
it 'filters with IP when valid' do
|
||||||
|
filter = described_class.new(ip: '127.0.0.1')
|
||||||
|
allow(User).to receive(:with_recent_ip_address).and_return(User.none)
|
||||||
|
|
||||||
|
filter.results
|
||||||
|
expect(User).to have_received(:with_recent_ip_address).with('127.0.0.1')
|
||||||
|
end
|
||||||
|
|
||||||
|
it 'skips IP when invalid' do
|
||||||
|
filter = described_class.new(ip: '345.678.901.234')
|
||||||
|
expect(User).not_to receive(:with_recent_ip_address)
|
||||||
|
|
||||||
|
filter.results
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
describe 'with valid params' do
|
describe 'with valid params' do
|
||||||
it 'combines filters on Account' do
|
it 'combines filters on Account' do
|
||||||
filter = described_class.new(by_domain: 'test.com', silenced: true)
|
filter = described_class.new(
|
||||||
|
by_domain: 'test.com',
|
||||||
|
silenced: true,
|
||||||
|
username: 'test',
|
||||||
|
display_name: 'name',
|
||||||
|
email: 'user@example.com',
|
||||||
|
)
|
||||||
|
|
||||||
allow(Account).to receive(:where).and_return(Account.none)
|
allow(Account).to receive(:where).and_return(Account.none)
|
||||||
allow(Account).to receive(:silenced).and_return(Account.none)
|
allow(Account).to receive(:silenced).and_return(Account.none)
|
||||||
|
allow(Account).to receive(:matches_display_name).and_return(Account.none)
|
||||||
|
allow(Account).to receive(:matches_username).and_return(Account.none)
|
||||||
|
allow(User).to receive(:matches_email).and_return(User.none)
|
||||||
|
|
||||||
filter.results
|
filter.results
|
||||||
|
|
||||||
expect(Account).to have_received(:where).with(domain: 'test.com')
|
expect(Account).to have_received(:where).with(domain: 'test.com')
|
||||||
expect(Account).to have_received(:silenced)
|
expect(Account).to have_received(:silenced)
|
||||||
|
expect(Account).to have_received(:matches_username).with('test')
|
||||||
|
expect(Account).to have_received(:matches_display_name).with('name')
|
||||||
|
expect(User).to have_received(:matches_email).with('user@example.com')
|
||||||
end
|
end
|
||||||
|
|
||||||
describe 'that call account methods' do
|
describe 'that call account methods' do
|
||||||
%i(local remote silenced recent).each do |option|
|
%i(local remote silenced recent suspended).each do |option|
|
||||||
it "delegates the #{option} option" do
|
it "delegates the #{option} option" do
|
||||||
allow(Account).to receive(option).and_return(Account.none)
|
allow(Account).to receive(option).and_return(Account.none)
|
||||||
filter = described_class.new({ option => true })
|
filter = described_class.new({ option => true })
|
||||||
|
|
|
@ -19,12 +19,13 @@ describe ReportFilter do
|
||||||
|
|
||||||
describe 'with valid params' do
|
describe 'with valid params' do
|
||||||
it 'combines filters on Report' do
|
it 'combines filters on Report' do
|
||||||
filter = ReportFilter.new(account_id: '123', resolved: true)
|
filter = ReportFilter.new(account_id: '123', resolved: true, target_account_id: '456')
|
||||||
|
|
||||||
allow(Report).to receive(:where).and_return(Report.none)
|
allow(Report).to receive(:where).and_return(Report.none)
|
||||||
allow(Report).to receive(:resolved).and_return(Report.none)
|
allow(Report).to receive(:resolved).and_return(Report.none)
|
||||||
filter.results
|
filter.results
|
||||||
expect(Report).to have_received(:where).with(account_id: '123')
|
expect(Report).to have_received(:where).with(account_id: '123')
|
||||||
|
expect(Report).to have_received(:where).with(target_account_id: '456')
|
||||||
expect(Report).to have_received(:resolved)
|
expect(Report).to have_received(:resolved)
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
|
|
Loading…
Reference in a new issue