test

Author: jpinkney-aws

packages/amazonq/test/unit/codewhisperer/testUtil.ts

@@ -12,19 +12,18 @@ import { assertTextEditorContains } from 'aws-core-vscode/test'
 // Note: RecommendationHandler.isSuggestionVisible seems not to work well, hence not using it
 export async function waitUntilSuggestionSeen(index: number = 0) {
     const session = CodeWhispererSessionState.instance.getSession()
-    const state = await waitUntil(
+    const ok = await waitUntil(
         async () => {
-            const r = session.getSuggestionState(index)
-            if (r) {
-                return r
-            }
+            console.log('Suggestion state: %O', session.getSuggestionState(index))
+            return session.getSuggestionState(index) === 'Showed'
         },
         {
             interval: 500,
+            timeout: 5000,
         }
     )
 
-    assert.deepStrictEqual(state, 'Showed')
+    assert.ok(ok === true)
 }
 
 export async function acceptByTab() {