summaryrefslogtreecommitdiff
path: root/spec/lib/gitlab/metrics/dashboard/importers/prometheus_metrics_spec.rb
blob: 09d5e048f6a7170f48ead39527d77d731261efa7 (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
74
75
76
77
78
79
# frozen_string_literal: true

require 'spec_helper'

RSpec.describe Gitlab::Metrics::Dashboard::Importers::PrometheusMetrics do
  include MetricsDashboardHelpers

  describe '#execute' do
    let(:project) { create(:project) }
    let(:dashboard_path) { 'path/to/dashboard.yml' }

    subject { described_class.new(dashboard_hash, project: project, dashboard_path: dashboard_path) }

    context 'valid dashboard' do
      let(:dashboard_hash) { load_sample_dashboard }

      context 'with all new metrics' do
        it 'creates PrometheusMetrics' do
          expect { subject.execute }.to change { PrometheusMetric.count }.by(3)
        end
      end

      context 'with existing metrics' do
        let!(:existing_metric) do
          create(:prometheus_metric, {
            project:    project,
            identifier: 'metric_b',
            title:      'overwrite',
            y_label:    'overwrite',
            query:      'overwrite',
            unit:       'overwrite',
            legend:     'overwrite'
          })
        end

        it 'updates existing PrometheusMetrics' do
          described_class.new(dashboard_hash, project: project, dashboard_path: dashboard_path).execute

          expect(existing_metric.reload.attributes.with_indifferent_access).to include({
            title:   'Super Chart B',
            y_label: 'y_label',
            query:   'query',
            unit:    'unit',
            legend:  'Legend Label'
          })
        end

        it 'creates new PrometheusMetrics' do
          expect { subject.execute }.to change { PrometheusMetric.count }.by(2)
        end

        context 'with stale metrics' do
          let!(:stale_metric) do
            create(:prometheus_metric,
              project: project,
              identifier: 'stale_metric',
              dashboard_path: dashboard_path,
              group: 3
            )
          end

          it 'deletes stale metrics' do
            subject.execute

            expect { stale_metric.reload }.to raise_error(ActiveRecord::RecordNotFound)
          end
        end
      end
    end

    context 'invalid dashboard' do
      let(:dashboard_hash) { {} }

      it 'returns false' do
        expect(subject.execute).to eq(false)
      end
    end
  end
end