1 files changed, 4 insertions, 1 deletions
diff --git a/gems/gitlab-schema-validation/spec/support/shared_examples/foreign_key_validators_shared_examples.rb b/gems/gitlab-schema-validation/spec/support/shared_examples/foreign_key_validators_shared_examples.rb
index 1f33c8bd760..8de9c165f8a 100644
--- a/gems/gitlab-schema-validation/spec/support/shared_examples/foreign_key_validators_shared_examples.rb
+++ b/gems/gitlab-schema-validation/spec/support/shared_examples/foreign_key_validators_shared_examples.rb
@@ -10,8 +10,11 @@ RSpec.shared_examples 'foreign key validators' do |validator, expected_result|
   let(:inconsistency_type) { validator.to_s }
   let(:database_name) { 'main' }
   let(:schema) { 'public' }
+  let(:connection_class) { class_double(Class, name: 'ActiveRecord::ConnectionAdapters::PostgreSQLAdapter') }
   # rubocop:disable RSpec/VerifiedDoubleReference
-  let(:connection) { instance_double('connection', exec_query: database_query, current_schema: 'public') }
+  let(:connection) do
+    instance_double('connection', class: connection_class, exec_query: database_query, current_schema: 'public')
+  end
   # rubocop:enable RSpec/VerifiedDoubleReference
 
   let(:database) { Gitlab::Schema::Validation::Sources::Database.new(connection) }