diff --git a/spec/support/database_load_balancing.rb b/spec/support/database_load_balancing.rb index cc176dce3c944efb2b6c3c1ccd786132ec6b5c69..75432cd7002ea87a82cd4c10872b964f586bbd3e 100644 --- a/spec/support/database_load_balancing.rb +++ b/spec/support/database_load_balancing.rb @@ -23,6 +23,10 @@ Gitlab::Database::LoadBalancing::SessionMap.clear_session redis_shared_state_cleanup! + Gitlab::Database::LoadBalancing.base_models.each do |model| + model.load_balancer.host_list.hosts.each(&:force_disconnect!) + end + old_proxies.each do |model, proxy| model.load_balancer, model.connection, model.sticking = proxy end