spec/services/prometheus/create_default_alerts_service_spec.rb


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
80
81
82
83
84
85
86
87
88
89
90
91

# frozen_string_literal: true

require 'spec_helper'

describe Prometheus::CreateDefaultAlertsService do
  let_it_be(:project) { create(:project, :repository) }
  let(:instance) { described_class.new(project: project) }
  let(:expected_alerts) { described_class::DEFAULT_ALERTS }

  describe '#execute' do
    subject(:execute) { instance.execute }

    shared_examples 'no alerts created' do
      it 'does not create alerts' do
        expect { execute }.not_to change { project.reload.prometheus_alerts.count }
      end
    end

    context 'no environment' do
      it_behaves_like 'no alerts created'
    end

    context 'environment exists' do
      let_it_be(:environment) { create(:environment, project: project) }

      context 'no found metric' do
        it_behaves_like 'no alerts created'
      end

      context 'metric exists' do
        before do
          create_expected_metrics!
        end

        context 'alert exists already' do
          before do
            create_pre_existing_alerts!(environment)
          end

          it_behaves_like 'no alerts created'
        end

        it 'creates alerts' do
          expect { execute }.to change { project.reload.prometheus_alerts.count }
           .by(expected_alerts.size)
        end

        it 'does not schedule an update to prometheus' do
          expect(::Clusters::Applications::ScheduleUpdateService).not_to receive(:new)
          execute
        end

        context 'cluster with prometheus exists' do
          let!(:cluster) { create(:cluster, :with_installed_prometheus, :provided_by_user, projects: [project]) }

          it 'schedules an update to prometheus' do
            expect_next_instance_of(::Clusters::Applications::ScheduleUpdateService) do |instance|
              expect(instance).to receive(:execute)
            end

            execute
          end
        end

        context 'multiple environments' do
          let!(:production) { create(:environment, project: project, name: 'production') }

          it 'uses the production environment' do
            expect { execute }.to change { production.reload.prometheus_alerts.count }
              .by(expected_alerts.size)
          end
        end
      end
    end
  end

  private

  def create_expected_metrics!
    expected_alerts.each do |alert_hash|
      create(:prometheus_metric, :common, identifier: alert_hash.fetch(:identifier))
    end
  end

  def create_pre_existing_alerts!(environment)
    expected_alerts.each do |alert_hash|
      metric = PrometheusMetric.for_identifier(alert_hash[:identifier]).first!
      create(:prometheus_alert, prometheus_metric: metric, project: project, environment: environment)
    end
  end
end