de-flake test

vercel · May 20, 2024 · 74616f9 · 74616f9
1 parent 66ecd71
commit 74616f9
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/test/e2e/app-dir/prefetch-searchparam/prefetch-searchparam.test.ts b/test/e2e/app-dir/prefetch-searchparam/prefetch-searchparam.test.ts
@@ -1,4 +1,5 @@
 import { nextTestSetup } from 'e2e-utils'
+import { retry } from '../../../lib/next-test-utils'
 
 describe('prefetch-searchparam', () => {
   const { next } = nextTestSetup({
@@ -11,12 +12,14 @@ describe('prefetch-searchparam', () => {
 
     // navigate to different search param, should update the search param
     await browser.elementByCss('[href="/?q=bar"]').click()
-    await browser.waitForElementByCss('p', 5000)
-    expect(await browser.elementByCss('p').text()).toBe('{"q":"bar"}')
+    await retry(async () => {
+      expect(await browser.elementByCss('p').text()).toBe('{"q":"bar"}')
+    })
 
     // navigate to home, should clear the searchParams value
     await browser.elementByCss('[href="/"]').click()
-    await browser.waitForElementByCss('p', 5000)
-    expect(await browser.elementByCss('p').text()).toBe('{}')
+    await retry(async () => {
+      expect(await browser.elementByCss('p').text()).toBe('{}')
+    })
   })
 })