2 files changed, 87 insertions, 61 deletions
diff --git a/spec/services/incident_management/create_incident_label_service_spec.rb b/spec/services/incident_management/create_incident_label_service_spec.rb
index 4771dfc9e64..441cddf1d2e 100644
--- a/spec/services/incident_management/create_incident_label_service_spec.rb
+++ b/spec/services/incident_management/create_incident_label_service_spec.rb
@@ -3,65 +3,5 @@
 require 'spec_helper'
 
 RSpec.describe IncidentManagement::CreateIncidentLabelService do
-  let_it_be(:project) { create(:project, :private) }
-  let_it_be(:user) { User.alert_bot }
-  let(:service) { described_class.new(project, user) }
-
-  subject(:execute) { service.execute }
-
-  describe 'execute' do
-    let(:incident_label_attributes) { attributes_for(:label, :incident) }
-    let(:title) { incident_label_attributes[:title] }
-    let(:color) { incident_label_attributes[:color] }
-    let(:description) { incident_label_attributes[:description] }
-
-    shared_examples 'existing label' do
-      it 'returns the existing label' do
-        expect { execute }.not_to change(Label, :count)
-
-        expect(execute).to be_success
-        expect(execute.payload).to eq(label: label)
-      end
-    end
-
-    shared_examples 'new label' do
-      it 'creates a new label' do
-        expect { execute }.to change(Label, :count).by(1)
-
-        label = project.reload.labels.last
-        expect(execute).to be_success
-        expect(execute.payload).to eq(label: label)
-        expect(label.title).to eq(title)
-        expect(label.color).to eq(color)
-        expect(label.description).to eq(description)
-      end
-    end
-
-    context 'with predefined project label' do
-      it_behaves_like 'existing label' do
-        let!(:label) { create(:label, project: project, title: title) }
-      end
-    end
-
-    context 'with predefined group label' do
-      let(:project) { create(:project, group: group) }
-      let(:group) { create(:group) }
-
-      it_behaves_like 'existing label' do
-        let!(:label) { create(:group_label, group: group, title: title) }
-      end
-    end
-
-    context 'without label' do
-      context 'when user has permissions to create labels' do
-        it_behaves_like 'new label'
-      end
-
-      context 'when user has no permissions to create labels' do
-        let_it_be(:user) { create(:user) }
-
-        it_behaves_like 'new label'
-      end
-    end
-  end
+  it_behaves_like 'incident management label service'
 end
diff --git a/spec/services/incident_management/incidents/update_severity_service_spec.rb b/spec/services/incident_management/incidents/update_severity_service_spec.rb
new file mode 100644
index 00000000000..bc1abf82cf2
--- /dev/null
+++ b/spec/services/incident_management/incidents/update_severity_service_spec.rb
@@ -0,0 +1,86 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+RSpec.describe IncidentManagement::Incidents::UpdateSeverityService do
+  let_it_be(:user) { create(:user) }
+
+  describe '#execute' do
+    let(:severity) { 'low' }
+    let(:system_note_worker) { ::IncidentManagement::AddSeveritySystemNoteWorker }
+
+    subject(:update_severity) { described_class.new(issuable, user, severity).execute }
+
+    before do
+      allow(system_note_worker).to receive(:perform_async)
+    end
+
+    shared_examples 'adds a system note' do
+      it 'calls AddSeveritySystemNoteWorker' do
+        update_severity
+
+        expect(system_note_worker).to have_received(:perform_async).with(issuable.id, user.id)
+      end
+    end
+
+    context 'when issuable not an incident' do
+      %i(issue merge_request).each do |issuable_type|
+        let(:issuable) { build_stubbed(issuable_type) }
+
+        it { is_expected.to be_nil }
+
+        it 'does not set severity' do
+          expect { update_severity }.not_to change(IssuableSeverity, :count)
+        end
+
+        it 'does not add a system note' do
+          update_severity
+
+          expect(system_note_worker).not_to have_received(:perform_async)
+        end
+      end
+    end
+
+    context 'when issuable is an incident' do
+      let!(:issuable) { create(:incident) }
+
+      context 'when issuable does not have issuable severity yet' do
+        it 'creates new record' do
+          expect { update_severity }.to change { IssuableSeverity.where(issue: issuable).count }.to(1)
+        end
+
+        it 'sets severity to specified value' do
+          expect { update_severity }.to change { issuable.severity }.to('low')
+        end
+
+        it_behaves_like 'adds a system note'
+      end
+
+      context 'when issuable has an issuable severity' do
+        let!(:issuable_severity) { create(:issuable_severity, issue: issuable, severity: 'medium') }
+
+        it 'does not create new record' do
+          expect { update_severity }.not_to change(IssuableSeverity, :count)
+        end
+
+        it 'updates existing issuable severity' do
+          expect { update_severity }.to change { issuable_severity.severity }.to(severity)
+        end
+
+        it_behaves_like 'adds a system note'
+      end
+
+      context 'when severity value is unsupported' do
+        let(:severity) { 'unsupported-severity' }
+
+        it 'sets the severity to default value' do
+          update_severity
+
+          expect(issuable.issuable_severity.severity).to eq(IssuableSeverity::DEFAULT)
+        end
+
+        it_behaves_like 'adds a system note'
+      end
+    end
+  end
+end