diff options
Diffstat (limited to 'spec/lib/gitlab/alert_management/payload_spec.rb')
-rw-r--r-- | spec/lib/gitlab/alert_management/payload_spec.rb | 60 |
1 files changed, 60 insertions, 0 deletions
diff --git a/spec/lib/gitlab/alert_management/payload_spec.rb b/spec/lib/gitlab/alert_management/payload_spec.rb new file mode 100644 index 00000000000..44b55e228c5 --- /dev/null +++ b/spec/lib/gitlab/alert_management/payload_spec.rb @@ -0,0 +1,60 @@ +# frozen_string_literal: true + +require 'spec_helper' + +RSpec.describe Gitlab::AlertManagement::Payload do + describe '#parse' do + let_it_be(:project) { build_stubbed(:project) } + let(:payload) { {} } + + context 'without a monitoring_tool specified by caller' do + subject { described_class.parse(project, payload) } + + context 'without a monitoring tool in the payload' do + it { is_expected.to be_a Gitlab::AlertManagement::Payload::Generic } + end + + context 'with the payload specifying Prometheus' do + let(:payload) { { 'monitoring_tool' => 'Prometheus' } } + + it { is_expected.to be_a Gitlab::AlertManagement::Payload::Prometheus } + + context 'with gitlab-managed attributes' do + let(:payload) { { 'monitoring_tool' => 'Prometheus', 'labels' => { 'gitlab_alert_id' => '12' } } } + + it { is_expected.to be_a Gitlab::AlertManagement::Payload::ManagedPrometheus } + end + end + + context 'with the payload specifying an unknown tool' do + let(:payload) { { 'monitoring_tool' => 'Custom Tool' } } + + it { is_expected.to be_a Gitlab::AlertManagement::Payload::Generic } + end + end + + context 'with monitoring_tool specified by caller' do + subject { described_class.parse(project, payload, monitoring_tool: monitoring_tool) } + + context 'as Prometheus' do + let(:monitoring_tool) { 'Prometheus' } + + context 'with an externally managed prometheus payload' do + it { is_expected.to be_a Gitlab::AlertManagement::Payload::Prometheus } + end + + context 'with a self-managed prometheus payload' do + let(:payload) { { 'labels' => { 'gitlab_alert_id' => '14' } } } + + it { is_expected.to be_a Gitlab::AlertManagement::Payload::ManagedPrometheus } + end + end + + context 'as an unknown tool' do + let(:monitoring_tool) { 'Custom Tool' } + + it { is_expected.to be_a Gitlab::AlertManagement::Payload::Generic } + end + end + end +end |