Adding statusCode for exportResultArchive API

Author: laileni-aws

packages/core/src/amazonqTest/error.ts

@@ -74,7 +74,12 @@ export class TestGenFailedError extends TestGenError {
     }
 }
 export class ExportResultsArchiveError extends TestGenError {
-    constructor(error?: string) {
-        super(error ?? 'Test generation failed', 'ExportResultsArchiveError', '500', techinalErrorCustomerFacingMessage)
+    constructor(error?: string, statusCode?: string) {
+        super(
+            error ?? 'Test generation failed',
+            'ExportResultsArchiveError',
+            statusCode ?? '400',
+            techinalErrorCustomerFacingMessage
+        )
     }
 }

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

@@ -297,7 +297,7 @@ export async function downloadResultArchive(
     } catch (e: any) {
         downloadErrorMessage = (e as Error).message
         getLogger().error(`Unit Test Generation: ExportResultArchive error = ${downloadErrorMessage}`)
-        throw new ExportResultsArchiveError(downloadErrorMessage)
+        throw new ExportResultsArchiveError(downloadErrorMessage, e.statusCode)
     } finally {
         cwStreamingClient.destroy()
     }