Security Scans: Improving telemetry error messages (#4500)

laileni-aws · web-flow · commit d334cf859636 · 2024-06-03T12:36:35.000-07:00
* Updating telemetry error messages
diff --git a/plugins/toolkit/jetbrains-core/src/software/aws/toolkits/jetbrains/services/codewhisperer/codescan/CodeWhispererCodeScanException.kt b/plugins/toolkit/jetbrains-core/src/software/aws/toolkits/jetbrains/services/codewhisperer/codescan/CodeWhispererCodeScanException.kt
@@ -7,7 +7,7 @@ import software.aws.toolkits.resources.message
 
 open class CodeWhispererCodeScanException(override val message: String?) : RuntimeException()
 
-open class UploadCodeScanException(override val message: String?) : Exception()
+open class CodeWhispererCodeScanServerException(override val message: String?) : RuntimeException()
 
 internal fun noFileOpenError(): Nothing =
     throw CodeWhispererCodeScanException(message("codewhisperer.codescan.no_file_open"))
@@ -27,8 +27,8 @@ internal fun fileFormatNotSupported(format: String): Nothing =
 internal fun fileTooLarge(): Nothing =
     throw CodeWhispererCodeScanException(message("codewhisperer.codescan.file_too_large"))
 
-internal fun uploadArtifactFailedError(errorMessage: String): Nothing =
-    throw UploadCodeScanException(errorMessage)
+internal fun codeScanServerException(errorMessage: String): Nothing =
+    throw CodeWhispererCodeScanServerException(errorMessage)
 
 internal fun invalidSourceZipError(): Nothing =
     throw CodeWhispererCodeScanException(message("codewhisperer.codescan.invalid_source_zip_telemetry"))
diff --git a/plugins/toolkit/jetbrains-core/src/software/aws/toolkits/jetbrains/services/codewhisperer/codescan/CodeWhispererCodeScanManager.kt b/plugins/toolkit/jetbrains-core/src/software/aws/toolkits/jetbrains/services/codewhisperer/codescan/CodeWhispererCodeScanManager.kt
@@ -323,14 +323,26 @@ class CodeWhispererCodeScanManager(val project: Project) {
         return errorMessage
     }
 
+    private fun getCodeScanExceptionMessage(e: CodeWhispererCodeScanException): String? {
+        val message = e.message
+        return when {
+            message.isNullOrBlank() -> null
+            message == message("codewhisperer.codescan.invalid_source_zip_telemetry") -> {
+                message("codewhisperer.codescan.run_scan_error")
+            }
+            else -> message
+        }
+    }
+
+    private fun getCodeScanServerExceptionMessage(e: CodeWhispererCodeScanServerException): String? =
+        e.message?.takeIf { it.startsWith("UploadArtifactToS3Exception:") }
+            ?.let { message("codewhisperer.codescan.upload_to_s3_failed") }
+
     fun handleException(coroutineContext: CoroutineContext, e: Exception, scope: CodeWhispererConstants.CodeAnalysisScope): String {
         val errorMessage = when (e) {
             is CodeWhispererException -> e.awsErrorDetails().errorMessage() ?: message("codewhisperer.codescan.run_scan_error")
-            is CodeWhispererCodeScanException -> when (e.message) {
-                message("codewhisperer.codescan.invalid_source_zip_telemetry") -> message("codewhisperer.codescan.run_scan_error")
-                else -> e.message
-            }
-            is UploadCodeScanException -> message("codewhisperer.codescan.upload_to_s3_failed")
+            is CodeWhispererCodeScanException -> getCodeScanExceptionMessage(e)
+            is CodeWhispererCodeScanServerException -> getCodeScanServerExceptionMessage(e)
             is WaiterTimeoutException, is TimeoutCancellationException -> message("codewhisperer.codescan.scan_timed_out")
             is CancellationException -> message("codewhisperer.codescan.cancelled_by_user_exception")
             else -> null
@@ -372,10 +384,17 @@ class CodeWhispererCodeScanManager(val project: Project) {
                 message("codewhisperer.codescan.file_too_large") -> message("codewhisperer.codescan.file_too_large_telemetry")
                 else -> e.message
             }
-            is UploadCodeScanException -> e.message
+            is CodeWhispererCodeScanServerException -> e.message
             is WaiterTimeoutException, is TimeoutCancellationException -> message("codewhisperer.codescan.scan_timed_out")
             is CancellationException -> message("codewhisperer.codescan.cancelled_by_user_exception")
-            else -> e.message
+            else -> when {
+                /**
+                 * Error message has text with user details(like requestId) which is specific so sending a custom error message to calculate the occurence of this event.
+                 */
+                e.message?.startsWith("Too many requests, please wait before trying again.") == true ->
+                    "Too many requests, please wait before trying again."
+                else -> e.message
+            }
         } ?: message("codewhisperer.codescan.run_scan_error_telemetry")
 
         return telemetryErrorMessage
diff --git a/plugins/toolkit/jetbrains-core/src/software/aws/toolkits/jetbrains/services/codewhisperer/codescan/CodeWhispererCodeScanSession.kt b/plugins/toolkit/jetbrains-core/src/software/aws/toolkits/jetbrains/services/codewhisperer/codescan/CodeWhispererCodeScanSession.kt
@@ -285,7 +285,8 @@ class CodeWhispererCodeScanSession(val sessionContext: CodeScanSessionContext) {
                 .build()
         )
     } catch (e: Exception) {
-        throw e
+        LOG.debug { "Create Upload URL failed: ${e.message}" }
+        throw codeScanServerException("CreateUploadUrlException: " + e.message?.let { it } ?: message("codewhisperer.codescan.run_scan_error_telemetry"))
     }
 
     private fun getUploadIntent(scope: CodeWhispererConstants.CodeAnalysisScope): UploadIntent = when (scope) {
@@ -317,8 +318,8 @@ class CodeWhispererCodeScanSession(val sessionContext: CodeScanSessionContext) {
                 IoUtils.copy(fileToUpload.inputStream(), connection.outputStream)
             }
         } catch (e: Exception) {
-            val errorMessage = e.message?.let { it } ?: message("codewhisperer.codescan.run_scan_error_telemetry")
-            throw uploadArtifactFailedError(errorMessage)
+            LOG.debug { "Artifact failed to upload in the S3 bucket: ${e.message}" }
+            throw codeScanServerException("UploadArtifactToS3Exception: " + e.message?.let { it } ?: message("codewhisperer.codescan.run_scan_error_telemetry"))
         }
     }
 
