1 files changed, 41 insertions, 2 deletions
diff --git a/spec/requests/api/graphql/mutations/incident_management/timeline_event/update_spec.rb b/spec/requests/api/graphql/mutations/incident_management/timeline_event/update_spec.rb
index 542d51b990f..163c689e399 100644
--- a/spec/requests/api/graphql/mutations/incident_management/timeline_event/update_spec.rb
+++ b/spec/requests/api/graphql/mutations/incident_management/timeline_event/update_spec.rb
@@ -2,24 +2,36 @@
 
 require 'spec_helper'
 
-RSpec.describe 'Updating an incident timeline event' do
+RSpec.describe 'Updating an incident timeline event', feature_category: :incident_management do
   include GraphqlHelpers
 
   let_it_be(:user) { create(:user) }
   let_it_be(:project) { create(:project) }
   let_it_be(:incident) { create(:incident, project: project) }
+  let_it_be(:tag1) { create(:incident_management_timeline_event_tag, project: project, name: 'Tag 1') }
+  let_it_be(:tag2) { create(:incident_management_timeline_event_tag, project: project, name: 'Tag 2') }
   let_it_be_with_reload(:timeline_event) do
     create(:incident_management_timeline_event, incident: incident, project: project)
   end
 
+  # Pre-attach a tag to the event
+  let_it_be(:tag_link1) do
+    create(:incident_management_timeline_event_tag_link,
+      timeline_event: timeline_event,
+      timeline_event_tag: tag1
+    )
+  end
+
   let(:occurred_at) { 1.minute.ago.iso8601 }
   let(:note) { 'Updated note' }
+  let(:tag_names) { [] }
 
   let(:variables) do
     {
       id: timeline_event.to_global_id.to_s,
       note: note,
-      occurred_at: occurred_at
+      occurred_at: occurred_at,
+      timeline_event_tag_names: tag_names
     }
   end
 
@@ -33,6 +45,7 @@ RSpec.describe 'Updating an incident timeline event' do
           author { id username }
           updatedByUser { id username }
           incident { id title }
+          timelineEventTags { nodes { name } }
           note
           noteHtml
           occurredAt
@@ -71,6 +84,9 @@ RSpec.describe 'Updating an incident timeline event' do
         'id' => incident.to_global_id.to_s,
         'title' => incident.title
       },
+      'timelineEventTags' => {
+        'nodes' => []
+      },
       'note' => note,
       'noteHtml' => timeline_event.note_html,
       'occurredAt' => occurred_at,
@@ -85,4 +101,27 @@ RSpec.describe 'Updating an incident timeline event' do
     it_behaves_like 'timeline event mutation responds with validation error',
       error_message: 'Timeline text is too long (maximum is 280 characters)'
   end
+
+  context 'when timeline event tag names are passed' do
+    context 'when tags exist' do
+      let(:tag_names) { [tag2.name] }
+
+      it 'removes tag1 and adds tag2' do
+        post_graphql_mutation(mutation, current_user: user)
+
+        timeline_event_response = mutation_response['timelineEvent']
+        tag_names = timeline_event_response['timelineEventTags']['nodes']
+
+        expect(response).to have_gitlab_http_status(:success)
+        expect(tag_names).to contain_exactly({ "name" => tag2.name })
+      end
+    end
+
+    context 'when tags do not exist' do
+      let(:tag_names) { ['some other tag'] }
+
+      it_behaves_like 'timeline event mutation responds with validation error',
+        error_message: "Following tags don't exist: [\"some other tag\"]"
+    end
+  end
 end