chore: update

Author: Timeless0911

tests/integration/cli/build-watch/build.test.ts

@@ -2,7 +2,7 @@ import { exec, spawn } from 'node:child_process';
 import path from 'node:path';
 import { after } from 'node:test';
 import fse from 'fs-extra';
-import { awaitFileChanges, awaitFileExists } from 'test-helper';
+import { expectFile, expectFileChanges } from 'test-helper';
 import { describe, expect, test } from 'vitest';
 
 describe('build --watch command', async () => {
@@ -30,7 +30,7 @@ export default defineConfig({
       cwd: __dirname,
     });
 
-    await awaitFileExists(distEsmIndexFile);
+    await expectFile(distEsmIndexFile);
 
     fse.outputFileSync(
       tempConfigFile,
@@ -49,7 +49,7 @@ export default defineConfig({
   `,
     );
 
-    await awaitFileExists(dist1EsmIndexFile);
+    await expectFile(dist1EsmIndexFile);
 
     process.kill();
   });
@@ -103,12 +103,12 @@ export default defineConfig({
         shell: true,
       },
     );
-    await awaitFileExists(distIndexFile);
+    await expectFile(distIndexFile);
 
     fse.outputFileSync(srcFooFile, `export const foo = 'foo';`);
     fse.outputFileSync(srcFoo2File, `export const foo2 = 'foo2';`);
-    await awaitFileExists(distFooFile);
-    await awaitFileExists(distFoo2File);
+    await expectFile(distFooFile);
+    await expectFile(distFoo2File);
     const content1 = await fse.readFile(distFooFile, 'utf-8');
     expect(content1!).toMatchInlineSnapshot(`
       "const foo = 'foo';
@@ -127,7 +127,7 @@ export default defineConfig({
     fse.removeSync(srcIndexFile);
 
     // change
-    const wait = await awaitFileChanges(distFooFile);
+    const wait = await expectFileChanges(distFooFile);
     fse.outputFileSync(srcFooFile, `export const foo = 'foo1';`);
     await wait();
     const content3 = await fse.readFile(distFooFile, 'utf-8');

tests/integration/cli/mf/mf.test.ts

@@ -4,7 +4,7 @@ import { describe } from 'node:test';
 import { pluginModuleFederation } from '@module-federation/rsbuild-plugin';
 import { startMFDevServer } from '@rslib/core';
 import fse, { existsSync } from 'fs-extra';
-import { awaitFileExists } from 'test-helper';
+import { expectFile } from 'test-helper';
 import { expect, test } from 'vitest';
 
 describe('mf-dev', () => {
@@ -25,7 +25,7 @@ describe('mf-dev', () => {
       },
     });
 
-    await awaitFileExists(distPath);
+    await expectFile(distPath);
 
     const rspackConfigContent = await fse.readFile(rspackConfigFile, 'utf-8');
     expect(rspackConfigContent).toContain(`nodeEnv: 'development'`);
@@ -51,8 +51,8 @@ describe('mf-dev', () => {
       cwd: fixturePath,
     });
 
-    await awaitFileExists(distPath1);
-    await awaitFileExists(distPath2);
+    await expectFile(distPath1);
+    await expectFile(distPath2);
 
     expect(existsSync(distPath1)).toBe(true);
     expect(existsSync(distPath2)).toBe(true);

tests/integration/plugins/index.test.ts

@@ -1,7 +1,7 @@
 import { exec } from 'node:child_process';
 import fs from 'node:fs';
 import { join } from 'node:path';
-import { awaitFileExists, buildAndGetResults } from 'test-helper';
+import { buildAndGetResults, expectFile } from 'test-helper';
 import { expect, test } from 'vitest';
 import { distIndex } from './basic/rslib.config';
 import { plugin1Path, plugin2Path } from './mf-dev/rslib.config';
@@ -20,8 +20,8 @@ test('should merge plugins correctly', async () => {
     cwd: fixturePath,
   });
 
-  await awaitFileExists(plugin1Path);
-  await awaitFileExists(plugin2Path);
+  await expectFile(plugin1Path);
+  await expectFile(plugin2Path);
 
   const content1 = fs.readFileSync(plugin1Path, 'utf-8');
   const content2 = fs.readFileSync(plugin2Path, 'utf-8');

tests/integration/server/index.test.ts

@@ -2,7 +2,7 @@ import { exec } from 'node:child_process';
 import { existsSync } from 'node:fs';
 import { join } from 'node:path';
 import fse from 'fs-extra';
-import { awaitFileExists, buildAndGetResults } from 'test-helper';
+import { buildAndGetResults, expectFile } from 'test-helper';
 import { describe, expect, test } from 'vitest';
 
 describe('server config', async () => {
@@ -32,7 +32,7 @@ describe('server config', async () => {
       },
     });
 
-    await awaitFileExists(doneFile);
+    await expectFile(doneFile);
 
     const rsbuildConfigContent = await fse.readFile(rsbuildConfigFile, 'utf-8');
     expect(rsbuildConfigContent).toContain('open: true');

tests/scripts/helper.ts

@@ -1,5 +1,7 @@
+import fs from 'node:fs';
 import { platform } from 'node:os';
 import { join } from 'node:path';
+import { expect } from '@playwright/test';
 import fse from 'fs-extra';
 import { type GlobOptions, convertPathToPattern, glob } from 'tinyglobby';
 
@@ -62,57 +64,23 @@ export const proxyConsole = (
   };
 };
 
-export const waitFor = async (
-  fn: () => boolean,
-  {
-    maxChecks = 100,
-    interval = 20,
-  }: {
-    maxChecks?: number;
-    interval?: number;
-  } = {},
-) => {
-  let checks = 0;
-
-  while (checks < maxChecks) {
-    if (fn()) {
-      return true;
-    }
-    checks++;
-    await new Promise((resolve) => setTimeout(resolve, interval));
-  }
-
-  return false;
-};
-
-export const awaitFileExists = async (dir: string) => {
-  const result = await waitFor(() => fse.existsSync(dir), {
-    interval: 50,
-    maxChecks: 400,
+/**
+ * A faster `expect.poll`
+ */
+export const expectPoll = (fn: () => boolean): any => {
+  return expect.poll(fn, {
+    intervals: [20, 30, 40, 50, 60, 70, 80, 90, 100],
   });
-  if (!result) {
-    throw new Error(`awaitFileExists failed: ${dir}`);
-  }
 };
 
-export const awaitFileChanges = async (file: string) => {
+export const expectFile = (dir: string) =>
+  expectPoll(() => fs.existsSync(dir)).toBeTruthy();
+
+export const expectFileChanges = async (file: string) => {
   const oldContent = await fse.readFile(file, 'utf-8');
   return async () => {
-    const result = await waitFor(
-      () => {
-        try {
-          return fse.readFileSync(file, 'utf-8') !== oldContent;
-        } catch (e) {
-          return false;
-        }
-      },
-      { interval: 50 },
-    );
-
-    if (!result) {
-      throw new Error(`awaitFileChanges failed: ${file}`);
-    }
-
-    return result;
+    await expectPoll(
+      () => fse.readFileSync(file, 'utf-8') !== oldContent,
+    ).toBeTruthy();
   };
 };