Fixes

plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonqCodeTest/CodeWhispererUTGChatManager.kt

@@ -53,9 +53,8 @@
 import software.aws.toolkits.jetbrains.settings.CodeWhispererSettings
 import software.aws.toolkits.jetbrains.utils.isQConnected
 import software.aws.toolkits.resources.message
 import software.aws.toolkits.telemetry.AmazonqTelemetry
 import software.aws.toolkits.telemetry.MetricResult
-//import software.aws.toolkits.telemetry.Status
 import java.io.ByteArrayInputStream
 import java.io.ByteArrayOutputStream
 import java.io.File

plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonqCodeTest/controller/CodeTestChatController.kt

@@ -103,7 +103,6 @@
 import software.aws.toolkits.telemetry.FeatureId
 import software.aws.toolkits.telemetry.InteractionType
 import software.aws.toolkits.telemetry.MetricResult
-//import software.aws.toolkits.telemetry.Status
 import software.aws.toolkits.telemetry.UiTelemetry
 import java.io.File
 import java.nio.file.Files
@@ -448,7 +447,7 @@
         val session = codeTestChatHelper.getActiveSession()
         when (message.vote) {
             "upvote" -> {
                 AmazonqTelemetry.feedback(
                     featureId = FeatureId.AmazonQTest,
                     interactionType = InteractionType.Upvote,
                     credentialStartUrl = getStartUrl(project = context.project),
@@ -599,7 +598,7 @@
 
                     codeTestChatHelper.updateUI(
                         promptInputDisabledState = true,
                         promptInputPlaceholder = message("testgen.placeholder.select_an_option"),
                     )
 
                     codeTestChatHelper.updateAnswer(
@@ -669,7 +668,7 @@
                         "Successfully sent test generation telemetry. RequestId: ${
                             testGenerationEventResponse.responseMetadata().requestId()}"
                     }
                     AmazonqTelemetry.utgGenerateTests(
                         cwsprChatProgrammingLanguage = session.programmingLanguage.languageId,
                         hasUserPromptSupplied = session.hasUserPromptSupplied,
                         isFileInWorkspace = true,
@@ -792,7 +791,7 @@
                             message = """
                                  You have gone through both iterations and this unit test generation workflow is complete.
                             """.trimIndent(),
                             type = ChatMessageType.Answer,
                         )
                     )
                     codeTestChatHelper.updateUI(
@@ -886,7 +885,7 @@
                     "Successfully sent test generation telemetry. RequestId: ${
                         testGenerationEventResponse.responseMetadata().requestId()}"
                 }
                 UiTelemetry.click(
                     null as Project?,
                     if (session.listOfTestGenerationJobId.size == 1) {
                         "unitTestGeneration_rejectDiff"
@@ -919,7 +918,7 @@
 //                        status = Status.REJECTED
                     )
                 } else {
                     AmazonqTelemetry.unitTestGeneration(
                         count = session.listOfTestGenerationJobId.size.toLong() - 1,
                         cwsprChatProgrammingLanguage = session.programmingLanguage.languageId,
                         hasUserPromptSupplied = session.hasUserPromptSupplied,
@@ -945,7 +944,7 @@
                         requestId = session.startTestGenerationRequestId,
                         update = session.updateBuildCommands,
                     )
                 }
 
                 sessionCleanUp(message.tabId)
             }
@@ -1072,7 +1071,7 @@
                 }
 
                 // TODO: only go to future iterations when buildExitCode or testExitCode > 0, right now iterate regardless
                 if (taskContext.buildExitCode > 0) {
                     // handle build failure case
 
                     taskContext.progressStatus = BuildAndExecuteProgressStatus.BUILD_FAILED
@@ -1131,7 +1130,7 @@
                 return
             }
             "stop_fixing_test_cases" -> {
                 UiTelemetry.click(null as Project?, "unitTestGeneration_cancelFixingTest")
                 session.isGeneratingTests = false
                 session.buildStatus = BuildStatus.CANCELLED
                 return
@@ -1150,10 +1149,10 @@
         sessionCleanUp(codeTestChatHelper.getActiveSession().tabId)
     }
 
     suspend fun updateBuildAndExecuteProgressCard(
         currentStatus: BuildAndExecuteProgressStatus,
         messageId: String?,
     ): String? {
         val updatedText = constructBuildAndExecutionSummaryText(currentStatus, codeTestChatHelper)
 
         if (currentStatus == BuildAndExecuteProgressStatus.RUN_BUILD) {

plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonqCodeTest/messages/CodeTestMessage.kt

@@ -17,7 +17,7 @@ const val CODE_TEST_TAB_NAME = "codetest"
 enum class CodeTestButtonId(val id: String) {
     StopTestGeneration("stop_test_generation"),
     StopTestGenBuildAndExecution("stop_test_gen_build_and_execution"),
-    StopFixingTestCases("stop_fixing_test_cases")
+    StopFixingTestCases("stop_fixing_test_cases"),
 }
 
 data class Button(