spec/tasks/gitlab/db/decomposition/connection_status_rake_spec.rb


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

# frozen_string_literal: true

require 'rake_helper'

RSpec.describe 'gitlab:db:decomposition:connection_status', feature_category: :cell do
  let(:max_connections) { 500 }
  let(:current_connections) { 300 }

  subject { run_rake_task('gitlab:db:decomposition:connection_status') }

  before(:all) do
    Rake.application.rake_require 'tasks/gitlab/db/decomposition/connection_status'
  end

  before do
    allow(ApplicationRecord.connection).to receive(:select_one).with(any_args).and_return(
      { "active" => current_connections, "max" => max_connections }
    )
  end

  context 'when separate ci database is not configured' do
    before do
      skip_if_multiple_databases_are_setup
    end

    context "when PostgreSQL max_connections is too low" do
      it 'suggests to increase it' do
        expect { subject }.to output(
          "Currently using #{current_connections} connections out of #{max_connections} max_connections,\n" \
          "which may run out when you switch to two database connections.\n\n" \
          "Consider increasing PostgreSQL 'max_connections' setting.\n" \
          "Depending on the installation method, there are different ways to\n" \
          "increase that setting. Please consult the GitLab documentation.\n"
        ).to_stdout
      end
    end

    context "when PostgreSQL max_connections is high enough" do
      let(:max_connections) { 1000 }

      it 'only shows current status' do
        expect { subject }.to output(
          "Currently using #{current_connections} connections out of #{max_connections} max_connections,\n" \
          "which is enough for running GitLab using two database connections.\n"
        ).to_stdout
      end
    end
  end

  context 'when separate ci database is configured' do
    before do
      skip_if_multiple_databases_not_setup(:ci)
    end

    it "does not show connection information" do
      expect { subject }.to output(
        "GitLab database already running on two connections\n"
      ).to_stdout
    end
  end
end