20 lines
530 B
Ruby
20 lines
530 B
Ruby
# 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
|