@@ -340,7 +341,7 @@ class CodeWhispererCodeScanSession(val sessionContext: CodeScanSessionContext) {
             )
         } catch (e: Exception) {
             LOG.debug { "Creating security scan failed: ${e.message}" }
-            throw e
+            throw codeScanServerException("CreateCodeScanException: " + e.message?.let { it } ?: message("codewhisperer.codescan.run_scan_error_telemetry"))
         }
     }
 
@@ -352,7 +353,7 @@ class CodeWhispererCodeScanSession(val sessionContext: CodeScanSessionContext) {
         )
     } catch (e: Exception) {
         LOG.debug { "Getting security scan failed: ${e.message}" }
-        throw e
+        throw codeScanServerException("GetCodeScanException: " + e.message?.let { it } ?: message("codewhisperer.codescan.run_scan_error_telemetry"))
     }
 
     fun listCodeScanFindings(jobId: String, nextToken: String?): ListCodeScanFindingsResponse = try {
@@ -365,7 +366,7 @@ class CodeWhispererCodeScanSession(val sessionContext: CodeScanSessionContext) {
         )
     } catch (e: Exception) {
         LOG.debug { "Listing security scan failed: ${e.message}" }
-        throw e
+        throw codeScanServerException("ListCodeScanFindingsException: " + e.message?.let { it } ?: message("codewhisperer.codescan.run_scan_error_telemetry"))
     }
 
     fun mapToCodeScanIssues(recommendations: List<String>): List<CodeWhispererCodeScanIssue> {
diff --git a/plugins/toolkit/jetbrains-core/tst/software/aws/toolkits/jetbrains/services/codewhisperer/codescan/CodeWhispererCodeFileScanTest.kt b/plugins/toolkit/jetbrains-core/tst/software/aws/toolkits/jetbrains/services/codewhisperer/codescan/CodeWhispererCodeFileScanTest.kt
@@ -311,14 +311,14 @@ class CodeWhispererCodeFileScanTest : CodeWhispererCodeScanTestBase(PythonCodeIn
     @Test
     fun `test run() - createCodeScan error`() {
         mockClient.stub {
-            onGeneric { createCodeScan(any(), any()) }.thenThrow(CodeWhispererCodeScanException::class.java)
+            onGeneric { createCodeScan(any(), any()) }.thenThrow(CodeWhispererCodeScanServerException::class.java)
         }
 
         runBlocking {
             val codeScanResponse = codeScanSessionSpy.run()
             assertThat(codeScanResponse).isInstanceOf<CodeScanResponse.Failure>()
             assertThat(codeScanResponse.responseContext.payloadContext).isEqualTo(payloadContext)
-            assertThat((codeScanResponse as CodeScanResponse.Failure).failureReason).isInstanceOf<CodeWhispererCodeScanException>()
+            assertThat((codeScanResponse as CodeScanResponse.Failure).failureReason).isInstanceOf<CodeWhispererCodeScanServerException>()
         }
     }
 
@@ -361,28 +361,28 @@ class CodeWhispererCodeFileScanTest : CodeWhispererCodeScanTestBase(PythonCodeIn
     @Test
     fun `test run() - getCodeScan error`() {
         mockClient.stub {
-            onGeneric { getCodeScan(any(), any()) }.thenThrow(CodeWhispererException::class.java)
+            onGeneric { getCodeScan(any(), any()) }.thenThrow(CodeWhispererCodeScanServerException::class.java)
         }
 
         runBlocking {
             val codeScanResponse = codeScanSessionSpy.run()
             assertThat(codeScanResponse).isInstanceOf<CodeScanResponse.Failure>()
             assertThat(codeScanResponse.responseContext.payloadContext).isEqualTo(payloadContext)
-            assertThat((codeScanResponse as CodeScanResponse.Failure).failureReason).isInstanceOf<CodeWhispererException>()
+            assertThat((codeScanResponse as CodeScanResponse.Failure).failureReason).isInstanceOf<CodeWhispererCodeScanServerException>()
         }
     }
 
     @Test
     fun `test run() - listCodeScanFindings error`() {
         mockClient.stub {
-            onGeneric { listCodeScanFindings(any(), any()) }.thenThrow(CodeWhispererException::class.java)
+            onGeneric { listCodeScanFindings(any(), any()) }.thenThrow(CodeWhispererCodeScanServerException::class.java)
         }
 
         runBlocking {
             val codeScanResponse = codeScanSessionSpy.run()
             assertThat(codeScanResponse).isInstanceOf<CodeScanResponse.Failure>()
             assertThat(codeScanResponse.responseContext.payloadContext).isEqualTo(payloadContext)
-            assertThat((codeScanResponse as CodeScanResponse.Failure).failureReason).isInstanceOf<CodeWhispererException>()
+            assertThat((codeScanResponse as CodeScanResponse.Failure).failureReason).isInstanceOf<CodeWhispererCodeScanServerException>()
         }
     }
 
diff --git a/plugins/toolkit/jetbrains-core/tst/software/aws/toolkits/jetbrains/services/codewhisperer/codescan/CodeWhispererCodeScanTest.kt b/plugins/toolkit/jetbrains-core/tst/software/aws/toolkits/jetbrains/services/codewhisperer/codescan/CodeWhispererCodeScanTest.kt
@@ -222,14 +222,14 @@ class CodeWhispererCodeScanTest : CodeWhispererCodeScanTestBase(PythonCodeInsigh
     @Test
     fun `test run() - createCodeScan error`() {
         mockClient.stub {
-            onGeneric { createCodeScan(any(), any()) }.thenThrow(CodeWhispererCodeScanException::class.java)
+            onGeneric { createCodeScan(any(), any()) }.thenThrow(CodeWhispererCodeScanServerException::class.java)
         }
 
         runBlocking {
             val codeScanResponse = codeScanSessionSpy.run()
             assertThat(codeScanResponse).isInstanceOf<CodeScanResponse.Failure>()
             assertThat(codeScanResponse.responseContext.payloadContext).isEqualTo(payloadContext)
-            assertThat((codeScanResponse as CodeScanResponse.Failure).failureReason).isInstanceOf<CodeWhispererCodeScanException>()
+            assertThat((codeScanResponse as CodeScanResponse.Failure).failureReason).isInstanceOf<CodeWhispererCodeScanServerException>()
         }
     }
 
@@ -272,28 +272,28 @@ class CodeWhispererCodeScanTest : CodeWhispererCodeScanTestBase(PythonCodeInsigh
     @Test
     fun `test run() - getCodeScan error`() {
         mockClient.stub {
-            onGeneric { getCodeScan(any(), any()) }.thenThrow(CodeWhispererException::class.java)
+            onGeneric { getCodeScan(any(), any()) }.thenThrow(CodeWhispererCodeScanServerException::class.java)
         }
 
         runBlocking {
             val codeScanResponse = codeScanSessionSpy.run()
             assertThat(codeScanResponse).isInstanceOf<CodeScanResponse.Failure>()
             assertThat(codeScanResponse.responseContext.payloadContext).isEqualTo(payloadContext)
-            assertThat((codeScanResponse as CodeScanResponse.Failure).failureReason).isInstanceOf<CodeWhispererException>()
+            assertThat((codeScanResponse as CodeScanResponse.Failure).failureReason).isInstanceOf<CodeWhispererCodeScanServerException>()
         }
     }
 
     @Test
     fun `test run() - listCodeScanFindings error`() {
         mockClient.stub {
-            onGeneric { listCodeScanFindings(any(), any()) }.thenThrow(CodeWhispererException::class.java)
+            onGeneric { listCodeScanFindings(any(), any()) }.thenThrow(CodeWhispererCodeScanServerException::class.java)
         }
 
         runBlocking {
             val codeScanResponse = codeScanSessionSpy.run()
             assertThat(codeScanResponse).isInstanceOf<CodeScanResponse.Failure>()
             assertThat(codeScanResponse.responseContext.payloadContext).isEqualTo(payloadContext)
-            assertThat((codeScanResponse as CodeScanResponse.Failure).failureReason).isInstanceOf<CodeWhispererException>()
+            assertThat((codeScanResponse as CodeScanResponse.Failure).failureReason).isInstanceOf<CodeWhispererCodeScanServerException>()
         }
     }
 

Original file line number	Diff line number	Diff line change
`@@ -285,7 +285,8 @@ class CodeWhispererCodeScanSession(val sessionContext: CodeScanSessionContext) {`
`285`	`285`	`.build()`
`286`	`286`	`)`
`287`	`287`	`} catch (e: Exception) {`
`288`		`- throw e`
	`288`	`+ LOG.debug { "Create Upload URL failed: ${e.message}" }`
	`289`	`+ throw codeScanServerException("CreateUploadUrlException: " + e.message?.let { it } ?: message("codewhisperer.codescan.run_scan_error_telemetry"))`
`289`	`290`	`}`
`290`	`291`
`291`	`292`	`private fun getUploadIntent(scope: CodeWhispererConstants.CodeAnalysisScope): UploadIntent = when (scope) {`
`@@ -317,8 +318,8 @@ class CodeWhispererCodeScanSession(val sessionContext: CodeScanSessionContext) {`
`317`	`318`	`IoUtils.copy(fileToUpload.inputStream(), connection.outputStream)`
`318`	`319`	`}`
`319`	`320`	`} catch (e: Exception) {`
`320`		`- val errorMessage = e.message?.let { it } ?: message("codewhisperer.codescan.run_scan_error_telemetry")`
`321`		`- throw uploadArtifactFailedError(errorMessage)`
	`321`	`+ LOG.debug { "Artifact failed to upload in the S3 bucket: ${e.message}" }`
	`322`	`+ throw codeScanServerException("UploadArtifactToS3Exception: " + e.message?.let { it } ?: message("codewhisperer.codescan.run_scan_error_telemetry"))`
`322`	`323`	`}`
`323`	`324`	`}`
`324`	`325`
`@@ -340,7 +341,7 @@ class CodeWhispererCodeScanSession(val sessionContext: CodeScanSessionContext) {`
`340`	`341`	`)`
`341`	`342`	`} catch (e: Exception) {`
`342`	`343`	`LOG.debug { "Creating security scan failed: ${e.message}" }`
`343`		`- throw e`
	`344`	`+ throw codeScanServerException("CreateCodeScanException: " + e.message?.let { it } ?: message("codewhisperer.codescan.run_scan_error_telemetry"))`
`344`	`345`	`}`
`345`	`346`	`}`
`346`	`347`
`@@ -352,7 +353,7 @@ class CodeWhispererCodeScanSession(val sessionContext: CodeScanSessionContext) {`
`352`	`353`	`)`
`353`	`354`	`} catch (e: Exception) {`
`354`	`355`	`LOG.debug { "Getting security scan failed: ${e.message}" }`
`355`		`- throw e`
	`356`	`+ throw codeScanServerException("GetCodeScanException: " + e.message?.let { it } ?: message("codewhisperer.codescan.run_scan_error_telemetry"))`
`356`	`357`	`}`
`357`	`358`
`358`	`359`	`fun listCodeScanFindings(jobId: String, nextToken: String?): ListCodeScanFindingsResponse = try {`
`@@ -365,7 +366,7 @@ class CodeWhispererCodeScanSession(val sessionContext: CodeScanSessionContext) {`
`365`	`366`	`)`
`366`	`367`	`} catch (e: Exception) {`
`367`	`368`	`LOG.debug { "Listing security scan failed: ${e.message}" }`
`368`		`- throw e`
	`369`	`+ throw codeScanServerException("ListCodeScanFindingsException: " + e.message?.let { it } ?: message("codewhisperer.codescan.run_scan_error_telemetry"))`
`369`	`370`	`}`
`370`	`371`
`371`	`372`	`fun mapToCodeScanIssues(recommendations: List<String>): List<CodeWhispererCodeScanIssue> {`
Original file line number	Diff line number	Diff line change
`@@ -311,14 +311,14 @@ class CodeWhispererCodeFileScanTest : CodeWhispererCodeScanTestBase(PythonCodeIn`
`311`	`311`	`@Test`
`312`	`312`	fun `test run() - createCodeScan error`() {
`313`	`313`	`mockClient.stub {`
`314`		`- onGeneric { createCodeScan(any(), any()) }.thenThrow(CodeWhispererCodeScanException::class.java)`
	`314`	`+ onGeneric { createCodeScan(any(), any()) }.thenThrow(CodeWhispererCodeScanServerException::class.java)`
`315`	`315`	`}`
`316`	`316`
`317`	`317`	`runBlocking {`
`318`	`318`	`val codeScanResponse = codeScanSessionSpy.run()`
`319`	`319`	`assertThat(codeScanResponse).isInstanceOf<CodeScanResponse.Failure>()`
`320`	`320`	`assertThat(codeScanResponse.responseContext.payloadContext).isEqualTo(payloadContext)`
`321`		`- assertThat((codeScanResponse as CodeScanResponse.Failure).failureReason).isInstanceOf<CodeWhispererCodeScanException>()`
	`321`	`+ assertThat((codeScanResponse as CodeScanResponse.Failure).failureReason).isInstanceOf<CodeWhispererCodeScanServerException>()`
`322`	`322`	`}`
`323`	`323`	`}`
`324`	`324`
`@@ -361,28 +361,28 @@ class CodeWhispererCodeFileScanTest : CodeWhispererCodeScanTestBase(PythonCodeIn`
`361`	`361`	`@Test`
`362`	`362`	fun `test run() - getCodeScan error`() {
`363`	`363`	`mockClient.stub {`
`364`		`- onGeneric { getCodeScan(any(), any()) }.thenThrow(CodeWhispererException::class.java)`
	`364`	`+ onGeneric { getCodeScan(any(), any()) }.thenThrow(CodeWhispererCodeScanServerException::class.java)`
`365`	`365`	`}`
`366`	`366`
`367`	`367`	`runBlocking {`
`368`	`368`	`val codeScanResponse = codeScanSessionSpy.run()`
`369`	`369`	`assertThat(codeScanResponse).isInstanceOf<CodeScanResponse.Failure>()`
`370`	`370`	`assertThat(codeScanResponse.responseContext.payloadContext).isEqualTo(payloadContext)`
`371`		`- assertThat((codeScanResponse as CodeScanResponse.Failure).failureReason).isInstanceOf<CodeWhispererException>()`
	`371`	`+ assertThat((codeScanResponse as CodeScanResponse.Failure).failureReason).isInstanceOf<CodeWhispererCodeScanServerException>()`
`372`	`372`	`}`
`373`	`373`	`}`
`374`	`374`
`375`	`375`	`@Test`
`376`	`376`	fun `test run() - listCodeScanFindings error`() {
`377`	`377`	`mockClient.stub {`
`378`		`- onGeneric { listCodeScanFindings(any(), any()) }.thenThrow(CodeWhispererException::class.java)`
	`378`	`+ onGeneric { listCodeScanFindings(any(), any()) }.thenThrow(CodeWhispererCodeScanServerException::class.java)`
`379`	`379`	`}`
`380`	`380`
`381`	`381`	`runBlocking {`
`382`	`382`	`val codeScanResponse = codeScanSessionSpy.run()`
`383`	`383`	`assertThat(codeScanResponse).isInstanceOf<CodeScanResponse.Failure>()`
`384`	`384`	`assertThat(codeScanResponse.responseContext.payloadContext).isEqualTo(payloadContext)`
`385`		`- assertThat((codeScanResponse as CodeScanResponse.Failure).failureReason).isInstanceOf<CodeWhispererException>()`
	`385`	`+ assertThat((codeScanResponse as CodeScanResponse.Failure).failureReason).isInstanceOf<CodeWhispererCodeScanServerException>()`
`386`	`386`	`}`
`387`	`387`	`}`
`388`	`388`
Original file line number	Diff line number	Diff line change
`@@ -222,14 +222,14 @@ class CodeWhispererCodeScanTest : CodeWhispererCodeScanTestBase(PythonCodeInsigh`
`222`	`222`	`@Test`
`223`	`223`	fun `test run() - createCodeScan error`() {
`224`	`224`	`mockClient.stub {`
`225`		`- onGeneric { createCodeScan(any(), any()) }.thenThrow(CodeWhispererCodeScanException::class.java)`
	`225`	`+ onGeneric { createCodeScan(any(), any()) }.thenThrow(CodeWhispererCodeScanServerException::class.java)`
`226`	`226`	`}`
`227`	`227`
`228`	`228`	`runBlocking {`
`229`	`229`	`val codeScanResponse = codeScanSessionSpy.run()`
`230`	`230`	`assertThat(codeScanResponse).isInstanceOf<CodeScanResponse.Failure>()`
`231`	`231`	`assertThat(codeScanResponse.responseContext.payloadContext).isEqualTo(payloadContext)`
`232`		`- assertThat((codeScanResponse as CodeScanResponse.Failure).failureReason).isInstanceOf<CodeWhispererCodeScanException>()`
	`232`	`+ assertThat((codeScanResponse as CodeScanResponse.Failure).failureReason).isInstanceOf<CodeWhispererCodeScanServerException>()`
`233`	`233`	`}`
`234`	`234`	`}`
`235`	`235`
`@@ -272,28 +272,28 @@ class CodeWhispererCodeScanTest : CodeWhispererCodeScanTestBase(PythonCodeInsigh`
`272`	`272`	`@Test`
`273`	`273`	fun `test run() - getCodeScan error`() {
`274`	`274`	`mockClient.stub {`
`275`		`- onGeneric { getCodeScan(any(), any()) }.thenThrow(CodeWhispererException::class.java)`
	`275`	`+ onGeneric { getCodeScan(any(), any()) }.thenThrow(CodeWhispererCodeScanServerException::class.java)`
`276`	`276`	`}`
`277`	`277`
`278`	`278`	`runBlocking {`
`279`	`279`	`val codeScanResponse = codeScanSessionSpy.run()`
`280`	`280`	`assertThat(codeScanResponse).isInstanceOf<CodeScanResponse.Failure>()`
`281`	`281`	`assertThat(codeScanResponse.responseContext.payloadContext).isEqualTo(payloadContext)`
`282`		`- assertThat((codeScanResponse as CodeScanResponse.Failure).failureReason).isInstanceOf<CodeWhispererException>()`
	`282`	`+ assertThat((codeScanResponse as CodeScanResponse.Failure).failureReason).isInstanceOf<CodeWhispererCodeScanServerException>()`
`283`	`283`	`}`
`284`	`284`	`}`
`285`	`285`
`286`	`286`	`@Test`
`287`	`287`	fun `test run() - listCodeScanFindings error`() {
`288`	`288`	`mockClient.stub {`
`289`		`- onGeneric { listCodeScanFindings(any(), any()) }.thenThrow(CodeWhispererException::class.java)`
	`289`	`+ onGeneric { listCodeScanFindings(any(), any()) }.thenThrow(CodeWhispererCodeScanServerException::class.java)`
`290`	`290`	`}`
`291`	`291`
`292`	`292`	`runBlocking {`
`293`	`293`	`val codeScanResponse = codeScanSessionSpy.run()`
`294`	`294`	`assertThat(codeScanResponse).isInstanceOf<CodeScanResponse.Failure>()`
`295`	`295`	`assertThat(codeScanResponse.responseContext.payloadContext).isEqualTo(payloadContext)`
`296`		`- assertThat((codeScanResponse as CodeScanResponse.Failure).failureReason).isInstanceOf<CodeWhispererException>()`
	`296`	`+ assertThat((codeScanResponse as CodeScanResponse.Failure).failureReason).isInstanceOf<CodeWhispererCodeScanServerException>()`
`297`	`297`	`}`
`298`	`298`	`}`
`299`	`299`