1 files changed, 22 insertions, 30 deletions

diff --git a/app/assets/javascripts/graphql_shared/issuable_client.js b/app/assets/javascripts/graphql_shared/issuable_client.js
index e86103c332b..3b737dfff33 100644
--- a/app/assets/javascripts/graphql_shared/issuable_client.js
+++ b/app/assets/javascripts/graphql_shared/issuable_client.js
@@ -4,14 +4,13 @@ import { concatPagination } from '@apollo/client/utilities';
 import getIssueStateQuery from '~/issues/show/queries/get_issue_state.query.graphql';
 import createDefaultClient from '~/lib/graphql';
 import typeDefs from '~/work_items/graphql/typedefs.graphql';
-import workItemQuery from '~/work_items/graphql/work_item.query.graphql';
-import { WIDGET_TYPE_LABELS } from '~/work_items/constants';
+import { WIDGET_TYPE_MILESTONE } from '~/work_items/constants';
 
 export const temporaryConfig = {
   typeDefs,
   cacheConfig: {
     possibleTypes: {
-      LocalWorkItemWidget: ['LocalWorkItemLabels'],
+      LocalWorkItemWidget: ['LocalWorkItemMilestone'],
     },
     typePolicies: {
       Project: {
@@ -28,18 +27,32 @@ export const temporaryConfig = {
               return (
                 widgets || [
                   {
-                    __typename: 'LocalWorkItemLabels',
-                    type: WIDGET_TYPE_LABELS,
-                    allowScopedLabels: true,
-                    nodes: [],
+                    __typename: 'LocalWorkItemMilestone',
+                    type: WIDGET_TYPE_MILESTONE,
+                    nodes: [
+                      {
+                        dueDate: null,
+                        expired: false,
+                        id: 'gid://gitlab/Milestone/30',
+                        title: 'v4.0',
+                        // eslint-disable-next-line @gitlab/require-i18n-strings
+                        __typename: 'Milestone',
+                      },
+                    ],
                   },
                 ]
               );
             },
           },
           widgets: {
-            merge(_, incoming) {
-              return incoming;
+            merge(existing = [], incoming) {
+              if (existing.length === 0) {
+                return incoming;
+              }
+              return existing.map((existingWidget) => {
+                const incomingWidget = incoming.find((w) => w.type === existingWidget.type);
+                return incomingWidget || existingWidget;
+              });
             },
           },
         },
@@ -62,27 +75,6 @@ export const resolvers = {
       });
       cache.writeQuery({ query: getIssueStateQuery, data });
     },
-    localUpdateWorkItem(_, { input }, { cache }) {
-      const sourceData = cache.readQuery({
-        query: workItemQuery,
-        variables: { id: input.id },
-      });
-
-      const data = produce(sourceData, (draftData) => {
-        if (input.labels) {
-          const labelsWidget = draftData.workItem.mockWidgets.find(
-            (widget) => widget.type === WIDGET_TYPE_LABELS,
-          );
-          labelsWidget.nodes = [...input.labels];
-        }
-      });
-
-      cache.writeQuery({
-        query: workItemQuery,
-        variables: { id: input.id },
-        data,
-      });
-    },
   },
 };