Add latest changes from gitlab-org/gitlab@14-10-stable-eev14.10.0-rc42

1 files changed, 67 insertions, 0 deletions

diff --git a/spec/workers/database/ci_namespace_mirrors_consistency_check_worker_spec.rb b/spec/workers/database/ci_namespace_mirrors_consistency_check_worker_spec.rb
new file mode 100644
index 00000000000..116026ea8f7
--- /dev/null
+++ b/spec/workers/database/ci_namespace_mirrors_consistency_check_worker_spec.rb
@@ -0,0 +1,67 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+RSpec.describe Database::CiNamespaceMirrorsConsistencyCheckWorker do
+  let(:worker) { described_class.new }
+
+  describe '#perform' do
+    context 'feature flag is disabled' do
+      before do
+        stub_feature_flags(ci_namespace_mirrors_consistency_check: false)
+      end
+
+      it 'does not perform the consistency check on namespaces' do
+        expect(Database::ConsistencyCheckService).not_to receive(:new)
+        expect(worker).not_to receive(:log_extra_metadata_on_done)
+        worker.perform
+      end
+    end
+
+    context 'feature flag is enabled' do
+      before do
+        stub_feature_flags(ci_namespace_mirrors_consistency_check: true)
+      end
+
+      it 'executes the consistency check on namespaces' do
+        expect(Database::ConsistencyCheckService).to receive(:new).and_call_original
+        expected_result = { batches: 0, matches: 0, mismatches: 0, mismatches_details: [] }
+        expect(worker).to receive(:log_extra_metadata_on_done).with(:results, expected_result)
+        worker.perform
+      end
+    end
+
+    context 'logs should contain the detailed mismatches' do
+      let(:first_namespace) { Namespace.all.order(:id).limit(1).first }
+      let(:missing_namespace) { Namespace.all.order(:id).limit(2).last }
+
+      before do
+        redis_shared_state_cleanup!
+        stub_feature_flags(ci_namespace_mirrors_consistency_check: true)
+        create_list(:namespace, 10) # This will also create Ci::NameSpaceMirror objects
+        missing_namespace.delete
+
+        allow_next_instance_of(Database::ConsistencyCheckService) do |instance|
+          allow(instance).to receive(:random_start_id).and_return(Namespace.first.id)
+        end
+      end
+
+      it 'reports the differences to the logs' do
+        expected_result = {
+          batches: 1,
+          matches: 9,
+          mismatches: 1,
+          mismatches_details: [{
+            id: missing_namespace.id,
+            source_table: nil,
+            target_table: [missing_namespace.traversal_ids]
+          }],
+          start_id: first_namespace.id,
+          next_start_id: first_namespace.id # The batch size > number of namespaces
+        }
+        expect(worker).to receive(:log_extra_metadata_on_done).with(:results, expected_result)
+        worker.perform
+      end
+    end
+  end
+end