diff options
Diffstat (limited to 'spec/support_specs/database/multiple_databases_spec.rb')
-rw-r--r-- | spec/support_specs/database/multiple_databases_spec.rb | 22 |
1 files changed, 22 insertions, 0 deletions
diff --git a/spec/support_specs/database/multiple_databases_spec.rb b/spec/support_specs/database/multiple_databases_spec.rb index b4cfa253813..0b019462077 100644 --- a/spec/support_specs/database/multiple_databases_spec.rb +++ b/spec/support_specs/database/multiple_databases_spec.rb @@ -3,6 +3,28 @@ require 'spec_helper' RSpec.describe 'Database::MultipleDatabases' do + let(:query) do + <<~SQL + WITH cte AS #{Gitlab::Database::AsWithMaterialized.materialized_if_supported} (SELECT 1) SELECT 1; + SQL + end + + it 'preloads database version for ApplicationRecord' do + counts = ActiveRecord::QueryRecorder + .new { ApplicationRecord.connection.execute(query) } + .count + + expect(counts).to eq(1) + end + + it 'preloads database version for Ci::ApplicationRecord' do + counts = ActiveRecord::QueryRecorder + .new { Ci::ApplicationRecord.connection.execute(query) } + .count + + expect(counts).to eq(1) + end + describe '.with_reestablished_active_record_base' do context 'when doing establish_connection' do context 'on ActiveRecord::Base' do |