From d42f8210224c3d021755488418e3c1b2ffc02d36 Mon Sep 17 00:00:00 2001
From: Josh Pinkney <jpink@amazon.com>
Date: Sat, 22 Feb 2025 13:36:09 -0500
Subject: [PATCH] test

---
 packages/amazonq/test/e2e/amazonq/chat.test.ts           | 3 +--
 packages/amazonq/test/e2e/amazonq/framework/framework.ts | 3 +++
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/packages/amazonq/test/e2e/amazonq/chat.test.ts b/packages/amazonq/test/e2e/amazonq/chat.test.ts
index afceb364013..bfb1ea6afde 100644
--- a/packages/amazonq/test/e2e/amazonq/chat.test.ts
+++ b/packages/amazonq/test/e2e/amazonq/chat.test.ts
@@ -36,7 +36,7 @@ describe('Amazon Q Chat', function () {
         // Make sure you're logged in before every test
         registerAuthHook('amazonq-test-account')
         framework = new qTestingFramework('cwc', true, [])
-        tab = framework.createTab()
+        tab = framework.getSelectedTab()
 
         /**
          * Since sending messages to the UI is asynchronous, race conditions can occur
@@ -45,7 +45,6 @@ describe('Amazon Q Chat', function () {
          */
         const ok = await waitUntil(
             async () => {
-                console.log(JSON.stringify(tab.getStore()))
                 return tab.getStore().tabTitle === 'Chat'
             },
             {
diff --git a/packages/amazonq/test/e2e/amazonq/framework/framework.ts b/packages/amazonq/test/e2e/amazonq/framework/framework.ts
index 0cc9a9298ee..5cef594f3d2 100644
--- a/packages/amazonq/test/e2e/amazonq/framework/framework.ts
+++ b/packages/amazonq/test/e2e/amazonq/framework/framework.ts
@@ -132,5 +132,8 @@ export class qTestingFramework {
     public dispose() {
         vscode.Disposable.from(...this.disposables).dispose()
         this.mynahUI.destroy()
+
+        // create a refresh instance of the dom each time
+        injectJSDOM()
     }
 }