diff options
Diffstat (limited to 'spec/lib/gitlab/consul')
-rw-r--r-- | spec/lib/gitlab/consul/internal_spec.rb | 139 |
1 files changed, 139 insertions, 0 deletions
diff --git a/spec/lib/gitlab/consul/internal_spec.rb b/spec/lib/gitlab/consul/internal_spec.rb new file mode 100644 index 00000000000..5889dd8b41d --- /dev/null +++ b/spec/lib/gitlab/consul/internal_spec.rb @@ -0,0 +1,139 @@ +# frozen_string_literal: true + +require 'spec_helper' + +RSpec.describe Gitlab::Consul::Internal do + let(:api_url) { 'http://127.0.0.1:8500' } + + let(:consul_settings) do + { + api_url: api_url + } + end + + before do + stub_config(consul: consul_settings) + end + + describe '.api_url' do + it 'returns correct value' do + expect(described_class.api_url).to eq(api_url) + end + + context 'when consul setting is not present in gitlab.yml' do + before do + allow(Gitlab.config).to receive(:consul).and_raise(Settingslogic::MissingSetting) + end + + it 'does not fail' do + expect(described_class.api_url).to be_nil + end + end + end + + shared_examples 'handles failure response' do + it 'raises Gitlab::Consul::Internal::SocketError when SocketError is rescued' do + stub_consul_discover_prometheus.to_raise(::SocketError) + + expect { subject } + .to raise_error(described_class::SocketError) + end + + it 'raises Gitlab::Consul::Internal::SSLError when OpenSSL::SSL::SSLError is rescued' do + stub_consul_discover_prometheus.to_raise(OpenSSL::SSL::SSLError) + + expect { subject } + .to raise_error(described_class::SSLError) + end + + it 'raises Gitlab::Consul::Internal::ECONNREFUSED when Errno::ECONNREFUSED is rescued' do + stub_consul_discover_prometheus.to_raise(Errno::ECONNREFUSED) + + expect { subject } + .to raise_error(described_class::ECONNREFUSED) + end + + it 'raises Consul::Internal::UnexpectedResponseError when StandardError is rescued' do + stub_consul_discover_prometheus.to_raise(StandardError) + + expect { subject } + .to raise_error(described_class::UnexpectedResponseError) + end + + it 'raises Consul::Internal::UnexpectedResponseError when request returns 500' do + stub_consul_discover_prometheus.to_return(status: 500, body: '{ message: "FAIL!" }') + + expect { subject } + .to raise_error(described_class::UnexpectedResponseError) + end + + it 'raises Consul::Internal::UnexpectedResponseError when request returns non json data' do + stub_consul_discover_prometheus.to_return(status: 200, body: 'not json') + + expect { subject } + .to raise_error(described_class::UnexpectedResponseError) + end + end + + shared_examples 'returns nil given blank value of' do |input_symbol| + [nil, ''].each do |value| + let(input_symbol) { value } + + it { is_expected.to be_nil } + end + end + + describe '.discover_service' do + subject { described_class.discover_service(service_name: service_name) } + + let(:service_name) { 'prometheus' } + + it_behaves_like 'returns nil given blank value of', :api_url + + it_behaves_like 'returns nil given blank value of', :service_name + + context 'one service discovered' do + before do + stub_consul_discover_prometheus.to_return(status: 200, body: '[{"ServiceAddress":"prom.net","ServicePort":9090}]') + end + + it 'returns the service address and port' do + is_expected.to eq(["prom.net", 9090]) + end + end + + context 'multiple services discovered' do + before do + stub_consul_discover_prometheus + .to_return(status: 200, body: '[{"ServiceAddress":"prom_1.net","ServicePort":9090},{"ServiceAddress":"prom.net","ServicePort":9090}]') + end + + it 'uses the first service' do + is_expected.to eq(["prom_1.net", 9090]) + end + end + + it_behaves_like 'handles failure response' + end + + describe '.discover_prometheus_server_address' do + subject { described_class.discover_prometheus_server_address } + + before do + stub_consul_discover_prometheus + .to_return(status: 200, body: '[{"ServiceAddress":"prom.net","ServicePort":9090}]') + end + + it 'returns the server address' do + is_expected.to eq('prom.net:9090') + end + + it_behaves_like 'returns nil given blank value of', :api_url + + it_behaves_like 'handles failure response' + end + + def stub_consul_discover_prometheus + stub_request(:get, /v1\/catalog\/service\/prometheus/) + end +end |