1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
|
# frozen_string_literal: true
require 'spec_helper'
RSpec.describe Packages::CleanupPackageRegistryWorker, feature_category: :package_registry do
describe '#perform' do
let_it_be_with_reload(:package_files) { create_list(:package_file, 2, :pending_destruction) }
let_it_be(:policy) { create(:packages_cleanup_policy, :runnable) }
let_it_be(:package) { create(:package, project: policy.project) }
let(:worker) { described_class.new }
subject(:perform) { worker.perform }
context 'with package files pending destruction' do
it_behaves_like 'an idempotent worker'
it 'queues the cleanup job' do
expect(Packages::CleanupPackageFileWorker).to receive(:perform_with_capacity)
perform
end
end
context 'with no package files pending destruction' do
before do
::Packages::PackageFile.update_all(status: :default)
end
it_behaves_like 'an idempotent worker'
it 'does not queue the cleanup job' do
expect(Packages::CleanupPackageFileWorker).not_to receive(:perform_with_capacity)
perform
end
end
context 'with runnable policies' do
it_behaves_like 'an idempotent worker'
it 'queues the cleanup job' do
expect(Packages::Cleanup::ExecutePolicyWorker).to receive(:perform_with_capacity)
perform
end
end
context 'with no runnable policies' do
before do
policy.update_column(:next_run_at, 5.minutes.from_now)
end
it 'does not queue the cleanup job' do
expect(Packages::Cleanup::ExecutePolicyWorker).not_to receive(:perform_with_capacity)
perform
end
end
context 'with npm metadata caches pending destruction' do
let_it_be(:npm_metadata_cache) { create(:npm_metadata_cache, :stale) }
it_behaves_like 'an idempotent worker'
it 'queues the cleanup job' do
expect(Packages::Npm::CleanupStaleMetadataCacheWorker).to receive(:perform_with_capacity)
perform
end
end
context 'with no npm metadata caches pending destruction' do
it_behaves_like 'an idempotent worker'
it 'does not queue the cleanup job' do
expect(Packages::Npm::CleanupStaleMetadataCacheWorker).not_to receive(:perform_with_capacity)
perform
end
end
context 'with nuget symbols pending destruction' do
let_it_be(:nuget_symbol) { create(:nuget_symbol, :stale) }
include_examples 'an idempotent worker' do
it 'queues the cleanup job' do
expect(Packages::Nuget::CleanupStaleSymbolsWorker).to receive(:perform_with_capacity)
perform
end
end
end
context 'with no nuget symbols pending destruction' do
include_examples 'an idempotent worker' do
it 'does not queue the cleanup job' do
expect(Packages::Nuget::CleanupStaleSymbolsWorker).not_to receive(:perform_with_capacity)
perform
end
end
end
describe 'counts logging' do
let_it_be(:processing_package_file) { create(:package_file, status: :processing) }
it 'logs all the counts', :aggregate_failures do
expect(worker).to receive(:log_extra_metadata_on_done).with(:pending_destruction_package_files_count, 2)
expect(worker).to receive(:log_extra_metadata_on_done).with(:processing_package_files_count, 1)
expect(worker).to receive(:log_extra_metadata_on_done).with(:error_package_files_count, 0)
expect(worker).to receive(:log_extra_metadata_on_done).with(:pending_cleanup_policies_count, 1)
perform
end
context 'with load balancing enabled', :db_load_balancing do
it 'reads the count from the replica' do
expect(Gitlab::Database::LoadBalancing::Session.current).to receive(:use_replicas_for_read_queries).and_call_original
perform
end
end
end
end
end
|
