1 files changed, 46 insertions, 36 deletions

diff --git a/spec/frontend/work_items/components/notes/work_item_note_awards_list_spec.js b/spec/frontend/work_items/components/notes/work_item_note_awards_list_spec.js
index ce915635946..6ce4c09329f 100644
--- a/spec/frontend/work_items/components/notes/work_item_note_awards_list_spec.js
+++ b/spec/frontend/work_items/components/notes/work_item_note_awards_list_spec.js
@@ -9,6 +9,7 @@ import AwardsList from '~/vue_shared/components/awards_list.vue';
 import WorkItemNoteAwardsList from '~/work_items/components/notes/work_item_note_awards_list.vue';
 import addAwardEmojiMutation from '~/work_items/graphql/notes/work_item_note_add_award_emoji.mutation.graphql';
 import removeAwardEmojiMutation from '~/work_items/graphql/notes/work_item_note_remove_award_emoji.mutation.graphql';
+import groupWorkItemNotesByIidQuery from '~/work_items/graphql/notes/group_work_item_notes_by_iid.query.graphql';
 import workItemNotesByIidQuery from '~/work_items/graphql/notes/work_item_notes_by_iid.query.graphql';
 import {
   mockWorkItemNotesResponseWithComments,
@@ -45,7 +46,9 @@ describe('Work Item Note Awards List', () => {
   const findAwardsList = () => wrapper.findComponent(AwardsList);
 
   const createComponent = ({
+    isGroup = false,
     note = firstNote,
+    query = workItemNotesByIidQuery,
     addAwardEmojiMutationHandler = addAwardEmojiMutationSuccessHandler,
     removeAwardEmojiMutationHandler = removeAwardEmojiMutationSuccessHandler,
   } = {}) => {
@@ -55,12 +58,15 @@ describe('Work Item Note Awards List', () => {
     ]);
 
     apolloProvider.clients.defaultClient.writeQuery({
-      query: workItemNotesByIidQuery,
+      query,
       variables: { fullPath, iid: workItemIid },
       ...mockWorkItemNotesResponseWithComments,
     });
 
     wrapper = shallowMount(WorkItemNoteAwardsList, {
+      provide: {
+        isGroup,
+      },
       propsData: {
         fullPath,
         workItemIid,
@@ -89,54 +95,58 @@ describe('Work Item Note Awards List', () => {
       expect(findAwardsList().props('canAwardEmoji')).toBe(hasAwardEmojiPermission);
     });
 
-    it('adds award if not already awarded', async () => {
-      createComponent();
-      await waitForPromises();
-
-      findAwardsList().vm.$emit('award', EMOJI_THUMBSUP);
-
-      expect(addAwardEmojiMutationSuccessHandler).toHaveBeenCalledWith({
-        awardableId: firstNote.id,
-        name: EMOJI_THUMBSUP,
-      });
-    });
+    it.each`
+      isGroup  | query
+      ${true}  | ${groupWorkItemNotesByIidQuery}
+      ${false} | ${workItemNotesByIidQuery}
+    `(
+      'adds award if not already awarded in both group and project contexts',
+      async ({ isGroup, query }) => {
+        createComponent({ isGroup, query });
+        await waitForPromises();
+
+        findAwardsList().vm.$emit('award', EMOJI_THUMBSUP);
+
+        expect(addAwardEmojiMutationSuccessHandler).toHaveBeenCalledWith({
+          awardableId: firstNote.id,
+          name: EMOJI_THUMBSUP,
+        });
+      },
+    );
 
     it('emits error if awarding emoji fails', async () => {
-      createComponent({
-        addAwardEmojiMutationHandler: jest.fn().mockRejectedValue('oh no'),
-      });
-      await waitForPromises();
+      createComponent({ addAwardEmojiMutationHandler: jest.fn().mockRejectedValue('oh no') });
 
       findAwardsList().vm.$emit('award', EMOJI_THUMBSUP);
-
       await waitForPromises();
 
       expect(wrapper.emitted('error')).toEqual([[__('Failed to add emoji. Please try again')]]);
     });
 
-    it('removes award if already awarded', async () => {
-      const removeAwardEmojiMutationHandler = removeAwardEmojiMutationSuccessHandler;
-
-      createComponent({ removeAwardEmojiMutationHandler });
-
-      findAwardsList().vm.$emit('award', EMOJI_THUMBSDOWN);
-
-      await waitForPromises();
-
-      expect(removeAwardEmojiMutationHandler).toHaveBeenCalledWith({
-        awardableId: firstNote.id,
-        name: EMOJI_THUMBSDOWN,
-      });
-    });
+    it.each`
+      isGroup  | query
+      ${true}  | ${groupWorkItemNotesByIidQuery}
+      ${false} | ${workItemNotesByIidQuery}
+    `(
+      'removes award if already awarded in both group and project contexts',
+      async ({ isGroup, query }) => {
+        const removeAwardEmojiMutationHandler = removeAwardEmojiMutationSuccessHandler;
+        createComponent({ isGroup, query, removeAwardEmojiMutationHandler });
+
+        findAwardsList().vm.$emit('award', EMOJI_THUMBSDOWN);
+        await waitForPromises();
+
+        expect(removeAwardEmojiMutationHandler).toHaveBeenCalledWith({
+          awardableId: firstNote.id,
+          name: EMOJI_THUMBSDOWN,
+        });
+      },
+    );
 
     it('restores award if remove fails', async () => {
-      createComponent({
-        removeAwardEmojiMutationHandler: jest.fn().mockRejectedValue('oh no'),
-      });
-      await waitForPromises();
+      createComponent({ removeAwardEmojiMutationHandler: jest.fn().mockRejectedValue('oh no') });
 
       findAwardsList().vm.$emit('award', EMOJI_THUMBSDOWN);
-
       await waitForPromises();
 
       expect(wrapper.emitted('error')).toEqual([[__('Failed to remove emoji. Please try again')]]);