test: use expect.poll in fs-serve playground (vitejs#20240)

Author: sapphi-red

playground/fs-serve/__tests__/base/fs-serve-base.spec.ts

@@ -15,93 +15,120 @@ describe.runIf(isServe)('main', () => {
   })
 
   test('default import', async () => {
-    expect(await page.textContent('.full')).toBe(stringified)
+    await expect.poll(() => page.textContent('.full')).toBe(stringified)
   })
 
   test('named import', async () => {
-    expect(await page.textContent('.named')).toBe(testJSON.msg)
+    await expect.poll(() => page.textContent('.named')).toBe(testJSON.msg)
   })
 
   test('safe fetch', async () => {
     expect(await page.textContent('.safe-fetch')).toMatch('KEY=safe')
-    expect(await page.textContent('.safe-fetch-status')).toBe('200')
+    await expect.poll(() => page.textContent('.safe-fetch-status')).toBe('200')
   })
 
   test('safe fetch with query', async () => {
     expect(await page.textContent('.safe-fetch-query')).toMatch('KEY=safe')
-    expect(await page.textContent('.safe-fetch-query-status')).toBe('200')
+    await expect
+      .poll(() => page.textContent('.safe-fetch-query-status'))
+      .toBe('200')
   })
 
   test('safe fetch with special characters', async () => {
     expect(
       await page.textContent('.safe-fetch-subdir-special-characters'),
     ).toMatch('KEY=safe')
-    expect(
-      await page.textContent('.safe-fetch-subdir-special-characters-status'),
-    ).toBe('200')
+    await expect
+      .poll(() =>
+        page.textContent('.safe-fetch-subdir-special-characters-status'),
+      )
+      .toBe('200')
   })
 
   test('unsafe fetch', async () => {
     expect(await page.textContent('.unsafe-fetch')).toMatch('403 Restricted')
-    expect(await page.textContent('.unsafe-fetch-status')).toBe('403')
+    await expect
+      .poll(() => page.textContent('.unsafe-fetch-status'))
+      .toBe('403')
   })
 
   test('unsafe fetch with special characters (#8498)', async () => {
-    expect(await page.textContent('.unsafe-fetch-8498')).toBe('')
-    expect(await page.textContent('.unsafe-fetch-8498-status')).toBe('404')
+    await expect.poll(() => page.textContent('.unsafe-fetch-8498')).toBe('')
+    await expect
+      .poll(() => page.textContent('.unsafe-fetch-8498-status'))
+      .toBe('404')
   })
 
   test('unsafe fetch with special characters 2 (#8498)', async () => {
-    expect(await page.textContent('.unsafe-fetch-8498-2')).toBe('')
-    expect(await page.textContent('.unsafe-fetch-8498-2-status')).toBe('404')
+    await expect.poll(() => page.textContent('.unsafe-fetch-8498-2')).toBe('')
+    await expect
+      .poll(() => page.textContent('.unsafe-fetch-8498-2-status'))
+      .toBe('404')
   })
 
   test('safe fs fetch', async () => {
-    expect(await page.textContent('.safe-fs-fetch')).toBe(stringified)
-    expect(await page.textContent('.safe-fs-fetch-status')).toBe('200')
+    await expect
+      .poll(() => page.textContent('.safe-fs-fetch'))
+      .toBe(stringified)
+    await expect
+      .poll(() => page.textContent('.safe-fs-fetch-status'))
+      .toBe('200')
   })
 
   test('safe fs fetch', async () => {
-    expect(await page.textContent('.safe-fs-fetch-query')).toBe(stringified)
-    expect(await page.textContent('.safe-fs-fetch-query-status')).toBe('200')
+    await expect
+      .poll(() => page.textContent('.safe-fs-fetch-query'))
+      .toBe(stringified)
+    await expect
+      .poll(() => page.textContent('.safe-fs-fetch-query-status'))
+      .toBe('200')
   })
 
   test('safe fs fetch with special characters', async () => {
-    expect(await page.textContent('.safe-fs-fetch-special-characters')).toBe(
-      stringified,
-    )
-    expect(
-      await page.textContent('.safe-fs-fetch-special-characters-status'),
-    ).toBe('200')
+    await expect
+      .poll(() => page.textContent('.safe-fs-fetch-special-characters'))
+      .toBe(stringified)
+    await expect
+      .poll(() => page.textContent('.safe-fs-fetch-special-characters-status'))
+      .toBe('200')
   })
 
   test('unsafe fs fetch', async () => {
-    expect(await page.textContent('.unsafe-fs-fetch')).toBe('')
-    expect(await page.textContent('.unsafe-fs-fetch-status')).toBe('403')
+    await expect.poll(() => page.textContent('.unsafe-fs-fetch')).toBe('')
+    await expect
+      .poll(() => page.textContent('.unsafe-fs-fetch-status'))
+      .toBe('403')
   })
 
   test('unsafe fs fetch with special characters (#8498)', async () => {
-    expect(await page.textContent('.unsafe-fs-fetch-8498')).toBe('')
-    expect(await page.textContent('.unsafe-fs-fetch-8498-status')).toBe('404')
+    await expect.poll(() => page.textContent('.unsafe-fs-fetch-8498')).toBe('')
+    await expect
+      .poll(() => page.textContent('.unsafe-fs-fetch-8498-status'))
+      .toBe('404')
   })
 
   test('unsafe fs fetch with special characters 2 (#8498)', async () => {
-    expect(await page.textContent('.unsafe-fs-fetch-8498-2')).toBe('')
-    expect(await page.textContent('.unsafe-fs-fetch-8498-2-status')).toBe('404')
+    await expect
+      .poll(() => page.textContent('.unsafe-fs-fetch-8498-2'))
+      .toBe('')
+    await expect
+      .poll(() => page.textContent('.unsafe-fs-fetch-8498-2-status'))
+      .toBe('404')
   })
 
   test('nested entry', async () => {
-    expect(await page.textContent('.nested-entry')).toBe('foobar')
+    await expect.poll(() => page.textContent('.nested-entry')).toBe('foobar')
   })
 
   test('denied', async () => {
-    expect(await page.textContent('.unsafe-dotenv')).toBe('403')
+    await expect.poll(() => page.textContent('.unsafe-dotenv')).toBe('403')
   })
 
   test('denied EnV casing', async () => {
     // It is 403 in case insensitive system, 404 in others
-    const code = await page.textContent('.unsafe-dotEnV-casing')
-    expect(code === '403' || code === '404').toBeTruthy()
+    await expect
+      .poll(() => page.textContent('.unsafe-dotEnV-casing'))
+      .toStrictEqual(expect.toBeOneOf(['403', '404']))
   })
 })