45 lines
		
	
	
		
			1.6 KiB
		
	
	
	
		
			Ruby
		
	
	
	
			
		
		
	
	
			45 lines
		
	
	
		
			1.6 KiB
		
	
	
	
		
			Ruby
		
	
	
	
| # frozen_string_literal: true
 | |
| 
 | |
| require 'spec_helper'
 | |
| 
 | |
| RSpec.describe Gitlab::Redis::DbLoadBalancing, feature_category: :scalability do
 | |
|   include_examples "redis_new_instance_shared_examples", 'db_load_balancing', Gitlab::Redis::SharedState
 | |
|   include_examples "redis_shared_examples"
 | |
| 
 | |
|   describe '#pool' do
 | |
|     let(:config_new_format_host) { "spec/fixtures/config/redis_new_format_host.yml" }
 | |
|     let(:config_new_format_socket) { "spec/fixtures/config/redis_new_format_socket.yml" }
 | |
| 
 | |
|     subject { described_class.pool }
 | |
| 
 | |
|     before do
 | |
|       allow(described_class).to receive(:config_file_name).and_return(config_new_format_host)
 | |
| 
 | |
|       # Override rails root to avoid having our fixtures overwritten by `redis.yml` if it exists
 | |
|       allow(Gitlab::Redis::SharedState).to receive(:rails_root).and_return(mktmpdir)
 | |
|       allow(Gitlab::Redis::SharedState).to receive(:config_file_name).and_return(config_new_format_socket)
 | |
|     end
 | |
| 
 | |
|     around do |example|
 | |
|       clear_pool
 | |
|       example.run
 | |
|     ensure
 | |
|       clear_pool
 | |
|     end
 | |
| 
 | |
|     it 'instantiates an instance of MultiStore' do
 | |
|       subject.with do |redis_instance|
 | |
|         expect(redis_instance).to be_instance_of(::Gitlab::Redis::MultiStore)
 | |
| 
 | |
|         expect(redis_instance.primary_store.connection[:id]).to eq("redis://test-host:6379/99")
 | |
|         expect(redis_instance.secondary_store.connection[:id]).to eq("unix:///path/to/redis.sock/0")
 | |
| 
 | |
|         expect(redis_instance.instance_name).to eq('DbLoadBalancing')
 | |
|       end
 | |
|     end
 | |
| 
 | |
|     it_behaves_like 'multi store feature flags', :use_primary_and_secondary_stores_for_db_load_balancing,
 | |
|       :use_primary_store_as_default_for_db_load_balancing
 | |
|   end
 | |
| end
 |