spec/support/shared_contexts/load_balancing_configuration_shared_context.rb


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19

# frozen_string_literal: true

RSpec.shared_context 'clear DB Load Balancing configuration' do
  def clear_load_balancing_configuration
    proxy = ::Gitlab::Database::LoadBalancing.instance_variable_get(:@proxy)
    proxy.load_balancer.release_host if proxy
    ::Gitlab::Database::LoadBalancing.instance_variable_set(:@proxy, nil)

    ::Gitlab::Database::LoadBalancing::Session.clear_session
  end

  around do |example|
    clear_load_balancing_configuration

    example.run

    clear_load_balancing_configuration
  end
end