blob: 014575e8a820f24f0cdf743723d48b3eefb0521c (
plain)
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
|
# frozen_string_literal: true
RSpec.configure do |config|
config.around(:each, :database_replica) do |example|
old_proxies = []
Gitlab::Database::LoadBalancing.base_models.each do |model|
config = Gitlab::Database::LoadBalancing::Configuration
.new(model, [model.connection_db_config.configuration_hash[:host]])
lb = Gitlab::Database::LoadBalancing::LoadBalancer.new(config)
old_proxies << [model, model.connection]
model.connection =
Gitlab::Database::LoadBalancing::ConnectionProxy.new(lb)
end
Gitlab::Database::LoadBalancing::Session.clear_session
redis_shared_state_cleanup!
example.run
Gitlab::Database::LoadBalancing::Session.clear_session
redis_shared_state_cleanup!
old_proxies.each do |(model, proxy)|
model.connection = proxy
end
end
end
|
