1 files changed, 32 insertions, 4 deletions
diff --git a/spec/lib/gitlab/database/reindexing/index_selection_spec.rb b/spec/lib/gitlab/database/reindexing/index_selection_spec.rb
index 2ae9037959d..e82a2ab467d 100644
--- a/spec/lib/gitlab/database/reindexing/index_selection_spec.rb
+++ b/spec/lib/gitlab/database/reindexing/index_selection_spec.rb
@@ -2,14 +2,16 @@
 
 require 'spec_helper'
 
-RSpec.describe Gitlab::Database::Reindexing::IndexSelection do
+RSpec.describe Gitlab::Database::Reindexing::IndexSelection, feature_category: :database do
   include Database::DatabaseHelpers
 
   subject { described_class.new(Gitlab::Database::PostgresIndex.all).to_a }
 
+  let(:connection) { ApplicationRecord.connection }
+
   before do
-    swapout_view_for_table(:postgres_index_bloat_estimates)
-    swapout_view_for_table(:postgres_indexes)
+    swapout_view_for_table(:postgres_index_bloat_estimates, connection: connection)
+    swapout_view_for_table(:postgres_indexes, connection: connection)
 
     create_list(:postgres_index, 10, ondisk_size_bytes: 10.gigabytes).each_with_index do |index, i|
       create(:postgres_index_bloat_estimate, index: index, bloat_size_bytes: 2.gigabyte * (i + 1))
@@ -17,7 +19,7 @@ RSpec.describe Gitlab::Database::Reindexing::IndexSelection do
   end
 
   def execute(sql)
-    ActiveRecord::Base.connection.execute(sql)
+    connection.execute(sql)
   end
 
   it 'orders by highest relative bloat first' do
@@ -74,4 +76,30 @@ RSpec.describe Gitlab::Database::Reindexing::IndexSelection do
       expect(subject.map(&:name).sort).to eq(not_recently_reindexed.map(&:name).sort)
     end
   end
+
+  context 'with restricted tables' do
+    let!(:ci_builds) do
+      create(
+        :postgres_index_bloat_estimate,
+        index: create(:postgres_index, ondisk_size_bytes: 100.gigabytes, tablename: 'ci_builds'),
+        bloat_size_bytes: 20.gigabyte
+      )
+    end
+
+    context 'when executed on Fridays', time_travel_to: '2022-12-16T09:44:07Z' do
+      it { expect(subject).not_to include(ci_builds.index) }
+    end
+
+    context 'when executed on Saturdays', time_travel_to: '2022-12-17T09:44:07Z' do
+      it { expect(subject).to include(ci_builds.index) }
+    end
+
+    context 'when executed on Sundays', time_travel_to: '2022-12-18T09:44:07Z' do
+      it { expect(subject).not_to include(ci_builds.index) }
+    end
+
+    context 'when executed on Mondays', time_travel_to: '2022-12-19T09:44:07Z' do
+      it { expect(subject).not_to include(ci_builds.index) }
+    end
+  end
 end