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/codewhisperer/service/testGenHandler.ts

@@ -18,6 +18,7 @@ import {
     CreateUploadUrlError,
     ExportResultsArchiveError,
     InvalidSourceZipError,
+    TestGenFailedError,
     TestGenStoppedError,
     TestGenTimedOutError,
 } from '../../amazonqTest/error'
@@ -193,9 +194,10 @@ 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) {
+            throw new TestGenFailedError(resp.testGenerationJob?.jobStatusReason)
+        } else if (status === CodeWhispererConstants.TestGenerationJobStatus.COMPLETED) {
             logger.verbose(`testgen job status: ${status}`)
             logger.verbose(`Complete polling test job status.`)
             break