debian-mirror-gitlab/spec/lib/gitlab/consul/internal_spec.rb

# 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