From 565cf5fee8cbca5957c74143d5f57151e963ab8b Mon Sep 17 00:00:00 2001
From: Josh Pinkney <jpink@amazon.com>
Date: Sat, 22 Feb 2025 22:29:46 -0500
Subject: [PATCH] test

---
 packages/amazonq/test/e2e/amazonq/chat.test.ts  |  6 +++---
 packages/amazonq/test/e2e/inline/inline.test.ts | 10 +++++-----
 2 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/packages/amazonq/test/e2e/amazonq/chat.test.ts b/packages/amazonq/test/e2e/amazonq/chat.test.ts
index 9f66c0bb07c..4fd6e2532e0 100644
--- 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
index 260f6b528f9..21903b87bef 100644
--- 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)}`