summaryrefslogtreecommitdiff
path: root/spec/commands/metrics_server/metrics_server_spec.rb
blob: f3936e6b3468c4c003051a58316768f66f74de90 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
# frozen_string_literal: true

require 'spec_helper'

require_relative '../../../metrics_server/metrics_server'

# End-to-end tests for the metrics server process we use to serve metrics
# from forking applications (Sidekiq, Puma) to the Prometheus scraper.
RSpec.describe 'bin/metrics-server', :aggregate_failures do
  let(:config_file) { Tempfile.new('gitlab.yml') }
  let(:config) do
    {
      'test' => {
        'monitoring' => {
          'sidekiq_exporter' => {
            'address' => 'localhost',
            'enabled' => true,
            'port' => 3807
          }
        }
      }
    }
  end

  context 'with a running server' do
    before do
      # We need to send a request to localhost
      WebMock.allow_net_connect!

      config_file.write(YAML.dump(config))
      config_file.close

      env = {
        'GITLAB_CONFIG' => config_file.path,
        'METRICS_SERVER_TARGET' => 'sidekiq',
        'WIPE_METRICS_DIR' => '1'
      }
      @pid = Process.spawn(env, 'bin/metrics-server', pgroup: true)
    end

    after do
      webmock_enable!

      if @pid
        pgrp = Process.getpgid(@pid)

        Timeout.timeout(5) do
          Process.kill('TERM', -pgrp)
          Process.waitpid(@pid)
        end

        expect(Gitlab::ProcessManagement.process_alive?(@pid)).to be(false)
      end
    rescue Errno::ESRCH => _
      # 'No such process' means the process died before
    ensure
      config_file.unlink
    end

    it 'serves /metrics endpoint' do
      expect do
        Timeout.timeout(5) do
          http_ok = false
          until http_ok
            sleep 1
            response = Gitlab::HTTP.try_get("http://localhost:3807/metrics", allow_local_requests: true)
            http_ok = response&.success?
          end
        end
      end.not_to raise_error
    end
  end
end