mirror of https://github.com/mastodon/mastodon
Reduce `.times` usage in `StatusPin` and add `PIN_LIMIT` constant in validator (#27945)
This commit is contained in:
parent
1fabf20a88
commit
00c6ebd86f
|
@ -1,10 +1,12 @@
|
||||||
# frozen_string_literal: true
|
# frozen_string_literal: true
|
||||||
|
|
||||||
class StatusPinValidator < ActiveModel::Validator
|
class StatusPinValidator < ActiveModel::Validator
|
||||||
|
PIN_LIMIT = 5
|
||||||
|
|
||||||
def validate(pin)
|
def validate(pin)
|
||||||
pin.errors.add(:base, I18n.t('statuses.pin_errors.reblog')) if pin.status.reblog?
|
pin.errors.add(:base, I18n.t('statuses.pin_errors.reblog')) if pin.status.reblog?
|
||||||
pin.errors.add(:base, I18n.t('statuses.pin_errors.ownership')) if pin.account_id != pin.status.account_id
|
pin.errors.add(:base, I18n.t('statuses.pin_errors.ownership')) if pin.account_id != pin.status.account_id
|
||||||
pin.errors.add(:base, I18n.t('statuses.pin_errors.direct')) if pin.status.direct_visibility?
|
pin.errors.add(:base, I18n.t('statuses.pin_errors.direct')) if pin.status.direct_visibility?
|
||||||
pin.errors.add(:base, I18n.t('statuses.pin_errors.limit')) if pin.account.status_pins.count > 4 && pin.account.local?
|
pin.errors.add(:base, I18n.t('statuses.pin_errors.limit')) if pin.account.status_pins.count >= PIN_LIMIT && pin.account.local?
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
|
|
|
@ -40,35 +40,34 @@ RSpec.describe StatusPin do
|
||||||
expect(described_class.new(account: account, status: status).save).to be false
|
expect(described_class.new(account: account, status: status).save).to be false
|
||||||
end
|
end
|
||||||
|
|
||||||
max_pins = 5
|
context 'with a pin limit' do
|
||||||
|
before { stub_const('StatusPinValidator::PIN_LIMIT', 2) }
|
||||||
|
|
||||||
it 'does not allow pins above the max' do
|
it 'does not allow pins above the max' do
|
||||||
account = Fabricate(:account)
|
account = Fabricate(:account)
|
||||||
status = []
|
|
||||||
|
|
||||||
(max_pins + 1).times do |i|
|
Fabricate.times(StatusPinValidator::PIN_LIMIT, :status_pin, account: account)
|
||||||
status[i] = Fabricate(:status, account: account)
|
|
||||||
end
|
|
||||||
|
|
||||||
max_pins.times do |i|
|
pin = described_class.new(account: account, status: Fabricate(:status, account: account))
|
||||||
expect(described_class.new(account: account, status: status[i]).save).to be true
|
expect(pin.save)
|
||||||
end
|
.to be(false)
|
||||||
|
|
||||||
expect(described_class.new(account: account, status: status[max_pins]).save).to be false
|
expect(pin.errors[:base])
|
||||||
|
.to contain_exactly(I18n.t('statuses.pin_errors.limit'))
|
||||||
end
|
end
|
||||||
|
|
||||||
it 'allows pins above the max for remote accounts' do
|
it 'allows pins above the max for remote accounts' do
|
||||||
account = Fabricate(:account, domain: 'remote.test', username: 'bob', url: 'https://remote.test/')
|
account = Fabricate(:account, domain: 'remote.test', username: 'bob', url: 'https://remote.test/')
|
||||||
status = []
|
|
||||||
|
|
||||||
(max_pins + 1).times do |i|
|
Fabricate.times(StatusPinValidator::PIN_LIMIT, :status_pin, account: account)
|
||||||
status[i] = Fabricate(:status, account: account)
|
|
||||||
|
pin = described_class.new(account: account, status: Fabricate(:status, account: account))
|
||||||
|
expect(pin.save)
|
||||||
|
.to be(true)
|
||||||
|
|
||||||
|
expect(pin.errors[:base])
|
||||||
|
.to be_empty
|
||||||
end
|
end
|
||||||
|
|
||||||
max_pins.times do |i|
|
|
||||||
expect(described_class.new(account: account, status: status[i]).save).to be true
|
|
||||||
end
|
|
||||||
|
|
||||||
expect(described_class.new(account: account, status: status[max_pins]).save).to be true
|
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
|
|
Loading…
Reference in New Issue