Adding ui_click event for fixing tests

Author: laileni-aws

packages/core/src/amazonqTest/chat/controller/controller.ts

@@ -384,6 +384,13 @@ export class TestController {
                 this.messenger.sendChatInputEnabled(data.tabID, true)
                 await this.sessionCleanUp()
                 break
+            case ButtonActions.STOP_FIXING_TEST:
+                testGenState.setToCancelling()
+                void this.handleUpdatePromptProgress({ status: 'cancel', tabID: data.tabID })
+                telemetry.ui_click.emit({ elementId: 'unitTestGeneration_cancelFixingTest' })
+                this.messenger.sendChatInputEnabled(data.tabID, true)
+                await this.sessionCleanUp()
+                break
         }
     }
     // This function handles actions if user gives any input from the chatInput box

packages/core/src/amazonqTest/chat/controller/messenger/messengerUtils.ts

@@ -12,6 +12,7 @@ export enum ButtonActions {
     VIEW_DIFF = 'View-Diff',
     STOP_TEST_GEN = 'Stop-Test-Generation',
     STOP_BUILD = 'Stop-Build-Process',
+    STOP_FIXING_TEST = 'Stop-Fixing-Test',
 }
 
 // TODO: Refactor the common functionality between Transform, FeatureDev, CWSPRChat, Scan and UTG to a new Folder.

packages/core/src/amazonqTest/models/constants.ts

@@ -13,15 +13,19 @@ export const testChat = 'testChat'
 
 export const maxUserPromptLength = 4096 // user prompt character limit from MPS and API model.
 
+const baseProgressField: Partial<ProgressField> = {
+    status: 'default',
+    value: -1,
+}
+
 export const cancelTestGenButton: ChatItemButton = {
     id: ButtonActions.STOP_TEST_GEN,
     text: 'Cancel',
     icon: 'cancel' as MynahIcons,
 }
 
 export const testGenProgressField: ProgressField = {
-    status: 'default',
-    value: -1,
+    ...baseProgressField,
     text: 'Generating unit tests...',
     actions: [cancelTestGenButton],
 }
@@ -46,25 +50,29 @@ export const cancelBuildProgressButton: ChatItemButton = {
     icon: 'cancel' as MynahIcons,
 }
 
+export const cancelFixingTestButton: ChatItemButton = {
+    id: ButtonActions.STOP_FIXING_TEST,
+    text: 'Cancel',
+    icon: 'cancel' as MynahIcons,
+}
+
 export const buildProgressField: ProgressField = {
-    status: 'default',
-    value: -1,
+    ...baseProgressField,
     text: 'Compiling project...',
     actions: [cancelBuildProgressButton],
 }
 
 export const fixingTestProgressField: ProgressField = {
-    status: 'default',
-    value: -1,
+    ...baseProgressField,
     text: 'Fixing test failures...',
-    actions: [cancelBuildProgressButton],
+    actions: [cancelFixingTestButton],
 }
 
 export const errorProgressField: ProgressField = {
     status: 'error',
     text: 'Error...Input needed',
     value: -1,
-    actions: [cancelBuildProgressButton],
+    actions: [cancelTestGenButton],
 }
 
 export const testGenSummaryMessage = (