140 lines
4 KiB
Ruby
140 lines
4 KiB
Ruby
# 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, %r{v1/catalog/service/prometheus})
|
|
end
|
|
end
|