diff options
Diffstat (limited to 'spec/lib/backup/database_model_spec.rb')
-rw-r--r-- | spec/lib/backup/database_model_spec.rb | 82 |
1 files changed, 82 insertions, 0 deletions
diff --git a/spec/lib/backup/database_model_spec.rb b/spec/lib/backup/database_model_spec.rb new file mode 100644 index 00000000000..5758ad2c1aa --- /dev/null +++ b/spec/lib/backup/database_model_spec.rb @@ -0,0 +1,82 @@ +# frozen_string_literal: true + +require 'spec_helper' + +RSpec.describe Backup::DatabaseModel, :reestablished_active_record_base, feature_category: :backup_restore do + let(:gitlab_database_name) { 'main' } + + describe '#connection' do + subject { described_class.new(gitlab_database_name).connection } + + it 'an instance of a ActiveRecord::Base.connection' do + subject.is_a? ActiveRecord::Base.connection.class # rubocop:disable Database/MultipleDatabases + end + end + + describe '#config' do + let(:application_config) do + { + adapter: 'postgresql', + host: 'some_host', + port: '5432' + } + end + + subject { described_class.new(gitlab_database_name).config } + + before do + allow( + Gitlab::Database.database_base_models_with_gitlab_shared[gitlab_database_name].connection_db_config + ).to receive(:configuration_hash).and_return(application_config) + end + + context 'when no GITLAB_BACKUP_PG* variables are set' do + it 'ActiveRecord backup configuration is expected to equal application configuration' do + expect(subject[:activerecord]).to eq(application_config) + end + + it 'PostgreSQL ENV is expected to equal application configuration' do + expect(subject[:pg_env]).to eq( + { + 'PGHOST' => application_config[:host], + 'PGPORT' => application_config[:port] + } + ) + end + end + + context 'when GITLAB_BACKUP_PG* variables are set' do + using RSpec::Parameterized::TableSyntax + + where(:env_variable, :overridden_value) do + 'GITLAB_BACKUP_PGHOST' | 'test.invalid.' + 'GITLAB_BACKUP_PGUSER' | 'some_user' + 'GITLAB_BACKUP_PGPORT' | '1543' + 'GITLAB_BACKUP_PGPASSWORD' | 'secret' + 'GITLAB_BACKUP_PGSSLMODE' | 'allow' + 'GITLAB_BACKUP_PGSSLKEY' | 'some_key' + 'GITLAB_BACKUP_PGSSLCERT' | '/path/to/cert' + 'GITLAB_BACKUP_PGSSLROOTCERT' | '/path/to/root/cert' + 'GITLAB_BACKUP_PGSSLCRL' | '/path/to/crl' + 'GITLAB_BACKUP_PGSSLCOMPRESSION' | '1' + end + + with_them do + let(:pg_env) { env_variable[/GITLAB_BACKUP_(\w+)/, 1] } + let(:active_record_key) { described_class::SUPPORTED_OVERRIDES.invert[pg_env] } + + before do + stub_env(env_variable, overridden_value) + end + + it 'ActiveRecord backup configuration overrides application configuration' do + expect(subject[:activerecord]).to eq(application_config.merge(active_record_key => overridden_value)) + end + + it 'PostgreSQL ENV overrides application configuration' do + expect(subject[:pg_env]).to include({ pg_env => overridden_value }) + end + end + end + end +end |