chore: update e2e

janhq · Dec 13, 2024 · ee26bdd · ee26bdd
1 parent c9027a9
commit ee26bdd
Showing 1 changed file with 7 additions and 2 deletions.
diff --git 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(
     () => {