1 files changed, 57 insertions, 0 deletions
diff --git a/spec/workers/packages/cleanup_package_registry_worker_spec.rb b/spec/workers/packages/cleanup_package_registry_worker_spec.rb
new file mode 100644
index 00000000000..e43864975f6
--- /dev/null
+++ b/spec/workers/packages/cleanup_package_registry_worker_spec.rb
@@ -0,0 +1,57 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+RSpec.describe Packages::CleanupPackageRegistryWorker do
+  describe '#perform' do
+    let_it_be_with_reload(:package_files) { create_list(:package_file, 2, :pending_destruction) }
+
+    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
+
+    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)
+
+        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