96 lines
3.2 KiB
Ruby
96 lines
3.2 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require 'spec_helper'
|
|
|
|
RSpec.describe Packages::MarkPackageFilesForDestructionService, :aggregate_failures do
|
|
let(:service) { described_class.new(package_files) }
|
|
|
|
describe '#execute', :aggregate_failures do
|
|
let(:batch_deadline) { nil }
|
|
|
|
subject { service.execute(batch_deadline: batch_deadline) }
|
|
|
|
shared_examples 'executing successfully' do |marked_package_files_count: 0|
|
|
it 'marks package files for destruction' do
|
|
expect { subject }
|
|
.to change { ::Packages::PackageFile.pending_destruction.count }.by(package_files.size)
|
|
end
|
|
|
|
it 'executes successfully' do
|
|
expect(subject).to be_success
|
|
expect(subject.message).to eq('Package files are now pending destruction')
|
|
expect(subject.payload).to eq(marked_package_files_count: marked_package_files_count)
|
|
end
|
|
end
|
|
|
|
context 'with no package files' do
|
|
let_it_be(:package_files) { ::Packages::PackageFile.none }
|
|
|
|
it_behaves_like 'executing successfully'
|
|
end
|
|
|
|
context 'with a single package file' do
|
|
let_it_be(:package_file) { create(:package_file) }
|
|
let_it_be(:package_files) { ::Packages::PackageFile.id_in(package_file.id) }
|
|
|
|
it_behaves_like 'executing successfully', marked_package_files_count: 1
|
|
end
|
|
|
|
context 'with many package files' do
|
|
let_it_be(:package_files) { ::Packages::PackageFile.id_in(create_list(:package_file, 3).map(&:id)) }
|
|
|
|
it_behaves_like 'executing successfully', marked_package_files_count: 3
|
|
|
|
context 'with a batch deadline' do
|
|
let_it_be(:batch_deadline) { 250.seconds.from_now }
|
|
|
|
context 'when the deadline is not hit' do
|
|
before do
|
|
expect(Time.zone).to receive(:now).and_return(batch_deadline - 10.seconds)
|
|
end
|
|
|
|
it_behaves_like 'executing successfully', marked_package_files_count: 3
|
|
end
|
|
|
|
context 'when the deadline is hit' do
|
|
it 'does not execute the batch loop' do
|
|
expect(Time.zone).to receive(:now).and_return(batch_deadline + 10.seconds)
|
|
expect { subject }.to not_change { ::Packages::PackageFile.pending_destruction.count }
|
|
expect(subject).to be_error
|
|
expect(subject.message).to eq('Timeout while marking package files as pending destruction')
|
|
expect(subject.payload).to eq(marked_package_files_count: 0)
|
|
end
|
|
end
|
|
end
|
|
|
|
context 'when a batch size is defined' do
|
|
let_it_be(:batch_deadline) { 250.seconds.from_now }
|
|
|
|
let(:batch_size) { 2 }
|
|
|
|
subject { service.execute(batch_deadline: batch_deadline, batch_size: batch_size) }
|
|
|
|
before do
|
|
expect(Time.zone).to receive(:now).twice.and_call_original
|
|
end
|
|
|
|
it_behaves_like 'executing successfully', marked_package_files_count: 3
|
|
end
|
|
end
|
|
|
|
context 'with an error during the update' do
|
|
let_it_be(:package_files) { ::Packages::PackageFile.none }
|
|
|
|
before do
|
|
expect(package_files).to receive(:each_batch).and_raise('error!')
|
|
end
|
|
|
|
it 'raises the error' do
|
|
expect { subject }
|
|
.to raise_error('error!')
|
|
.and not_change { ::Packages::PackageFile.pending_destruction.count }
|
|
end
|
|
end
|
|
end
|
|
end
|