Updating the statusCode from String to Int

Author: laileni-aws

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

@@ -254,7 +254,7 @@ export class TestController {
             session.startTestGenerationRequestId,
             performance.now() - session.testGenerationStartTime,
             telemetryErrorMessage,
-            data.error.statusCode ?? '0',
+            data.error.statusCode.toString() ?? '0',
             session.isCodeBlockSelected,
             session.artifactsUploadDuration,
             session.srcPayloadSize,
@@ -280,10 +280,7 @@ export class TestController {
         return
     }
     // Client side error messages
-    private sendErrorMessage(data: {
-        tabID: string
-        error: { uiMessage: string; message: string; code: string; statusCode: string }
-    }) {
+    private sendErrorMessage(data: { tabID: string; error: { uiMessage: string; message: string; code: string } }) {
         const { error, tabID } = data
 
         // If user reached monthly limit for builderId

packages/core/src/amazonqTest/error.ts

@@ -11,68 +11,68 @@ export class TestGenError extends ToolkitError {
     constructor(
         error: string,
         code: string,
-        public statusCode: string,
+        public statusCode: number,
         public uiMessage: string
     ) {
         super(error, { code })
     }
 }
 export class ProjectZipError extends TestGenError {
     constructor(error: string) {
-        super(error, 'ProjectZipError', '400', defaultTestGenErrorMessage)
+        super(error, 'ProjectZipError', 400, defaultTestGenErrorMessage)
     }
 }
 export class InvalidSourceZipError extends TestGenError {
     constructor() {
-        super('Failed to create valid source zip', 'InvalidSourceZipError', '400', defaultTestGenErrorMessage)
+        super('Failed to create valid source zip', 'InvalidSourceZipError', 400, defaultTestGenErrorMessage)
     }
 }
 export class CreateUploadUrlError extends TestGenError {
-    constructor(errorMessage: string, errorCode: string) {
+    constructor(errorMessage: string, errorCode: number) {
         super(errorMessage, 'CreateUploadUrlError', errorCode, technicalErrorCustomerFacingMessage)
     }
 }
 export class UploadTestArtifactToS3Error extends TestGenError {
-    constructor(error: string, statusCode?: string) {
-        super(error, 'UploadTestArtifactToS3Error', statusCode ?? '400', technicalErrorCustomerFacingMessage)
+    constructor(error: string, statusCode?: number) {
+        super(error, 'UploadTestArtifactToS3Error', statusCode ?? 400, technicalErrorCustomerFacingMessage)
     }
 }
 export class CreateTestJobError extends TestGenError {
-    constructor(error: string, code: string) {
-        super(error, 'CreateTestJobError', code, technicalErrorCustomerFacingMessage)
+    constructor(error: string, statusCode: number) {
+        super(error, 'CreateTestJobError', statusCode, technicalErrorCustomerFacingMessage)
     }
 }
 export class TestGenTimedOutError extends TestGenError {
     constructor() {
         super(
             'Test generation failed. Amazon Q timed out.',
             'TestGenTimedOutError',
-            '500',
+            500,
             technicalErrorCustomerFacingMessage
         )
     }
 }
 export class TestGenStoppedError extends TestGenError {
     constructor() {
-        super('Unit test generation cancelled.', 'TestGenCancelled', '400', 'Unit test generation cancelled.')
+        super('Unit test generation cancelled.', 'TestGenCancelled', 400, 'Unit test generation cancelled.')
     }
 }
 export class TestGenFailedError extends TestGenError {
-    constructor(code: string, error?: string) {
+    constructor(statusCode: number, error?: string) {
         super(
             error ?? 'Test generation failed',
             'TestGenFailedError',
-            code,
+            statusCode,
             error ?? technicalErrorCustomerFacingMessage
         )
     }
 }
 export class ExportResultsArchiveError extends TestGenError {
-    constructor(error?: string, statusCode?: string) {
+    constructor(error?: string, statusCode?: number) {
         super(
             error ?? 'Test generation failed',
             'ExportResultsArchiveError',
-            statusCode ?? '400',
+            statusCode ?? 400,
             technicalErrorCustomerFacingMessage
         )
     }

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

@@ -122,7 +122,7 @@ export async function startTestGenerationProcess(
         // TODO: Send status to test summary
         if (jobStatus === TestGenerationJobStatus.FAILED) {
             logger.verbose(`Test generation failed.`)
-            throw new TestGenFailedError('500')
+            throw new TestGenFailedError(500)
         }
         throwIfCancelled()
         if (!shouldContinueRunning(tabID)) {

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

@@ -159,7 +159,7 @@ export async function pollTestJobStatus(
             // Stop the Unit test generation workflow if IDE receive stopIteration = true
             if (shortAnswer.stopIteration === 'true') {
                 session.stopIteration = true
-                throw new TestGenFailedError('400', shortAnswer.planSummary)
+                throw new TestGenFailedError(400, shortAnswer.planSummary)
             }
             if (shortAnswer.numberOfTestMethods) {
                 session.numberOfTestsGenerated = Number(shortAnswer.numberOfTestMethods)