540 lines
18 KiB
Ruby
540 lines
18 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require 'spec_helper'
|
|
|
|
# We stub Gitaly in `spec/support/gitaly.rb` for other tests. We don't want
|
|
# those stubs while testing the GitalyClient itself.
|
|
RSpec.describe Gitlab::GitalyClient do
|
|
def stub_repos_storages(address)
|
|
allow(Gitlab.config.repositories).to receive(:storages).and_return({
|
|
'default' => { 'gitaly_address' => address }
|
|
})
|
|
end
|
|
|
|
describe '.query_time', :request_store do
|
|
it 'increments query times' do
|
|
subject.add_query_time(0.4510004)
|
|
subject.add_query_time(0.3220004)
|
|
|
|
expect(subject.query_time).to eq(0.773001)
|
|
end
|
|
end
|
|
|
|
describe '.long_timeout' do
|
|
context 'default case' do
|
|
it { expect(subject.long_timeout).to eq(6.hours) }
|
|
end
|
|
|
|
context 'running in Puma' do
|
|
before do
|
|
allow(Gitlab::Runtime).to receive(:puma?).and_return(true)
|
|
end
|
|
|
|
it { expect(subject.long_timeout).to eq(55) }
|
|
end
|
|
end
|
|
|
|
describe '.filesystem_id_from_disk' do
|
|
it 'catches errors' do
|
|
[Errno::ENOENT, Errno::EACCES, JSON::ParserError].each do |error|
|
|
stub_file_read(described_class.storage_metadata_file_path('default'), error: error)
|
|
|
|
expect(described_class.filesystem_id_from_disk('default')).to be_nil
|
|
end
|
|
end
|
|
end
|
|
|
|
describe '.filesystem_id' do
|
|
it 'returns an empty string when the relevant storage status is not found in the response' do
|
|
response = double("response")
|
|
allow(response).to receive(:storage_statuses).and_return([])
|
|
allow_next_instance_of(Gitlab::GitalyClient::ServerService) do |instance|
|
|
allow(instance).to receive(:info).and_return(response)
|
|
end
|
|
|
|
expect(described_class.filesystem_id('default')).to eq(nil)
|
|
end
|
|
end
|
|
|
|
context 'when the relevant storage status is not found' do
|
|
before do
|
|
response = double('response')
|
|
allow(response).to receive(:storage_statuses).and_return([])
|
|
allow_next_instance_of(Gitlab::GitalyClient::ServerService) do |instance|
|
|
allow(instance).to receive(:disk_statistics).and_return(response)
|
|
expect(instance).to receive(:storage_disk_statistics)
|
|
end
|
|
end
|
|
|
|
describe '.filesystem_disk_available' do
|
|
it 'returns nil when the relevant storage status is not found in the response' do
|
|
expect(described_class.filesystem_disk_available('default')).to eq(nil)
|
|
end
|
|
end
|
|
|
|
describe '.filesystem_disk_used' do
|
|
it 'returns nil when the relevant storage status is not found in the response' do
|
|
expect(described_class.filesystem_disk_used('default')).to eq(nil)
|
|
end
|
|
end
|
|
end
|
|
|
|
context 'when the relevant storage status is found' do
|
|
let(:disk_available) { 42 }
|
|
let(:disk_used) { 42 }
|
|
let(:storage_status) { double('storage_status') }
|
|
|
|
before do
|
|
allow(storage_status).to receive(:storage_name).and_return('default')
|
|
allow(storage_status).to receive(:used).and_return(disk_used)
|
|
allow(storage_status).to receive(:available).and_return(disk_available)
|
|
response = double('response')
|
|
allow(response).to receive(:storage_statuses).and_return([storage_status])
|
|
allow_next_instance_of(Gitlab::GitalyClient::ServerService) do |instance|
|
|
allow(instance).to receive(:disk_statistics).and_return(response)
|
|
end
|
|
expect_next_instance_of(Gitlab::GitalyClient::ServerService) do |instance|
|
|
expect(instance).to receive(:storage_disk_statistics).and_return(storage_status)
|
|
end
|
|
end
|
|
|
|
describe '.filesystem_disk_available' do
|
|
it 'returns disk available when the relevant storage status is found in the response' do
|
|
expect(storage_status).to receive(:available)
|
|
expect(described_class.filesystem_disk_available('default')).to eq(disk_available)
|
|
end
|
|
end
|
|
|
|
describe '.filesystem_disk_used' do
|
|
it 'returns disk used when the relevant storage status is found in the response' do
|
|
expect(storage_status).to receive(:used)
|
|
expect(described_class.filesystem_disk_used('default')).to eq(disk_used)
|
|
end
|
|
end
|
|
end
|
|
|
|
describe '.stub_class' do
|
|
it 'returns the gRPC health check stub' do
|
|
expect(described_class.stub_class(:health_check)).to eq(::Grpc::Health::V1::Health::Stub)
|
|
end
|
|
|
|
it 'returns a Gitaly stub' do
|
|
expect(described_class.stub_class(:ref_service)).to eq(::Gitaly::RefService::Stub)
|
|
end
|
|
end
|
|
|
|
describe '.stub_address' do
|
|
it 'returns the same result after being called multiple times' do
|
|
address = 'tcp://localhost:9876'
|
|
stub_repos_storages address
|
|
|
|
2.times do
|
|
expect(described_class.stub_address('default')).to eq('localhost:9876')
|
|
end
|
|
end
|
|
end
|
|
|
|
describe '.stub_creds' do
|
|
it 'returns :this_channel_is_insecure if unix' do
|
|
address = 'unix:/tmp/gitaly.sock'
|
|
stub_repos_storages address
|
|
|
|
expect(described_class.stub_creds('default')).to eq(:this_channel_is_insecure)
|
|
end
|
|
|
|
it 'returns :this_channel_is_insecure if tcp' do
|
|
address = 'tcp://localhost:9876'
|
|
stub_repos_storages address
|
|
|
|
expect(described_class.stub_creds('default')).to eq(:this_channel_is_insecure)
|
|
end
|
|
|
|
it 'returns Credentials object if tls' do
|
|
address = 'tls://localhost:9876'
|
|
stub_repos_storages address
|
|
|
|
expect(described_class.stub_creds('default')).to be_a(GRPC::Core::ChannelCredentials)
|
|
end
|
|
end
|
|
|
|
describe '.stub' do
|
|
# Notice that this is referring to gRPC "stubs", not rspec stubs
|
|
before do
|
|
described_class.clear_stubs!
|
|
end
|
|
|
|
context 'when passed a UNIX socket address' do
|
|
it 'passes the address as-is to GRPC' do
|
|
address = 'unix:/tmp/gitaly.sock'
|
|
stub_repos_storages address
|
|
|
|
expect(Gitaly::CommitService::Stub).to receive(:new).with(address, any_args)
|
|
|
|
described_class.stub(:commit_service, 'default')
|
|
end
|
|
end
|
|
|
|
context 'when passed a TLS address' do
|
|
it 'strips tls:// prefix before passing it to GRPC::Core::Channel initializer' do
|
|
address = 'localhost:9876'
|
|
prefixed_address = "tls://#{address}"
|
|
stub_repos_storages prefixed_address
|
|
|
|
expect(Gitaly::CommitService::Stub).to receive(:new).with(address, any_args)
|
|
|
|
described_class.stub(:commit_service, 'default')
|
|
end
|
|
end
|
|
|
|
context 'when passed a TCP address' do
|
|
it 'strips tcp:// prefix before passing it to GRPC::Core::Channel initializer' do
|
|
address = 'localhost:9876'
|
|
prefixed_address = "tcp://#{address}"
|
|
stub_repos_storages prefixed_address
|
|
|
|
expect(Gitaly::CommitService::Stub).to receive(:new).with(address, any_args)
|
|
|
|
described_class.stub(:commit_service, 'default')
|
|
end
|
|
end
|
|
end
|
|
|
|
describe '.can_use_disk?' do
|
|
it 'properly caches a false result' do
|
|
# spec_helper stubs this globally
|
|
allow(described_class).to receive(:can_use_disk?).and_call_original
|
|
expect(described_class).to receive(:filesystem_id).once
|
|
expect(described_class).to receive(:filesystem_id_from_disk).once
|
|
|
|
2.times do
|
|
described_class.can_use_disk?('unknown')
|
|
end
|
|
end
|
|
end
|
|
|
|
describe '.connection_data' do
|
|
it 'returns connection data' do
|
|
address = 'tcp://localhost:9876'
|
|
stub_repos_storages address
|
|
|
|
expect(described_class.connection_data('default')).to eq({ 'address' => address, 'token' => 'secret' })
|
|
end
|
|
end
|
|
|
|
describe 'allow_n_plus_1_calls' do
|
|
context 'when RequestStore is enabled', :request_store do
|
|
it 'returns the result of the allow_n_plus_1_calls block' do
|
|
expect(described_class.allow_n_plus_1_calls { "result" }).to eq("result")
|
|
end
|
|
end
|
|
|
|
context 'when RequestStore is not active' do
|
|
it 'returns the result of the allow_n_plus_1_calls block' do
|
|
expect(described_class.allow_n_plus_1_calls { "something" }).to eq("something")
|
|
end
|
|
end
|
|
end
|
|
|
|
describe '.request_kwargs' do
|
|
it 'sets the gitaly-session-id in the metadata' do
|
|
results = described_class.request_kwargs('default', timeout: 1)
|
|
expect(results[:metadata]).to include('gitaly-session-id')
|
|
end
|
|
|
|
context 'when RequestStore is not enabled' do
|
|
it 'sets a different gitaly-session-id per request' do
|
|
gitaly_session_id = described_class.request_kwargs('default', timeout: 1)[:metadata]['gitaly-session-id']
|
|
|
|
expect(described_class.request_kwargs('default', timeout: 1)[:metadata]['gitaly-session-id']).not_to eq(gitaly_session_id)
|
|
end
|
|
end
|
|
|
|
context 'when RequestStore is enabled', :request_store do
|
|
it 'sets the same gitaly-session-id on every outgoing request metadata' do
|
|
gitaly_session_id = described_class.request_kwargs('default', timeout: 1)[:metadata]['gitaly-session-id']
|
|
|
|
3.times do
|
|
expect(described_class.request_kwargs('default', timeout: 1)[:metadata]['gitaly-session-id']).to eq(gitaly_session_id)
|
|
end
|
|
end
|
|
end
|
|
|
|
context 'gitlab_git_env' do
|
|
let(:policy) { 'gitaly-route-repository-accessor-policy' }
|
|
|
|
context 'when RequestStore is disabled' do
|
|
it 'does not force-route to primary' do
|
|
expect(described_class.request_kwargs('default', timeout: 1)[:metadata][policy]).to be_nil
|
|
end
|
|
end
|
|
|
|
context 'when RequestStore is enabled without git_env', :request_store do
|
|
it 'does not force-orute to primary' do
|
|
expect(described_class.request_kwargs('default', timeout: 1)[:metadata][policy]).to be_nil
|
|
end
|
|
end
|
|
|
|
context 'when RequestStore is enabled with empty git_env', :request_store do
|
|
before do
|
|
Gitlab::SafeRequestStore[:gitlab_git_env] = {}
|
|
end
|
|
|
|
it 'disables force-routing to primary' do
|
|
expect(described_class.request_kwargs('default', timeout: 1)[:metadata][policy]).to be_nil
|
|
end
|
|
end
|
|
|
|
context 'when RequestStore is enabled with populated git_env', :request_store do
|
|
before do
|
|
Gitlab::SafeRequestStore[:gitlab_git_env] = {
|
|
"GIT_OBJECT_DIRECTORY_RELATIVE" => "foo/bar"
|
|
}
|
|
end
|
|
|
|
it 'enables force-routing to primary' do
|
|
expect(described_class.request_kwargs('default', timeout: 1)[:metadata][policy]).to eq('primary-only')
|
|
end
|
|
end
|
|
end
|
|
|
|
context 'deadlines', :request_store do
|
|
let(:request_deadline) { real_time + 10.0 }
|
|
|
|
before do
|
|
allow(Gitlab::RequestContext.instance).to receive(:request_deadline).and_return(request_deadline)
|
|
end
|
|
|
|
it 'includes the deadline information' do
|
|
kword_args = described_class.request_kwargs('default', timeout: 2)
|
|
|
|
expect(kword_args[:deadline])
|
|
.to be_within(1).of(real_time + 2)
|
|
expect(kword_args[:metadata][:deadline_type]).to eq("regular")
|
|
end
|
|
|
|
it 'limits the deadline do the request deadline if that is closer', :aggregate_failures do
|
|
kword_args = described_class.request_kwargs('default', timeout: 15)
|
|
|
|
expect(kword_args[:deadline]).to eq(request_deadline)
|
|
expect(kword_args[:metadata][:deadline_type]).to eq("limited")
|
|
end
|
|
|
|
it 'does not limit calls in sidekiq' do
|
|
expect(Sidekiq).to receive(:server?).and_return(true)
|
|
|
|
kword_args = described_class.request_kwargs('default', timeout: 6.hours.to_i)
|
|
|
|
expect(kword_args[:deadline]).to be_within(1).of(real_time + 6.hours.to_i)
|
|
expect(kword_args[:metadata][:deadline_type]).to be_nil
|
|
end
|
|
|
|
it 'does not limit calls in sidekiq when allowed unlimited' do
|
|
expect(Sidekiq).to receive(:server?).and_return(true)
|
|
|
|
kword_args = described_class.request_kwargs('default', timeout: 0)
|
|
|
|
expect(kword_args[:deadline]).to be_nil
|
|
expect(kword_args[:metadata][:deadline_type]).to be_nil
|
|
end
|
|
|
|
it 'includes only the deadline specified by the timeout when there was no deadline' do
|
|
allow(Gitlab::RequestContext.instance).to receive(:request_deadline).and_return(nil)
|
|
kword_args = described_class.request_kwargs('default', timeout: 6.hours.to_i)
|
|
|
|
expect(kword_args[:deadline]).to be_within(1).of(Gitlab::Metrics::System.real_time + 6.hours.to_i)
|
|
expect(kword_args[:metadata][:deadline_type]).to be_nil
|
|
end
|
|
|
|
def real_time
|
|
Gitlab::Metrics::System.real_time
|
|
end
|
|
end
|
|
end
|
|
|
|
describe 'enforce_gitaly_request_limits?' do
|
|
def call_gitaly(count = 1)
|
|
(1..count).each do
|
|
described_class.enforce_gitaly_request_limits(:test)
|
|
end
|
|
end
|
|
|
|
context 'when RequestStore is enabled and the maximum number of calls is not enforced by a feature flag', :request_store do
|
|
before do
|
|
stub_feature_flags(gitaly_enforce_requests_limits: false)
|
|
end
|
|
|
|
it 'allows up the maximum number of allowed calls' do
|
|
expect { call_gitaly(Gitlab::GitalyClient::MAXIMUM_GITALY_CALLS) }.not_to raise_error
|
|
end
|
|
|
|
it 'allows the maximum number of calls to be exceeded if GITALY_DISABLE_REQUEST_LIMITS is set' do
|
|
stub_env('GITALY_DISABLE_REQUEST_LIMITS', 'true')
|
|
|
|
expect { call_gitaly(Gitlab::GitalyClient::MAXIMUM_GITALY_CALLS + 1) }.not_to raise_error
|
|
end
|
|
|
|
context 'when the maximum number of calls has been reached' do
|
|
before do
|
|
call_gitaly(Gitlab::GitalyClient::MAXIMUM_GITALY_CALLS)
|
|
end
|
|
|
|
it 'fails on the next call' do
|
|
expect { call_gitaly(1) }.to raise_error(Gitlab::GitalyClient::TooManyInvocationsError)
|
|
end
|
|
end
|
|
|
|
it 'allows the maximum number of calls to be exceeded within an allow_n_plus_1_calls block' do
|
|
expect do
|
|
described_class.allow_n_plus_1_calls do
|
|
call_gitaly(Gitlab::GitalyClient::MAXIMUM_GITALY_CALLS + 1)
|
|
end
|
|
end.not_to raise_error
|
|
end
|
|
|
|
context 'when the maximum number of calls has been reached within an allow_n_plus_1_calls block' do
|
|
before do
|
|
described_class.allow_n_plus_1_calls do
|
|
call_gitaly(Gitlab::GitalyClient::MAXIMUM_GITALY_CALLS)
|
|
end
|
|
end
|
|
|
|
it 'allows up to the maximum number of calls outside of an allow_n_plus_1_calls block' do
|
|
expect { call_gitaly(Gitlab::GitalyClient::MAXIMUM_GITALY_CALLS) }.not_to raise_error
|
|
end
|
|
|
|
it 'does not allow the maximum number of calls to be exceeded outside of an allow_n_plus_1_calls block' do
|
|
expect { call_gitaly(Gitlab::GitalyClient::MAXIMUM_GITALY_CALLS + 1) }.to raise_error(Gitlab::GitalyClient::TooManyInvocationsError)
|
|
end
|
|
end
|
|
end
|
|
|
|
context 'in production and when RequestStore is enabled', :request_store do
|
|
before do
|
|
stub_rails_env('production')
|
|
end
|
|
|
|
context 'when the maximum number of calls is enforced by a feature flag' do
|
|
before do
|
|
stub_feature_flags(gitaly_enforce_requests_limits: true)
|
|
end
|
|
|
|
it 'does not allow the maximum number of calls to be exceeded' do
|
|
expect { call_gitaly(Gitlab::GitalyClient::MAXIMUM_GITALY_CALLS + 1) }.to raise_error(Gitlab::GitalyClient::TooManyInvocationsError)
|
|
end
|
|
end
|
|
|
|
context 'when the maximum number of calls is not enforced by a feature flag' do
|
|
before do
|
|
stub_feature_flags(gitaly_enforce_requests_limits: false)
|
|
end
|
|
|
|
it 'allows the maximum number of calls to be exceeded' do
|
|
expect { call_gitaly(Gitlab::GitalyClient::MAXIMUM_GITALY_CALLS + 1) }.not_to raise_error
|
|
end
|
|
end
|
|
end
|
|
|
|
context 'when RequestStore is not active' do
|
|
it 'does not raise errors when the maximum number of allowed calls is exceeded' do
|
|
expect { call_gitaly(Gitlab::GitalyClient::MAXIMUM_GITALY_CALLS + 2) }.not_to raise_error
|
|
end
|
|
|
|
it 'does not fail when the maximum number of calls is exceeded within an allow_n_plus_1_calls block' do
|
|
expect do
|
|
described_class.allow_n_plus_1_calls do
|
|
call_gitaly(Gitlab::GitalyClient::MAXIMUM_GITALY_CALLS + 1)
|
|
end
|
|
end.not_to raise_error
|
|
end
|
|
end
|
|
end
|
|
|
|
describe 'get_request_count' do
|
|
context 'when RequestStore is enabled', :request_store do
|
|
context 'when enforce_gitaly_request_limits is called outside of allow_n_plus_1_calls blocks' do
|
|
before do
|
|
described_class.enforce_gitaly_request_limits(:call)
|
|
end
|
|
|
|
it 'counts gitaly calls' do
|
|
expect(described_class.get_request_count).to eq(1)
|
|
end
|
|
end
|
|
|
|
context 'when enforce_gitaly_request_limits is called inside and outside of allow_n_plus_1_calls blocks' do
|
|
before do
|
|
described_class.enforce_gitaly_request_limits(:call)
|
|
described_class.allow_n_plus_1_calls do
|
|
described_class.enforce_gitaly_request_limits(:call)
|
|
end
|
|
end
|
|
|
|
it 'counts gitaly calls' do
|
|
expect(described_class.get_request_count).to eq(2)
|
|
end
|
|
end
|
|
|
|
context 'when reset_counts is called' do
|
|
before do
|
|
described_class.enforce_gitaly_request_limits(:call)
|
|
described_class.reset_counts
|
|
end
|
|
|
|
it 'resets counts' do
|
|
expect(described_class.get_request_count).to eq(0)
|
|
end
|
|
end
|
|
end
|
|
|
|
context 'when RequestStore is not active' do
|
|
before do
|
|
described_class.enforce_gitaly_request_limits(:call)
|
|
end
|
|
|
|
it 'returns zero' do
|
|
expect(described_class.get_request_count).to eq(0)
|
|
end
|
|
end
|
|
end
|
|
|
|
describe 'timeouts' do
|
|
context 'with default values' do
|
|
before do
|
|
stub_application_setting(gitaly_timeout_default: 55)
|
|
stub_application_setting(gitaly_timeout_medium: 30)
|
|
stub_application_setting(gitaly_timeout_fast: 10)
|
|
end
|
|
|
|
it 'returns expected values' do
|
|
expect(described_class.default_timeout).to be(55)
|
|
expect(described_class.medium_timeout).to be(30)
|
|
expect(described_class.fast_timeout).to be(10)
|
|
end
|
|
end
|
|
end
|
|
|
|
describe 'Peek Performance bar details' do
|
|
let(:gitaly_server) { Gitaly::Server.all.first }
|
|
|
|
before do
|
|
Gitlab::SafeRequestStore[:peek_enabled] = true
|
|
end
|
|
|
|
context 'when the request store is active', :request_store do
|
|
it 'records call details if a RPC is called' do
|
|
gitaly_server.server_version
|
|
|
|
expect(described_class.list_call_details).not_to be_empty
|
|
expect(described_class.list_call_details.size).to be(1)
|
|
end
|
|
end
|
|
|
|
context 'when no request store is active' do
|
|
it 'records nothing' do
|
|
gitaly_server.server_version
|
|
|
|
expect(described_class.list_call_details).to be_empty
|
|
end
|
|
end
|
|
end
|
|
end
|