1 files changed, 21 insertions, 1 deletions
diff --git a/spec/frontend/notes/stores/getters_spec.js b/spec/frontend/notes/stores/getters_spec.js
index e03fa854e54..1514602d424 100644
--- a/spec/frontend/notes/stores/getters_spec.js
+++ b/spec/frontend/notes/stores/getters_spec.js
@@ -1,5 +1,5 @@
 import discussionWithTwoUnresolvedNotes from 'test_fixtures/merge_requests/resolved_diff_discussion.json';
-import { DESC, ASC } from '~/notes/constants';
+import { DESC, ASC, NOTEABLE_TYPE_MAPPING } from '~/notes/constants';
 import * as getters from '~/notes/stores/getters';
 import {
   notesDataMock,
@@ -536,4 +536,24 @@ describe('Getters Notes Store', () => {
       expect(getters.sortDirection(state)).toBe(DESC);
     });
   });
+
+  describe('canUserAddIncidentTimelineEvents', () => {
+    it.each`
+      userData                              | noteableData                                | expected
+      ${{ can_add_timeline_events: true }}  | ${{ type: NOTEABLE_TYPE_MAPPING.Incident }} | ${true}
+      ${{ can_add_timeline_events: true }}  | ${{ type: NOTEABLE_TYPE_MAPPING.Issue }}    | ${false}
+      ${null}                               | ${{ type: NOTEABLE_TYPE_MAPPING.Incident }} | ${false}
+      ${{ can_add_timeline_events: false }} | ${{ type: NOTEABLE_TYPE_MAPPING.Incident }} | ${false}
+    `(
+      'with userData=$userData and noteableData=$noteableData, expected=$expected',
+      ({ userData, noteableData, expected }) => {
+        Object.assign(state, {
+          userData,
+          noteableData,
+        });
+
+        expect(getters.canUserAddIncidentTimelineEvents(state)).toBe(expected);
+      },
+    );
+  });
 });