如何使用Jest测试依赖顶层环境变量的导出函数？

你的问题根因是模块顶层代码会在导入阶段直接执行，测试代码里修改process.env的操作晚于模块导入时机，所以常量b的值在模块加载完成后就已经固定为"undefined"，不会再随着环境变量修改更新。以下是三种常见的解决方式：

方案1：调整测试代码的导入顺序，先修改环境变量再加载目标模块 #

ES模块的静态import会自动提升到代码最顶层执行，因此写在import后的环境变量修改操作永远晚于模块加载。可以改用CommonJS的require或者ES动态导入来手动控制模块加载时机：

// temp.test.js 修改后
process.env.TEMP = "temp"; // 先修改环境变量
// 再导入目标模块
const { temp } = require("./temp");

describe("temp", () => {
  test("test temp", () => {
    expect(temp()).toBe("temp"); // 可正常通过
  });
});

如果要使用ES模块的导入规范，可以用异步动态导入实现：

// temp.test.js 异步版本
describe("temp", () => {
  test("test temp", async () => {
    process.env.TEMP = "temp";
    const { temp } = await import("./temp");
    expect(temp()).toBe("temp");
  });
});

方案2：修改源码封装环境变量读取逻辑，避免顶层预加载 #

如果不想修改测试逻辑，可以调整源码中常量b的定义，把固定的顶层常量改成懒加载的取值函数，既可以满足多个函数复用的需求，也能每次读取到最新的环境变量：

// temp.js 修改后
const getB = () => `${process.env.TEMP}`;

export function temp(a) {
  return getB();
}

// 其余导出函数也可以调用getB()复用逻辑，不需要重复编写

该方案不会破坏原有业务逻辑，同时后续如果有动态修改环境变量的业务需求也能直接支持。

方案3：使用Jest的模块重置能力处理多测试用例场景 #

如果多个测试用例需要使用不同的环境变量值，可以配合jest.resetModules()清空之前加载的模块缓存，每次测试用例执行前重新加载模块即可：

// temp.test.js
describe("temp", () => {
  beforeEach(() => {
    jest.resetModules(); // 每次执行用例前清空模块缓存
  });

  test("test env value is temp", () => {
    process.env.TEMP = "temp";
    const { temp } = require("./temp");
    expect(temp()).toBe("temp");
  });

  test("test env value is other", () => {
    process.env.TEMP = "other";
    const { temp } = require("./temp");
    expect(temp()).toBe("other");
  });
});

内容的提问来源于stack exchange，提问作者Koko