forked from fedi/mastodon
128 lines
4.2 KiB
Ruby
128 lines
4.2 KiB
Ruby
require 'rails_helper'
|
|
|
|
describe Scheduler::AccountsStatusesCleanupScheduler do
|
|
subject { described_class.new }
|
|
|
|
let!(:account1) { Fabricate(:account, domain: nil) }
|
|
let!(:account2) { Fabricate(:account, domain: nil) }
|
|
let!(:account3) { Fabricate(:account, domain: nil) }
|
|
let!(:account4) { Fabricate(:account, domain: nil) }
|
|
let!(:remote) { Fabricate(:account) }
|
|
|
|
let!(:policy1) { Fabricate(:account_statuses_cleanup_policy, account: account1) }
|
|
let!(:policy2) { Fabricate(:account_statuses_cleanup_policy, account: account3) }
|
|
let!(:policy3) { Fabricate(:account_statuses_cleanup_policy, account: account4, enabled: false) }
|
|
|
|
let(:queue_size) { 0 }
|
|
let(:queue_latency) { 0 }
|
|
let(:process_set_stub) do
|
|
[
|
|
{
|
|
'concurrency' => 2,
|
|
'queues' => ['push', 'default'],
|
|
},
|
|
]
|
|
end
|
|
let(:retry_size) { 0 }
|
|
|
|
before do
|
|
queue_stub = double
|
|
allow(queue_stub).to receive(:size).and_return(queue_size)
|
|
allow(queue_stub).to receive(:latency).and_return(queue_latency)
|
|
allow(Sidekiq::Queue).to receive(:new).and_return(queue_stub)
|
|
allow(Sidekiq::ProcessSet).to receive(:new).and_return(process_set_stub)
|
|
|
|
sidekiq_stats_stub = double
|
|
allow(sidekiq_stats_stub).to receive(:retry_size).and_return(retry_size)
|
|
allow(Sidekiq::Stats).to receive(:new).and_return(sidekiq_stats_stub)
|
|
|
|
# Create a bunch of old statuses
|
|
10.times do
|
|
Fabricate(:status, account: account1, created_at: 3.years.ago)
|
|
Fabricate(:status, account: account2, created_at: 3.years.ago)
|
|
Fabricate(:status, account: account3, created_at: 3.years.ago)
|
|
Fabricate(:status, account: account4, created_at: 3.years.ago)
|
|
Fabricate(:status, account: remote, created_at: 3.years.ago)
|
|
end
|
|
|
|
# Create a bunch of newer statuses
|
|
5.times do
|
|
Fabricate(:status, account: account1, created_at: 3.minutes.ago)
|
|
Fabricate(:status, account: account2, created_at: 3.minutes.ago)
|
|
Fabricate(:status, account: account3, created_at: 3.minutes.ago)
|
|
Fabricate(:status, account: account4, created_at: 3.minutes.ago)
|
|
Fabricate(:status, account: remote, created_at: 3.minutes.ago)
|
|
end
|
|
end
|
|
|
|
describe '#under_load?' do
|
|
context 'when nothing is queued' do
|
|
it 'returns false' do
|
|
expect(subject.under_load?).to be false
|
|
end
|
|
end
|
|
|
|
context 'when numerous jobs are queued' do
|
|
let(:queue_size) { 5 }
|
|
let(:queue_latency) { 120 }
|
|
|
|
it 'returns true' do
|
|
expect(subject.under_load?).to be true
|
|
end
|
|
end
|
|
|
|
context 'when there is a huge amount of jobs to retry' do
|
|
let(:retry_size) { 1_000_000 }
|
|
|
|
it 'returns true' do
|
|
expect(subject.under_load?).to be true
|
|
end
|
|
end
|
|
end
|
|
|
|
describe '#get_budget' do
|
|
context 'on a single thread' do
|
|
let(:process_set_stub) { [{ 'concurrency' => 1, 'queues' => ['push', 'default'] }] }
|
|
|
|
it 'returns a low value' do
|
|
expect(subject.compute_budget).to be < 10
|
|
end
|
|
end
|
|
|
|
context 'on a lot of threads' do
|
|
let(:process_set_stub) do
|
|
[
|
|
{ 'concurrency' => 2, 'queues' => ['push', 'default'] },
|
|
{ 'concurrency' => 2, 'queues' => ['push'] },
|
|
{ 'concurrency' => 2, 'queues' => ['push'] },
|
|
{ 'concurrency' => 2, 'queues' => ['push'] },
|
|
]
|
|
end
|
|
|
|
it 'returns a larger value' do
|
|
expect(subject.compute_budget).to be > 10
|
|
end
|
|
end
|
|
end
|
|
|
|
describe '#perform' do
|
|
context 'when the budget is lower than the number of toots to delete' do
|
|
it 'deletes as many statuses as the given budget' do
|
|
expect { subject.perform }.to change { Status.count }.by(-subject.compute_budget)
|
|
end
|
|
|
|
it 'does not delete from accounts with no cleanup policy' do
|
|
expect { subject.perform }.to_not change { account2.statuses.count }
|
|
end
|
|
|
|
it 'does not delete from accounts with disabled cleanup policies' do
|
|
expect { subject.perform }.to_not change { account4.statuses.count }
|
|
end
|
|
|
|
it 'eventually deletes every deletable toot' do
|
|
expect { subject.perform; subject.perform; subject.perform; subject.perform }.to change { Status.count }.by(-20)
|
|
end
|
|
end
|
|
end
|
|
end
|