1 files changed, 5 insertions, 1 deletions
diff --git a/spec/lib/gitlab/database/reindexing_spec.rb b/spec/lib/gitlab/database/reindexing_spec.rb
index 359e0597f4e..eb78a5fe8ea 100644
--- a/spec/lib/gitlab/database/reindexing_spec.rb
+++ b/spec/lib/gitlab/database/reindexing_spec.rb
@@ -6,12 +6,16 @@ RSpec.describe Gitlab::Database::Reindexing do
   include ExclusiveLeaseHelpers
 
   describe '.perform' do
-    subject { described_class.perform(indexes) }
+    subject { described_class.perform(candidate_indexes) }
 
     let(:coordinator) { instance_double(Gitlab::Database::Reindexing::Coordinator) }
+    let(:index_selection) { instance_double(Gitlab::Database::Reindexing::IndexSelection) }
+    let(:candidate_indexes) { double }
     let(:indexes) { double }
 
     it 'delegates to Coordinator' do
+      expect(Gitlab::Database::Reindexing::IndexSelection).to receive(:new).with(candidate_indexes).and_return(index_selection)
+      expect(index_selection).to receive(:take).with(2).and_return(indexes)
       expect(Gitlab::Database::Reindexing::Coordinator).to receive(:new).with(indexes).and_return(coordinator)
       expect(coordinator).to receive(:perform)