spec/lib/gitlab/redis/db_load_balancing_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

# frozen_string_literal: true

require 'spec_helper'

RSpec.describe Gitlab::Redis::DbLoadBalancing, feature_category: :scalability do
  include_examples "redis_new_instance_shared_examples", 'db_load_balancing', Gitlab::Redis::SharedState
  include_examples "redis_shared_examples"

  describe '#pool' do
    let(:config_new_format_host) { "spec/fixtures/config/redis_new_format_host.yml" }
    let(:config_new_format_socket) { "spec/fixtures/config/redis_new_format_socket.yml" }

    subject { described_class.pool }

    before do
      allow(described_class).to receive(:config_file_name).and_return(config_new_format_host)

      # Override rails root to avoid having our fixtures overwritten by `redis.yml` if it exists
      allow(Gitlab::Redis::SharedState).to receive(:rails_root).and_return(mktmpdir)
      allow(Gitlab::Redis::SharedState).to receive(:config_file_name).and_return(config_new_format_socket)
    end

    around do |example|
      clear_pool
      example.run
    ensure
      clear_pool
    end

    it 'instantiates an instance of MultiStore' do
      subject.with do |redis_instance|
        expect(redis_instance).to be_instance_of(::Gitlab::Redis::MultiStore)

        expect(redis_instance.primary_store.connection[:id]).to eq("redis://test-host:6379/99")
        expect(redis_instance.secondary_store.connection[:id]).to eq("unix:///path/to/redis.sock/0")

        expect(redis_instance.instance_name).to eq('DbLoadBalancing')
      end
    end

    it_behaves_like 'multi store feature flags', :use_primary_and_secondary_stores_for_db_load_balancing,
      :use_primary_store_as_default_for_db_load_balancing
  end

  describe '#raw_config_hash' do
    it 'has a legacy default URL' do
      expect(subject).to receive(:fetch_config).and_return(false)

      expect(subject.send(:raw_config_hash)).to eq(url: 'redis://localhost:6382')
    end
  end
end