spec/support/database_load_balancing.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.configure do |config|
  config.before(:each, :db_load_balancing) do
    allow(Gitlab::Database::LoadBalancing).to receive(:enable?).and_return(true)

    proxy = ::Gitlab::Database::LoadBalancing::ConnectionProxy.new([Gitlab::Database.main.config['host']])

    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