debian-mirror-gitlab/spec/services/projects/housekeeping_service_spec.rb

129 lines
4.2 KiB
Ruby
Raw Normal View History

2019-07-31 22:56:46 +05:30
# frozen_string_literal: true
2016-06-02 11:05:42 +05:30
require 'spec_helper'
2020-07-28 23:09:34 +05:30
RSpec.describe Projects::HousekeepingService do
2017-08-17 22:00:37 +05:30
subject { described_class.new(project) }
2019-12-21 20:55:43 +05:30
2020-03-13 15:44:24 +05:30
let_it_be(:project) { create(:project, :repository) }
2016-06-02 11:05:42 +05:30
2016-09-29 09:46:39 +05:30
before do
project.reset_pushes_since_gc
end
after do
project.reset_pushes_since_gc
end
2016-06-02 11:05:42 +05:30
2016-09-29 09:46:39 +05:30
describe '#execute' do
2016-06-02 11:05:42 +05:30
it 'enqueues a sidekiq job' do
2017-08-17 22:00:37 +05:30
expect(subject).to receive(:try_obtain_lease).and_return(:the_uuid)
expect(subject).to receive(:lease_key).and_return(:the_lease_key)
2018-11-08 19:23:39 +05:30
expect(subject).to receive(:task).and_return(:incremental_repack)
expect(GitGarbageCollectWorker).to receive(:perform_async).with(project.id, :incremental_repack, :the_lease_key, :the_uuid).and_call_original
2016-06-02 11:05:42 +05:30
2018-11-08 19:23:39 +05:30
Sidekiq::Testing.fake! do
expect { subject.execute }.to change(GitGarbageCollectWorker.jobs, :size).by(1)
end
2016-06-02 11:05:42 +05:30
end
2018-03-17 18:26:18 +05:30
it 'yields the block if given' do
expect do |block|
subject.execute(&block)
end.to yield_with_no_args
end
2018-11-08 19:23:39 +05:30
it 'resets counter after execution' do
expect(subject).to receive(:try_obtain_lease).and_return(:the_uuid)
allow(subject).to receive(:gc_period).and_return(1)
project.increment_pushes_since_gc
2018-11-18 11:00:15 +05:30
perform_enqueued_jobs do
2018-11-08 19:23:39 +05:30
expect { subject.execute }.to change { project.pushes_since_gc }.to(0)
end
end
2016-08-24 12:49:21 +05:30
context 'when no lease can be obtained' do
2018-03-17 18:26:18 +05:30
before do
2016-08-24 12:49:21 +05:30
expect(subject).to receive(:try_obtain_lease).and_return(false)
end
2016-06-02 11:05:42 +05:30
2016-08-24 12:49:21 +05:30
it 'does not enqueue a job' do
expect(GitGarbageCollectWorker).not_to receive(:perform_async)
expect { subject.execute }.to raise_error(Projects::HousekeepingService::LeaseTaken)
end
it 'does not reset pushes_since_gc' do
expect do
expect { subject.execute }.to raise_error(Projects::HousekeepingService::LeaseTaken)
2016-09-29 09:46:39 +05:30
end.not_to change { project.pushes_since_gc }
2016-08-24 12:49:21 +05:30
end
2018-03-17 18:26:18 +05:30
it 'does not yield' do
expect do |block|
expect { subject.execute(&block) }
.to raise_error(Projects::HousekeepingService::LeaseTaken)
end.not_to yield_with_no_args
end
2016-06-02 11:05:42 +05:30
end
2018-11-08 19:23:39 +05:30
context 'task type' do
it 'goes through all three housekeeping tasks, executing only the highest task when there is overlap' do
allow(subject).to receive(:try_obtain_lease).and_return(:the_uuid)
allow(subject).to receive(:lease_key).and_return(:the_lease_key)
# At push 200
expect(GitGarbageCollectWorker).to receive(:perform_async).with(project.id, :gc, :the_lease_key, :the_uuid)
2020-03-13 15:44:24 +05:30
.once
2018-11-08 19:23:39 +05:30
# At push 50, 100, 150
expect(GitGarbageCollectWorker).to receive(:perform_async).with(project.id, :full_repack, :the_lease_key, :the_uuid)
.exactly(3).times
# At push 10, 20, ... (except those above)
expect(GitGarbageCollectWorker).to receive(:perform_async).with(project.id, :incremental_repack, :the_lease_key, :the_uuid)
.exactly(16).times
2019-07-31 22:56:46 +05:30
# At push 6, 12, 18, ... (except those above)
expect(GitGarbageCollectWorker).to receive(:perform_async).with(project.id, :pack_refs, :the_lease_key, :the_uuid)
.exactly(27).times
2018-11-08 19:23:39 +05:30
201.times do
subject.increment!
subject.execute if subject.needed?
end
expect(project.pushes_since_gc).to eq(1)
end
end
2019-07-07 11:18:12 +05:30
it 'runs the task specifically requested' do
housekeeping = described_class.new(project, :gc)
allow(housekeeping).to receive(:try_obtain_lease).and_return(:gc_uuid)
allow(housekeeping).to receive(:lease_key).and_return(:gc_lease_key)
expect(GitGarbageCollectWorker).to receive(:perform_async).with(project.id, :gc, :gc_lease_key, :gc_uuid).twice
2.times do
housekeeping.execute
end
end
2016-06-02 11:05:42 +05:30
end
2016-09-29 09:46:39 +05:30
describe '#needed?' do
2016-06-02 11:05:42 +05:30
it 'when the count is low enough' do
expect(subject.needed?).to eq(false)
end
it 'when the count is high enough' do
allow(project).to receive(:pushes_since_gc).and_return(10)
expect(subject.needed?).to eq(true)
end
end
2016-09-29 09:46:39 +05:30
describe '#increment!' do
2016-06-02 11:05:42 +05:30
it 'increments the pushes_since_gc counter' do
2018-11-08 19:23:39 +05:30
expect { subject.increment! }.to change { project.pushes_since_gc }.by(1)
2016-08-24 12:49:21 +05:30
end
2016-06-02 11:05:42 +05:30
end
end