如何在Jest测试中安全访问.yaml文件中的环境变量？

你遇到的问题很常见——测试时不想把真实的RDS敏感信息硬编码到测试用例里，下面几种方案可以解决：

1. 封装环境变量获取逻辑，Mock这个封装函数 #

最推荐的做法是把获取环境变量的代码抽成单独模块，避免直接在业务代码里访问process.env，这样测试时只需要Mock这个模块就行：

首先新建config.js：

// config.js
module.exports = {
  getDbCredentials: () => ({
    dbName: process.env.DB_NAME,
    dbPassword: process.env.DB_PASSWORD,
    // 其他RDS连接参数
  })
};

然后在业务代码index.js里调用这个模块：

// index.js
const { getDbCredentials } = require('./config');

function initDbConnection() {
  const { dbName, dbPassword } = getDbCredentials();
  // 数据库连接逻辑
}

module.exports = { initDbConnection };

测试时（以Jest为例）直接Mockconfig.js，用模拟值代替真实变量：

// index.test.js
const { initDbConnection } = require('./index');
const config = require('./config');

// Mock整个config模块的方法
jest.mock('./config', () => ({
  getDbCredentials: jest.fn().mockReturnValue({
    dbName: 'mock_test_db',
    dbPassword: 'mock_test_pwd'
  })
}));

test('数据库连接初始化逻辑正常', () => {
  initDbConnection();
  // 可以断言方法被调用，或者验证逻辑行为
  expect(config.getDbCredentials).toHaveBeenCalled();
});

这种方式完全隔离了真实环境变量，模拟值随便设，也不会污染全局的process.env。

2. 用测试框架直接Mock process.env（Jest为例） #

如果不想额外封装模块，Jest支持直接修改process.env，并且能在测试前后自动恢复，避免测试间的变量污染：

// index.test.js
// 每个测试前重置环境变量
beforeEach(() => {
  jest.resetModules();
  // 扩展原有环境变量，只覆盖需要的RDS参数
  process.env = {
    ...process.env,
    DB_NAME: 'mock_test_db',
    DB_PASSWORD: 'mock_test_pwd'
  };
});

// 测试后恢复原始环境变量
afterEach(() => {
  process.env = jest.requireActual('./config').originalEnv;
});

test('环境变量能被正确获取', () => {
  const { initDbConnection } = require('./index');
  initDbConnection();
  // 执行断言
});

另外也可以用dotenv加载测试专用的环境变量文件，把.env.test加入.gitignore（避免提交敏感信息），然后在Jest的配置文件里全局加载：

jest.config.js：

module.exports = {
  setupFilesAfterEnv: ['./jest.setup.js']
};

jest.setup.js：

require('dotenv').config({ path: '.env.test' });

.env.test里放模拟值：

DB_NAME=mock_test_db
DB_PASSWORD=mock_test_pwd

3. 用Sinon.js Mock全局process.env（Mocha/Sinon组合） #

如果用Mocha+Sinon的测试栈，可以用Sinon的stub来临时替换process.env的属性，测试后再恢复：

// index.test.js
const sinon = require('sinon');
const { initDbConnection } = require('./index');

let dbNameStub, dbPasswordStub;

beforeEach(() => {
  // 替换环境变量值
  dbNameStub = sinon.stub(process.env, 'DB_NAME').value('mock_test_db');
  dbPasswordStub = sinon.stub(process.env, 'DB_PASSWORD').value('mock_test_pwd');
});

afterEach(() => {
  // 恢复原始值
  dbNameStub.restore();
  dbPasswordStub.restore();
});

test('数据库连接逻辑正常执行', () => {
  initDbConnection();
  // 执行断言
});

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