feat: stubEnv support import.meta.env (#594)

Author: 9aoy

e2e/spy/stubEnv.test.mts

@@ -0,0 +1,21 @@
+import { expect, it, rstest } from '@rstest/core';
+
+it('test stubEnv & unstubAllEnvs in import.meta.env', () => {
+  const env = import.meta.env.NODE_ENV;
+  const mockEnv = env === 'production' ? 'development' : 'production';
+  rstest.stubEnv('NODE_ENV', mockEnv);
+
+  rstest.stubEnv('TEST_111', '111');
+
+  expect(import.meta.env.NODE_ENV).toBe(mockEnv);
+
+  expect(import.meta.env.TEST_111).toBe('111');
+
+  rstest.stubEnv('NODE_ENV', undefined);
+  expect(import.meta.env.NODE_ENV).toBeUndefined();
+
+  rstest.unstubAllEnvs();
+
+  expect(import.meta.env.NODE_ENV).toBe(env);
+  expect(import.meta.env.TEST_111).toBeUndefined();
+});

packages/core/src/core/plugins/basic.ts

@@ -38,6 +38,7 @@ export const pluginBasic: (context: RstestContext) => RsbuildPlugin = (
             source: {
               define: {
                 'import.meta.rstest': "global['@rstest/core']",
+                'import.meta.env': 'process.env',
               },
             },
             output: {

packages/core/tests/core/__snapshots__/rsbuild.test.ts.snap

@@ -428,6 +428,7 @@ exports[`prepareRsbuild > should generate rspack config correctly (jsdom) 1`] =
     DefinePlugin {
       "_args": [
         {
+          "import.meta.env": "process.env",
           "import.meta.env.ASSET_PREFIX": """",
           "import.meta.env.BASE_URL": ""/"",
           "import.meta.env.DEV": false,
@@ -940,6 +941,7 @@ exports[`prepareRsbuild > should generate rspack config correctly (node) 1`] = `
     DefinePlugin {
       "_args": [
         {
+          "import.meta.env": "process.env",
           "import.meta.env.ASSET_PREFIX": """",
           "import.meta.env.BASE_URL": ""/"",
           "import.meta.env.DEV": false,
@@ -1454,6 +1456,7 @@ exports[`prepareRsbuild > should generate rspack config correctly in watch mode
     DefinePlugin {
       "_args": [
         {
+          "import.meta.env": "process.env",
           "import.meta.env.ASSET_PREFIX": """",
           "import.meta.env.BASE_URL": ""/"",
           "import.meta.env.DEV": false,
@@ -1986,6 +1989,7 @@ exports[`prepareRsbuild > should generate rspack config correctly with projects
     DefinePlugin {
       "_args": [
         {
+          "import.meta.env": "process.env",
           "import.meta.env.ASSET_PREFIX": """",
           "import.meta.env.BASE_URL": ""/"",
           "import.meta.env.DEV": false,
@@ -2509,6 +2513,7 @@ exports[`prepareRsbuild > should generate rspack config correctly with projects
     DefinePlugin {
       "_args": [
         {
+          "import.meta.env": "process.env",
           "import.meta.env.ASSET_PREFIX": """",
           "import.meta.env.BASE_URL": ""/"",
           "import.meta.env.DEV": false,

website/docs/en/api/rstest/utilities.mdx

@@ -8,9 +8,9 @@ A set of useful utility functions.
 
 - **Type:** `(name: string, value: string | undefined) => Rstest`
 
-Temporarily sets an environment variable in `process.env` to the specified value. Useful for testing code that depends on environment variables.
+Temporarily sets an environment variable in `process.env` and `import.meta.env` to the specified value. Useful for testing code that depends on environment variables.
 
-- If `value` is `undefined`, the variable will be removed from `process.env`.
+- If `value` is `undefined`, the variable will be removed from `process.env` and `import.meta.env`.
 - You can call this multiple times to stub multiple variables.
 - Use `rstest.unstubAllEnvs()` to restore all environment variables changed by this method.
 
@@ -19,9 +19,11 @@ Temporarily sets an environment variable in `process.env` to the specified value
 ```ts
 rstest.stubEnv('NODE_ENV', 'test');
 expect(process.env.NODE_ENV).toBe('test');
+expect(import.meta.env.NODE_ENV).toBe('test');
 
 rstest.stubEnv('MY_VAR', undefined);
 expect(process.env.MY_VAR).toBeUndefined();
+expect(import.meta.env.MY_VAR).toBeUndefined();
 ```
 
 ## rstest.unstubAllEnvs

website/docs/zh/api/rstest/utilities.mdx

@@ -8,9 +8,9 @@
 
 - **类型：** `(name: string, value: string | undefined) => Rstest`
 
-临时设置 `process.env` 中的环境变量为指定值。适用于测试依赖环境变量的代码。
+临时设置 `process.env` 和 `import.meta.env` 中的环境变量为指定值。适用于测试依赖环境变量的代码。
 
-- 如果 `value` 为 `undefined`，该变量会从 `process.env` 中移除。
+- 如果 `value` 为 `undefined`，该变量会从 `process.env` 和 `import.meta.env` 中移除。
 - 可多次调用以模拟多个环境变量。
 - 使用 `rstest.unstubAllEnvs()` 可恢复所有通过此方法更改的环境变量。
 
@@ -19,9 +19,11 @@
 ```ts
 rstest.stubEnv('NODE_ENV', 'test');
 expect(process.env.NODE_ENV).toBe('test');
+expect(import.meta.env.NODE_ENV).toBe('test');
 
 rstest.stubEnv('MY_VAR', undefined);
 expect(process.env.MY_VAR).toBeUndefined();
+expect(import.meta.env.MY_VAR).toBeUndefined();
 ```
 
 ## rstest.unstubAllEnvs