1 files changed, 4 insertions, 1 deletions
diff --git a/gems/gitlab-schema-validation/spec/support/shared_examples/index_validators_shared_examples.rb b/gems/gitlab-schema-validation/spec/support/shared_examples/index_validators_shared_examples.rb
index cc20c0dc765..36a621ffb35 100644
--- a/gems/gitlab-schema-validation/spec/support/shared_examples/index_validators_shared_examples.rb
+++ b/gems/gitlab-schema-validation/spec/support/shared_examples/index_validators_shared_examples.rb
@@ -13,9 +13,12 @@ RSpec.shared_examples 'index validators' do |validator, expected_result|
   end
 
   let(:inconsistency_type) { validator.name }
+  let(:connection_class) { class_double(Class, name: 'ActiveRecord::ConnectionAdapters::PostgreSQLAdapter') }
 
   # rubocop:disable RSpec/VerifiedDoubleReference
-  let(:connection) { instance_double('connection', select_rows: database_indexes, current_schema: 'public') }
+  let(:connection) do
+    instance_double('connection', class: connection_class, select_rows: database_indexes, current_schema: 'public')
+  end
   # rubocop:enable RSpec/VerifiedDoubleReference
 
   let(:schema) { 'public' }