Add latest changes from gitlab-org/gitlab@13-4-stable-ee

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