Fix infinite loop in AccountsStatusesCleanupScheduler (#24840)

This commit is contained in:
Claire 2023-05-04 10:13:00 +02:00 committed by GitHub
parent f95125bd04
commit c3c7d59072
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 15 additions and 0 deletions

View file

@ -79,6 +79,7 @@ class Scheduler::AccountsStatusesCleanupScheduler
# and start back after the last processed account otherwise
break if budget.zero? || (num_processed_accounts.zero? && !full_iteration)
full_iteration = false unless first_iteration
first_iteration = false
end
end

View file

@ -143,6 +143,20 @@ describe Scheduler::AccountsStatusesCleanupScheduler do
expect { subject.perform }.to change { Status.count }.by(-30)
end
end
context 'when there is no work to be done' do
let(:process_set_stub) { [{ 'concurrency' => 400, 'queues' => %w(push default) }] }
before do
stub_const 'Scheduler::AccountsStatusesCleanupScheduler::MAX_BUDGET', 400
subject.perform
end
it 'does not get stuck' do
expect(subject.compute_budget).to eq(400)
expect { subject.perform }.to_not change { Status.count }
end
end
end
end
end