From 0d3e1bf77d8fc337050702179fc760dcde87849b Mon Sep 17 00:00:00 2001
From: FredrikOseberg <fredrik.no@gmail.com>
Date: Thu, 23 Oct 2025 13:01:16 +0200
Subject: [PATCH] fix: show badges

---
 .../ChangeRequest/Changes/Change/MilestoneListRenderer.tsx  | 5 +++++
 .../FeatureView/FeatureOverview/ReleasePlan/ReleasePlan.tsx | 4 ++--
 .../ReleasePlanMilestoneItem/MilestoneAutomation.tsx        | 6 +++---
 3 files changed, 10 insertions(+), 5 deletions(-)

diff --git a/frontend/src/component/changeRequest/ChangeRequest/Changes/Change/MilestoneListRenderer.tsx b/frontend/src/component/changeRequest/ChangeRequest/Changes/Change/MilestoneListRenderer.tsx
index e11396e8fb..91492e374e 100644
--- a/frontend/src/component/changeRequest/ChangeRequest/Changes/Change/MilestoneListRenderer.tsx
+++ b/frontend/src/component/changeRequest/ChangeRequest/Changes/Change/MilestoneListRenderer.tsx
@@ -51,9 +51,14 @@ const MilestoneListRendererCore = ({
                 const hasPendingDelete = milestonesWithDeletedAutomation.has(
                     milestone.id,
                 );
+                const hasPendingModification = milestonesWithAutomation.has(
+                    milestone.id,
+                );
 
                 const badge = hasPendingDelete ? (
                     <Badge color='error'>Deleted in draft</Badge>
+                ) : hasPendingModification ? (
+                    <Badge color='warning'>Modified in draft</Badge>
                 ) : undefined;
 
                 const automationSection =
diff --git a/frontend/src/component/feature/FeatureView/FeatureOverview/ReleasePlan/ReleasePlan.tsx b/frontend/src/component/feature/FeatureView/FeatureOverview/ReleasePlan/ReleasePlan.tsx
index 83033d1ce9..86b981560e 100644
--- a/frontend/src/component/feature/FeatureView/FeatureOverview/ReleasePlan/ReleasePlan.tsx
+++ b/frontend/src/component/feature/FeatureView/FeatureOverview/ReleasePlan/ReleasePlan.tsx
@@ -134,10 +134,10 @@ export const ReleasePlan = ({
             );
             if (!featureInChangeRequest) continue;
 
-            // Look for update or delete progression changes
+            // Look for change or delete progression changes
             const progressionChange = featureInChangeRequest.changes.find(
                 (change: any) =>
-                    (change.action === 'updateMilestoneProgression' &&
+                    (change.action === 'changeMilestoneProgression' &&
                         (change.payload.sourceMilestoneId ===
                             sourceMilestoneId ||
                             change.payload.sourceMilestone ===
diff --git a/frontend/src/component/feature/FeatureView/FeatureOverview/ReleasePlan/ReleasePlanMilestoneItem/MilestoneAutomation.tsx b/frontend/src/component/feature/FeatureView/FeatureOverview/ReleasePlan/ReleasePlanMilestoneItem/MilestoneAutomation.tsx
index 6ce8d1e9c3..4b2d520903 100644
--- a/frontend/src/component/feature/FeatureView/FeatureOverview/ReleasePlan/ReleasePlanMilestoneItem/MilestoneAutomation.tsx
+++ b/frontend/src/component/feature/FeatureView/FeatureOverview/ReleasePlan/ReleasePlanMilestoneItem/MilestoneAutomation.tsx
@@ -84,14 +84,14 @@ export const MilestoneAutomation = ({
 
     const hasPendingCreate =
         pendingProgressionChange?.action === 'createMilestoneProgression';
-    const hasPendingUpdate =
-        pendingProgressionChange?.action === 'updateMilestoneProgression';
+    const hasPendingChange =
+        pendingProgressionChange?.action === 'changeMilestoneProgression';
     const hasPendingDelete =
         pendingProgressionChange?.action === 'deleteMilestoneProgression';
 
     const badge = hasPendingDelete ? (
         <Badge color='error'>Deleted in draft</Badge>
-    ) : hasPendingUpdate ? (
+    ) : hasPendingChange ? (
         <Badge color='warning'>Modified in draft</Badge>
     ) : undefined;