test(e2e): replace manual file lookup with helpers (#6248)

chenjiahan · Copilot · web-flow · commit d0238e763b41 · 2025-09-25T13:09:47.000+08:00
Co-authored-by: Copilot &lt;175728472+Copilot@users.noreply.github.com&gt;
diff --git a/e2e/cases/assets/asset-prefix/index.test.ts b/e2e/cases/assets/asset-prefix/index.test.ts
@@ -1,4 +1,4 @@
-import { expect, test } from '@e2e/helper';
+import { expect, getFileContent, test } from '@e2e/helper';
 
 test('should allow dev.assetPrefix to be `auto`', async ({ page, dev }) => {
   await dev({
@@ -98,8 +98,7 @@ test('should use output.assetPrefix in none mode', async ({ build }) => {
   });
 
   const files = result.getDistFiles();
-  const indexHtml =
-    files[Object.keys(files).find((file) => file.endsWith('index.html'))!];
+  const indexHtml = getFileContent(files, 'index.html');
   expect(indexHtml).toContain('http://prod.com');
   expect(indexHtml).not.toContain('http://dev.com');
 });
diff --git a/e2e/cases/browserslist/extends-browserslist/index.test.ts b/e2e/cases/browserslist/extends-browserslist/index.test.ts
@@ -1,15 +1,12 @@
-import { expect, test } from '@e2e/helper';
+import { expect, getFileContent, test } from '@e2e/helper';
 
 test('should extend browserslist and downgrade syntax', async ({ build }) => {
   const originalCwd = process.cwd();
   process.chdir(__dirname);
 
   const rsbuild = await build();
-
   const files = rsbuild.getDistFiles();
-
-  const indexFile =
-    files[Object.keys(files).find((file) => file.endsWith('.js'))!];
+  const indexFile = getFileContent(files, 'index.js');
 
   expect(indexFile.includes('async ')).toBeFalsy();
 
diff --git a/e2e/cases/browserslist/package-config-array/index.test.ts b/e2e/cases/browserslist/package-config-array/index.test.ts
@@ -1,12 +1,8 @@
-import { expect, test } from '@e2e/helper';
+import { expect, getFileContent, test } from '@e2e/helper';
 
 test('should read browserslist array from package.json', async ({ build }) => {
   const rsbuild = await build();
-
   const files = rsbuild.getDistFiles();
-
-  const indexFile =
-    files[Object.keys(files).find((file) => file.endsWith('.js'))!];
-
+  const indexFile = getFileContent(files, 'index.js');
   expect(indexFile.includes('async ')).toBeFalsy();
 });
diff --git a/e2e/cases/browserslist/package-config-object/index.test.ts b/e2e/cases/browserslist/package-config-object/index.test.ts
@@ -1,12 +1,8 @@
-import { expect, test } from '@e2e/helper';
+import { expect, getFileContent, test } from '@e2e/helper';
 
 test('should read browserslist string from package.json', async ({ build }) => {
   const rsbuild = await build();
-
   const files = rsbuild.getDistFiles();
-
-  const indexFile =
-    files[Object.keys(files).find((file) => file.endsWith('.js'))!];
-
+  const indexFile = getFileContent(files, 'index.js');
   expect(indexFile.includes('async ')).toBeFalsy();
 });
diff --git a/e2e/cases/browserslist/package-config-string/index.test.ts b/e2e/cases/browserslist/package-config-string/index.test.ts
@@ -1,12 +1,8 @@
-import { expect, test } from '@e2e/helper';
+import { expect, getFileContent, test } from '@e2e/helper';
 
 test('should read browserslist string from package.json', async ({ build }) => {
   const rsbuild = await build();
-
   const files = rsbuild.getDistFiles();
-
-  const indexFile =
-    files[Object.keys(files).find((file) => file.endsWith('.js'))!];
-
+  const indexFile = getFileContent(files, 'index.js');
   expect(indexFile.includes('async ')).toBeFalsy();
 });
diff --git a/e2e/cases/css/configure-css-extract/index.test.ts b/e2e/cases/css/configure-css-extract/index.test.ts
@@ -1,12 +1,10 @@
-import { expect, test } from '@e2e/helper';
+import { expect, getFileContent, test } from '@e2e/helper';
 
 test('should allow to configure options of CSSExtractPlugin', async ({
   build,
 }) => {
   const rsbuild = await build();
   const files = rsbuild.getDistFiles();
-  const content =
-    files[Object.keys(files).find((file) => file.endsWith('my-css.css'))!];
-
+  const content = getFileContent(files, 'my-css.css');
   expect(content).toEqual('body{color:red}');
 });
diff --git a/e2e/cases/css/css-layers/index.test.ts b/e2e/cases/css/css-layers/index.test.ts
@@ -1,14 +1,11 @@
-import { expect, rspackTest } from '@e2e/helper';
+import { expect, getFileContent, rspackTest } from '@e2e/helper';
 
 rspackTest(
   'should bundle CSS layers as expected in build',
   async ({ build }) => {
     const rsbuild = await build();
     const files = rsbuild.getDistFiles();
-
-    const content =
-      files[Object.keys(files).find((file) => file.endsWith('.css'))!];
-
+    const content = getFileContent(files, 'index.css');
     expect(content).toEqual(
       '@layer a{.a{color:red}}@layer b{.b{color:green}}@layer c{@layer{.c-sub{color:#00f}.c-sub2{color:green}}.c{color:#00f}}',
     );
@@ -18,10 +15,7 @@ rspackTest(
 rspackTest('should bundle CSS layers as expected in dev', async ({ dev }) => {
   const rsbuild = await dev();
   const files = rsbuild.getDistFiles();
-
-  const content =
-    files[Object.keys(files).find((file) => file.endsWith('.css'))!];
-
+  const content = getFileContent(files, 'index.css');
   expect(content.trim()).toEqual(`@layer a {
 .a {
   color: red;
diff --git a/e2e/cases/css/css-modules-composes/index.test.ts b/e2e/cases/css/css-modules-composes/index.test.ts
@@ -1,15 +1,12 @@
 import path from 'node:path';
-import { expect, rspackTest } from '@e2e/helper';
+import { expect, getFileContent, rspackTest } from '@e2e/helper';
 
 rspackTest(
   'should compile CSS Modules composes correctly',
   async ({ build }) => {
     const rsbuild = await build();
     const files = rsbuild.getDistFiles();
-
-    const content =
-      files[Object.keys(files).find((file) => file.endsWith('.css'))!];
-
+    const content = getFileContent(files, 'index.css');
     expect(content).toMatch(
       /.*\{color:#ff0;background:red\}.*\{background:#00f\}/,
     );
@@ -27,10 +24,7 @@ rspackTest(
       },
     });
     const files = rsbuild.getDistFiles();
-
-    const content =
-      files[Object.keys(files).find((file) => file.endsWith('.css'))!];
-
+    const content = getFileContent(files, 'external.css');
     expect(content).toMatch(
       /.*\{color:#000;background:#0ff\}.*\{background:green\}/,
     );
diff --git a/e2e/cases/css/css-modules-export-locals/index.test.ts b/e2e/cases/css/css-modules-export-locals/index.test.ts
@@ -1,4 +1,9 @@
-import { type BuildResult, expect, rspackTest } from '@e2e/helper';
+import {
+  type BuildResult,
+  expect,
+  getFileContent,
+  rspackTest,
+} from '@e2e/helper';
 
 declare global {
   interface Window {
@@ -8,8 +13,7 @@ declare global {
 
 const expectCSSContext = async (rsbuild: BuildResult) => {
   const files = rsbuild.getDistFiles();
-  const content =
-    files[Object.keys(files).find((file) => file.endsWith('.css'))!];
+  const content = getFileContent(files, 'index.css');
   expect(content).toMatch(
     /\.the-dash-class-\w{6}{color:#00f}\.theCamelClass-\w{6}{color:red}\.the_underscore_class-\w{6}{color:green}/,
   );
diff --git a/e2e/cases/css/css-modules-exports-global/index.test.ts b/e2e/cases/css/css-modules-exports-global/index.test.ts
@@ -1,4 +1,4 @@
-import { expect, rspackTest } from '@e2e/helper';
+import { expect, getFileContent, rspackTest } from '@e2e/helper';
 
 rspackTest(
   'should exports global in CSS Modules correctly in dev build',
@@ -25,8 +25,7 @@ rspackTest(
     await expect(test2Locator).toHaveCSS('color', 'rgb(0, 0, 255)');
 
     const files = rsbuild.getDistFiles();
-    const content =
-      files[Object.keys(files).find((file) => file.endsWith('.css'))!];
+    const content = getFileContent(files, 'index.css');
     expect(content).toMatch(/\.foo-\w{6}{color:red}\.bar{color:#00f}/);
   },
 );
diff --git a/e2e/cases/css/css-modules-global/index.test.ts b/e2e/cases/css/css-modules-global/index.test.ts
@@ -1,14 +1,11 @@
-import { expect, test } from '@e2e/helper';
+import { expect, getFileContent, test } from '@e2e/helper';
 
 test('should compile CSS Modules with :global() correctly', async ({
   build,
 }) => {
   const rsbuild = await build();
   const files = rsbuild.getDistFiles();
-
-  const content =
-    files[Object.keys(files).find((file) => file.endsWith('.css'))!];
-
+  const content = getFileContent(files, 'index.css');
   expect(content).toMatch(
     /.*\{position:relative\}.* \.bar,.* \.baz\{height:100%;overflow:hidden\}.* \.lol{width:80%}/,
   );
diff --git a/e2e/cases/css/css-modules-mode/index.test.ts b/e2e/cases/css/css-modules-mode/index.test.ts
@@ -1,12 +1,9 @@
-import { expect, test } from '@e2e/helper';
+import { expect, getFileContent, test } from '@e2e/helper';
 
 test('should compile CSS Modules global mode correctly', async ({ build }) => {
   const rsbuild = await build();
   const files = rsbuild.getDistFiles();
-
-  const content =
-    files[Object.keys(files).find((file) => file.endsWith('.css'))!];
-
+  const content = getFileContent(files, 'index.css');
   expect(content).toEqual(
     '.foo{position:relative}.foo .bar,.foo .baz{height:100%;overflow:hidden}.foo .lol{width:80%}',
   );
diff --git a/e2e/cases/css/css-modules-named-export/index.test.ts b/e2e/cases/css/css-modules-named-export/index.test.ts
@@ -1,21 +1,18 @@
-import { expect, rspackTest } from '@e2e/helper';
+import { expect, getFileContent, rspackTest } from '@e2e/helper';
 
 rspackTest(
   'should compile CSS Modules with named exports correctly',
   async ({ page, buildPreview }) => {
     const rsbuild = await buildPreview();
     const files = rsbuild.getDistFiles();
-
-    const content =
-      files[Object.keys(files).find((file) => file.endsWith('.css'))!];
+    const content = getFileContent(files, 'index.css');
 
     expect(content).toMatch(
       /\.classA-\w{6}{color:red}\.classB-\w{6}{color:#00f}\.classC-\w{6}{color:#ff0}/,
     );
 
     const root = page.locator('#root');
     const text = await root.innerHTML();
-
     expect(text).toMatch(/classA-\w{6} classB-\w{6} classC-\w{6}/);
   },
 );
diff --git a/e2e/cases/css/css-modules/index.test.ts b/e2e/cases/css/css-modules/index.test.ts
@@ -1,14 +1,11 @@
-import { expect, rspackTest } from '@e2e/helper';
+import { expect, getFileContent, rspackTest } from '@e2e/helper';
 
 rspackTest(
   'should compile CSS Modules with default configuration',
   async ({ build }) => {
     const rsbuild = await build();
     const files = rsbuild.getDistFiles();
-
-    const content =
-      files[Object.keys(files).find((file) => file.endsWith('.css'))!];
-
+    const content = getFileContent(files, 'index.css');
     expect(content).toMatch(
       /\.the-a-class{color:red}\.the-b-class-\w{6}{color:#00f}\.the-c-class-\w{6}{color:#ff0}\.the-d-class{color:green}/,
     );
@@ -30,10 +27,7 @@ rspackTest(
       },
     });
     const files = rsbuild.getDistFiles();
-
-    const content =
-      files[Object.keys(files).find((file) => file.endsWith('.css'))!];
-
+    const content = getFileContent(files, 'index.css');
     expect(content).toMatch(
       /.the-a-class{color:red}.the-b-class-\w{6}{color:#00f}.the-c-class{color:#ff0}.the-d-class{color:green}/,
     );
@@ -53,10 +47,7 @@ rspackTest(
       },
     });
     const files = rsbuild.getDistFiles();
-
-    const content =
-      files[Object.keys(files).find((file) => file.endsWith('.css'))!];
-
+    const content = getFileContent(files, 'index.css');
     expect(content).toMatch(
       /\.the-a-class{color:red}\.\w{8}{color:#00f}\.\w{8}{color:#ff0}\.the-d-class{color:green}/,
     );
@@ -76,10 +67,7 @@ rspackTest(
       },
     });
     const files = rsbuild.getDistFiles();
-
-    const content =
-      files[Object.keys(files).find((file) => file.endsWith('.css'))!];
-
+    const content = getFileContent(files, 'index.css');
     expect(content).toMatch(
       /\.the-a-class{color:red}\.\w{4}{color:#00f}\.\w{4}{color:#ff0}\.the-d-class{color:green}/,
     );
diff --git a/e2e/cases/css/emit-css/index.test.ts b/e2e/cases/css/emit-css/index.test.ts
@@ -1,4 +1,4 @@
-import { expect, test } from '@e2e/helper';
+import { expect, getFileContent, test } from '@e2e/helper';
 
 test('should not emit CSS files when build node target', async ({ build }) => {
   const rsbuild = await build({
@@ -11,8 +11,7 @@ test('should not emit CSS files when build node target', async ({ build }) => {
   const files = rsbuild.getDistFiles();
 
   // preserve CSS Modules mapping
-  const jsContent =
-    files[Object.keys(files).find((file) => file.endsWith('.js'))!];
+  const jsContent = getFileContent(files, 'index.js');
   expect(jsContent.includes('"title-class":')).toBeTruthy();
 
   const cssFiles = Object.keys(files).filter((file) => file.endsWith('.css'));
@@ -33,8 +32,7 @@ test('should allow to emit CSS with output.emitCss when build node target', asyn
   const files = rsbuild.getDistFiles();
 
   // preserve CSS Modules mapping
-  const jsContent =
-    files[Object.keys(files).find((file) => file.endsWith('.js'))!];
+  const jsContent = getFileContent(files, 'index.js');
   expect(jsContent.includes('"title-class":')).toBeTruthy();
 
   const cssFiles = Object.keys(files).filter((file) => file.endsWith('.css'));
@@ -54,8 +52,7 @@ test('should not emit CSS files when build web-worker target', async ({
   const files = rsbuild.getDistFiles();
 
   // preserve CSS Modules mapping
-  const jsContent =
-    files[Object.keys(files).find((file) => file.endsWith('.js'))!];
+  const jsContent = getFileContent(files, 'index.js');
   expect(jsContent.includes('"title-class":')).toBeTruthy();
 
   const cssFiles = Object.keys(files).filter((file) => file.endsWith('.css'));
@@ -76,8 +73,7 @@ test('should allow to emit CSS with output.emitCss when build web-worker target'
   const files = rsbuild.getDistFiles();
 
   // preserve CSS Modules mapping
-  const jsContent =
-    files[Object.keys(files).find((file) => file.endsWith('.js'))!];
+  const jsContent = getFileContent(files, 'index.js');
   expect(jsContent.includes('"title-class":')).toBeTruthy();
 
   const cssFiles = Object.keys(files).filter((file) => file.endsWith('.css'));
@@ -98,8 +94,7 @@ test('should allow to disable CSS emit with output.emitCss when build web target
   const files = rsbuild.getDistFiles();
 
   // preserve CSS Modules mapping
-  const jsContent =
-    files[Object.keys(files).find((file) => file.endsWith('.js'))!];
+  const jsContent = getFileContent(files, 'index.js');
   expect(jsContent.includes('"title-class":')).toBeTruthy();
 
   const cssFiles = Object.keys(files).filter((file) => file.endsWith('.css'));
diff --git a/e2e/cases/css/enable-experiments-css/index.test.ts b/e2e/cases/css/enable-experiments-css/index.test.ts
@@ -1,4 +1,4 @@
-import { expect, rspackTest } from '@e2e/helper';
+import { expect, getFileContent, rspackTest } from '@e2e/helper';
 
 const COMPILE_WARNING = 'Compile Warning';
 
@@ -7,8 +7,7 @@ rspackTest(
   async ({ build }) => {
     const rsbuild = await build();
     const files = rsbuild.getDistFiles();
-    const content =
-      files[Object.keys(files).find((file) => file.endsWith('index.css'))!];
+    const content = getFileContent(files, 'index.css');
 
     expect(content).toEqual('body{color:red}');
     // should have no warnings
@@ -28,8 +27,7 @@ rspackTest(
     });
 
     const files = rsbuild.getDistFiles();
-    const content =
-      files[Object.keys(files).find((file) => file.endsWith('index.js'))!];
+    const content = getFileContent(files, 'index.js');
     expect(content).toContain('color:red');
 
     // should have no warnings
diff --git a/e2e/cases/css/import-common-css/index.test.ts b/e2e/cases/css/import-common-css/index.test.ts
@@ -1,12 +1,8 @@
-import { expect, rspackTest } from '@e2e/helper';
+import { expect, getFileContent, rspackTest } from '@e2e/helper';
 
 rspackTest('should compile common CSS import correctly', async ({ build }) => {
   const rsbuild = await build();
-
   const files = rsbuild.getDistFiles();
-  const cssFiles = Object.keys(files).find((file) => file.endsWith('.css'))!;
-
-  expect(files[cssFiles]).toEqual(
-    'html{min-height:100%}#a{color:red}#b{color:#00f}',
-  );
+  const indexCss = getFileContent(files, 'index.css');
+  expect(indexCss).toEqual('html{min-height:100%}#a{color:red}#b{color:#00f}');
 });
diff --git a/e2e/cases/css/import-loaders/index.test.ts b/e2e/cases/css/import-loaders/index.test.ts
diff --git a/e2e/cases/css/multi-css/index.test.ts b/e2e/cases/css/multi-css/index.test.ts
diff --git a/e2e/cases/performance/remove-console/index.test.ts b/e2e/cases/performance/remove-console/index.test.ts
diff --git a/e2e/cases/plugin-less/import/index.test.ts b/e2e/cases/plugin-less/import/index.test.ts
diff --git a/e2e/cases/plugin-vue/sfc-css-modules-ssr/index.test.ts b/e2e/cases/plugin-vue/sfc-css-modules-ssr/index.test.ts
diff --git a/e2e/cases/polyfill/core-js-basic/index.test.ts b/e2e/cases/polyfill/core-js-basic/index.test.ts
diff --git a/e2e/cases/polyfill/helper.ts b/e2e/cases/polyfill/helper.ts
diff --git a/e2e/helper/utils.ts b/e2e/helper/utils.ts
