Adjusted CodeGenerationStateTest to be aligned with new polling rate

JuusticeT · JuusticeT · commit 35eea6a78034 · 2024-10-23T17:56:49.000-04:00
diff --git a/plugins/amazonq/chat/jetbrains-community/tst/software/aws/toolkits/jetbrains/services/amazonqFeatureDev/session/CodeGenerationStateTest.kt b/plugins/amazonq/chat/jetbrains-community/tst/software/aws/toolkits/jetbrains/services/amazonqFeatureDev/session/CodeGenerationStateTest.kt
@@ -164,7 +164,7 @@ class CodeGenerationStateTest : FeatureDevTestBase() {
         }
 
         verify(exactly = 1) { featureDevService.startTaskAssistCodeGeneration(testConversationId, testUploadId, userMessage) }
-        verify(exactly = 180) { featureDevService.getTaskAssistCodeGeneration(testConversationId, testCodeGenerationId) }
+        verify(exactly = 360) { featureDevService.getTaskAssistCodeGeneration(testConversationId, testCodeGenerationId) }
         coVerify(exactly = 0) { featureDevService.exportTaskAssistArchiveResult(testConversationId) }
     }
 }

Original file line number	Diff line number	Diff line change
`@@ -164,7 +164,7 @@ class CodeGenerationStateTest : FeatureDevTestBase() {`
`164`	`164`	`}`
`165`	`165`
`166`	`166`	`verify(exactly = 1) { featureDevService.startTaskAssistCodeGeneration(testConversationId, testUploadId, userMessage) }`
`167`		`- verify(exactly = 180) { featureDevService.getTaskAssistCodeGeneration(testConversationId, testCodeGenerationId) }`
	`167`	`+ verify(exactly = 360) { featureDevService.getTaskAssistCodeGeneration(testConversationId, testCodeGenerationId) }`
`168`	`168`	`coVerify(exactly = 0) { featureDevService.exportTaskAssistArchiveResult(testConversationId) }`
`169`	`169`	`}`
`170`	`170`	`}`