2022-01-18 15:14:54 +00:00
|
|
|
# frozen_string_literal: true
|
|
|
|
|
|
|
|
require 'spec_helper'
|
|
|
|
|
|
|
|
RSpec.describe Packages::CleanupPackageFileWorker do
|
2022-04-07 15:09:57 +00:00
|
|
|
let_it_be_with_reload(:package) { create(:package) }
|
2022-01-18 15:14:54 +00:00
|
|
|
|
|
|
|
let(:worker) { described_class.new }
|
|
|
|
|
|
|
|
describe '#perform_work' do
|
|
|
|
subject { worker.perform_work }
|
|
|
|
|
|
|
|
context 'with no work to do' do
|
|
|
|
it { is_expected.to be_nil }
|
|
|
|
end
|
|
|
|
|
|
|
|
context 'with work to do' do
|
|
|
|
let_it_be(:package_file1) { create(:package_file, package: package) }
|
|
|
|
let_it_be(:package_file2) { create(:package_file, :pending_destruction, package: package) }
|
|
|
|
let_it_be(:package_file3) { create(:package_file, :pending_destruction, package: package, updated_at: 1.year.ago, created_at: 1.year.ago) }
|
|
|
|
|
|
|
|
it 'deletes the oldest package file pending destruction based on id', :aggregate_failures do
|
2022-01-19 18:14:01 +00:00
|
|
|
expect(worker).to receive(:log_extra_metadata_on_done).twice
|
2022-01-18 15:14:54 +00:00
|
|
|
|
|
|
|
expect { subject }.to change { Packages::PackageFile.count }.by(-1)
|
2022-04-07 15:09:57 +00:00
|
|
|
.and not_change { Packages::Package.count }
|
|
|
|
expect { package_file2.reload }.to raise_error(ActiveRecord::RecordNotFound)
|
|
|
|
end
|
|
|
|
|
|
|
|
context 'with a duplicated PyPI package file' do
|
|
|
|
let_it_be_with_reload(:duplicated_package_file) { create(:package_file, package: package) }
|
|
|
|
|
|
|
|
before do
|
|
|
|
package.update!(package_type: :pypi, version: '1.2.3')
|
|
|
|
duplicated_package_file.update_column(:file_name, package_file2.file_name)
|
|
|
|
end
|
|
|
|
|
|
|
|
it 'deletes one of the duplicates' do
|
|
|
|
expect { subject }.to change { Packages::PackageFile.count }.by(-1)
|
|
|
|
.and not_change { Packages::Package.count }
|
|
|
|
expect { package_file2.reload }.to raise_error(ActiveRecord::RecordNotFound)
|
|
|
|
end
|
2022-01-18 15:14:54 +00:00
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2022-04-09 00:09:46 +00:00
|
|
|
context 'with a package file to destroy' do
|
2022-01-18 15:14:54 +00:00
|
|
|
let_it_be(:package_file) { create(:package_file, :pending_destruction) }
|
|
|
|
|
2022-04-09 00:09:46 +00:00
|
|
|
context 'with an error during the destroy' do
|
|
|
|
before do
|
2022-04-19 09:08:55 +00:00
|
|
|
allow(worker).to receive(:log_metadata).and_raise('Error!')
|
2022-04-09 00:09:46 +00:00
|
|
|
end
|
|
|
|
|
|
|
|
it 'handles the error' do
|
2022-07-12 18:08:46 +00:00
|
|
|
expect(Gitlab::ErrorTracking).to receive(:log_exception).with(instance_of(RuntimeError), class: described_class.name)
|
2022-04-09 00:09:46 +00:00
|
|
|
expect { subject }.to change { Packages::PackageFile.error.count }.from(0).to(1)
|
|
|
|
expect(package_file.reload).to be_error
|
|
|
|
end
|
2022-01-18 15:14:54 +00:00
|
|
|
end
|
|
|
|
|
2022-04-09 00:09:46 +00:00
|
|
|
context 'when trying to destroy a destroyed record' do
|
|
|
|
before do
|
|
|
|
allow_next_found_instance_of(Packages::PackageFile) do |package_file|
|
|
|
|
destroy_method = package_file.method(:destroy!)
|
|
|
|
|
|
|
|
allow(package_file).to receive(:destroy!) do
|
|
|
|
destroy_method.call
|
|
|
|
|
|
|
|
raise 'Error!'
|
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
it 'handles the error' do
|
2022-07-12 18:08:46 +00:00
|
|
|
expect(Gitlab::ErrorTracking).to receive(:log_exception).with(instance_of(RuntimeError), class: described_class.name)
|
|
|
|
expect { subject }.not_to change { Packages::PackageFile.count }
|
|
|
|
expect(package_file.reload).to be_error
|
2022-04-09 00:09:46 +00:00
|
|
|
end
|
2022-01-18 15:14:54 +00:00
|
|
|
end
|
|
|
|
end
|
2022-02-04 00:13:53 +00:00
|
|
|
|
2022-04-19 09:08:55 +00:00
|
|
|
describe 'removing the last package file' do
|
2022-02-04 00:13:53 +00:00
|
|
|
let_it_be(:package_file) { create(:package_file, :pending_destruction, package: package) }
|
|
|
|
|
|
|
|
it 'deletes the package file and the package' do
|
|
|
|
expect(worker).to receive(:log_extra_metadata_on_done).twice
|
|
|
|
|
|
|
|
expect { subject }.to change { Packages::PackageFile.count }.by(-1)
|
|
|
|
.and change { Packages::Package.count }.by(-1)
|
|
|
|
end
|
|
|
|
end
|
2022-01-18 15:14:54 +00:00
|
|
|
end
|
|
|
|
|
|
|
|
describe '#max_running_jobs' do
|
|
|
|
let(:capacity) { 5 }
|
|
|
|
|
|
|
|
subject { worker.max_running_jobs }
|
|
|
|
|
|
|
|
before do
|
|
|
|
stub_application_setting(packages_cleanup_package_file_worker_capacity: capacity)
|
|
|
|
end
|
|
|
|
|
|
|
|
it { is_expected.to eq(capacity) }
|
|
|
|
end
|
|
|
|
|
|
|
|
describe '#remaining_work_count' do
|
2022-04-19 09:08:55 +00:00
|
|
|
before_all do
|
|
|
|
create_list(:package_file, 2, :pending_destruction, package: package)
|
2022-01-18 15:14:54 +00:00
|
|
|
end
|
|
|
|
|
|
|
|
subject { worker.remaining_work_count }
|
|
|
|
|
2022-04-19 09:08:55 +00:00
|
|
|
it { is_expected.to eq(2) }
|
2022-01-18 15:14:54 +00:00
|
|
|
end
|
|
|
|
end
|