Welcome to mirror list, hosted at ThFree Co, Russian Federation.

database_model_spec.rb « backup « lib « spec - gitlab.com/gitlab-org/gitlab-foss.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
blob: 5758ad2c1aa3a599d1d0cef2ef57584d3bccd5c1 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
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