test

aws · Feb 23, 2025 · 565cf5f · 565cf5f
1 parent 0db682f
commit 565cf5f
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git a/packages/amazonq/test/e2e/amazonq/chat.test.ts b/packages/amazonq/test/e2e/amazonq/chat.test.ts
@@ -12,7 +12,6 @@ import { assertContextCommands, assertQuickActions } from './assert'
 import { registerAuthHook, using } from 'aws-core-vscode/test'
 import { loginToIdC } from './utils/setup'
 import { webviewConstants } from 'aws-core-vscode/amazonq'
-import { sleep } from 'aws-core-vscode/shared'
 
 describe('Amazon Q Chat', function () {
     let framework: qTestingFramework
@@ -35,9 +34,10 @@ describe('Amazon Q Chat', function () {
     beforeEach(async () => {
         // Make sure you're logged in before every test
         registerAuthHook('amazonq-test-account')
-        await sleep(5000)
+        // await sleep(5000)
         framework = new qTestingFramework('cwc', true, [])
-        tab = framework.getTabs()[0] // use the default tab that gets created
+        tab = framework.createTab()
+        console.log('number of tabs: ' + framework.getTabs().length)
         store = tab.getStore()
     })
 

diff --git a/packages/amazonq/test/e2e/inline/inline.test.ts b/packages/amazonq/test/e2e/inline/inline.test.ts
@@ -73,11 +73,11 @@ describe('Amazon Q Inline', async function () {
     }
 
     async function waitForRecommendations() {
-        const ok = await waitUntil(
-            async () =>
-                RecommendationHandler.instance.isSuggestionVisible() || session.getSuggestionState(0) === 'Showed',
-            waitOptions
-        )
+        const ok1 = await waitUntil(async () => session.getSuggestionState(0) === 'Showed', waitOptions)
+        if (!ok1) {
+            throw new Error(`Suggestion did not show: ${JSON.stringify(session.suggestionStates)}`)
+        }
+        const ok = await waitUntil(async () => RecommendationHandler.instance.isSuggestionVisible(), waitOptions)
         if (!ok) {
             throw new Error(
                 `Suggestions failed to become visible. Suggestion States: ${JSON.stringify(session.suggestionStates)}`