From ee26bdd6b01b21855a43464674ac392c86bc83f8 Mon Sep 17 00:00:00 2001
From: Faisal Amir <urmauur@gmail.com>
Date: Fri, 13 Dec 2024 15:10:00 +0700
Subject: [PATCH] chore: update e2e

---
 electron/tests/e2e/thread.e2e.spec.ts | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/electron/tests/e2e/thread.e2e.spec.ts b/electron/tests/e2e/thread.e2e.spec.ts
index 6231f6c049..54e5ffda07 100644
--- a/electron/tests/e2e/thread.e2e.spec.ts
+++ b/electron/tests/e2e/thread.e2e.spec.ts
@@ -15,8 +15,13 @@ test('Select GPT model from Hub and Chat with Invalid API Key', async ({
 
   await page.getByTestId('txt-input-chat').fill('dummy value')
 
-  await page.getByTestId('btn-deny-product-analytics').click({ force: true })
-  await page.getByTestId('btn-send-chat').click({ force: true })
+  const denyButton = page.locator('[data-testid="btn-deny-product-analytics"]')
+
+  if ((await denyButton.count()) > 0) {
+    await denyButton.click({ force: true })
+  } else {
+    await page.getByTestId('btn-send-chat').click({ force: true })
+  }
 
   await page.waitForFunction(
     () => {