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( () => {