21 lines
698 B
Ruby
21 lines
698 B
Ruby
# frozen_string_literal: true
|
|
|
|
RSpec.configure do |config|
|
|
config.before(:each, :db_load_balancing) do
|
|
allow(Gitlab::Database::LoadBalancing).to receive(:enable?).and_return(true)
|
|
|
|
lb = ::Gitlab::Database::LoadBalancing::LoadBalancer
|
|
.new([Gitlab::Database.main.config['host']])
|
|
proxy = ::Gitlab::Database::LoadBalancing::ConnectionProxy.new(lb)
|
|
|
|
allow(ActiveRecord::Base).to receive(:load_balancing_proxy).and_return(proxy)
|
|
|
|
::Gitlab::Database::LoadBalancing::Session.clear_session
|
|
redis_shared_state_cleanup!
|
|
end
|
|
|
|
config.after(:each, :db_load_balancing) do
|
|
::Gitlab::Database::LoadBalancing::Session.clear_session
|
|
redis_shared_state_cleanup!
|
|
end
|
|
end
|