show customer facing message

changelog

logic

Author: chungjac

packages/amazonq/.changes/next-release/Bug Fix-7bb4ae68-204d-48a1-a510-490d2a2a938a.json

@@ -0,0 +1,4 @@
+{
+	"type": "Bug Fix",
+	"description": "/test: show customer facing error message"
+}

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

@@ -309,26 +309,13 @@ export class TestController {
             undefined,
             isCancel ? 'CANCELLED' : 'FAILED'
         )
-        if (session.stopIteration) {
-            // Error from Science
-            this.messenger.sendMessage(
-                data.error.uiMessage.replaceAll('```', ''),
-                data.tabID,
-                'answer',
-                'testGenErrorMessage',
-                this.getFeedbackButtons()
-            )
-        } else {
-            isCancel
-                ? this.messenger.sendMessage(
-                      data.error.uiMessage,
-                      data.tabID,
-                      'answer',
-                      'testGenErrorMessage',
-                      this.getFeedbackButtons()
-                  )
-                : this.sendErrorMessage(data)
-        }
+        this.messenger.sendMessage(
+            data.error.uiMessage.replaceAll('```', ''),
+            data.tabID,
+            'answer',
+            'testGenErrorMessage',
+            this.getFeedbackButtons()
+        )
         await this.sessionCleanUp()
         return
     }

packages/core/src/codewhisperer/commands/startTestGeneration.ts

@@ -20,8 +20,6 @@ import { ChatSessionManager } from '../../amazonqTest/chat/storages/chatSession'
 import { ChildProcess, spawn } from 'child_process' // eslint-disable-line no-restricted-imports
 import { BuildStatus } from '../../amazonqTest/chat/session/session'
 import { fs } from '../../shared/fs/fs'
-import { TestGenerationJobStatus } from '../models/constants'
-import { TestGenFailedError } from '../../amazonqTest/error'
 import { Range } from '../client/codewhispereruserclient'
 
 // eslint-disable-next-line unicorn/no-null
@@ -112,18 +110,13 @@ export async function startTestGenerationProcess(
         if (!shouldContinueRunning(tabID)) {
             return
         }
-        const jobStatus = await pollTestJobStatus(
+        await pollTestJobStatus(
             testJob.testGenerationJob.testGenerationJobId,
             testJob.testGenerationJob.testGenerationJobGroupName,
             filePath,
             initialExecution
         )
         // TODO: Send status to test summary
-        if (jobStatus === TestGenerationJobStatus.FAILED) {
-            session.numberOfTestsGenerated = 0
-            logger.verbose(`Test generation failed.`)
-            throw new TestGenFailedError()
-        }
         throwIfCancelled()
         if (!shouldContinueRunning(tabID)) {
             return

packages/core/src/codewhisperer/service/testGenHandler.ts

@@ -18,6 +18,7 @@ import {
     CreateUploadUrlError,
     ExportResultsArchiveError,
     InvalidSourceZipError,
+    TestGenFailedError,
     TestGenStoppedError,
     TestGenTimedOutError,
 } from '../../amazonqTest/error'
@@ -193,9 +194,15 @@ export async function pollTestJobStatus(
             }
         }
         ChatSessionManager.Instance.getSession().targetFileInfo = targetFileInfo
-        if (resp.testGenerationJob?.status !== CodeWhispererConstants.TestGenerationJobStatus.IN_PROGRESS) {
-            // This can be FAILED or COMPLETED
-            status = resp.testGenerationJob?.status as CodeWhispererConstants.TestGenerationJobStatus
+        status = resp.testGenerationJob?.status as CodeWhispererConstants.TestGenerationJobStatus
+        if (status === CodeWhispererConstants.TestGenerationJobStatus.FAILED) {
+            session.numberOfTestsGenerated = 0
+            logger.verbose(`Test generation failed.`)
+            if (resp.testGenerationJob?.jobStatusReason) {
+                throw new TestGenFailedError(resp.testGenerationJob?.jobStatusReason)
+            }
+            throw new TestGenFailedError()
+        } else if (status === CodeWhispererConstants.TestGenerationJobStatus.COMPLETED) {
             logger.verbose(`testgen job status: ${status}`)
             logger.verbose(`Complete polling test job status.`)
             break