1 files changed, 25 insertions, 5 deletions
diff --git a/spec/frontend/work_items/mock_data.js b/spec/frontend/work_items/mock_data.js
index b66e250f428..0099dce77b3 100644
--- a/spec/frontend/work_items/mock_data.js
+++ b/spec/frontend/work_items/mock_data.js
@@ -301,11 +301,12 @@ export const workItemResponseFactory = ({
         milestoneWidgetPresent
           ? {
               __typename: 'WorkItemWidgetMilestone',
-              dueDate: null,
-              expired: false,
-              id: 'gid://gitlab/Milestone/30',
-              title: 'v4.0',
               type: 'MILESTONE',
+              milestone: {
+                expired: false,
+                id: 'gid://gitlab/Milestone/30',
+                title: 'v4.0',
+              },
             }
           : { type: 'MOCK TYPE' },
         {
@@ -610,6 +611,25 @@ export const workItemIterationSubscriptionResponse = {
   },
 };
 
+export const workItemMilestoneSubscriptionResponse = {
+  data: {
+    issuableMilestoneUpdated: {
+      id: 'gid://gitlab/WorkItem/1',
+      widgets: [
+        {
+          __typename: 'WorkItemWidgetMilestone',
+          type: 'MILESTONE',
+          milestone: {
+            id: 'gid://gitlab/Milestone/1125',
+            expired: false,
+            title: 'Milestone title',
+          },
+        },
+      ],
+    },
+  },
+};
+
 export const workItemHierarchyEmptyResponse = {
   data: {
     workItem: {
@@ -1116,7 +1136,7 @@ export const groupIterationsResponseWithNoIterations = {
 };
 
 export const mockMilestoneWidgetResponse = {
-  dueDate: null,
+  state: 'active',
   expired: false,
   id: 'gid://gitlab/Milestone/30',
   title: 'v4.0',