如何使用Mocha为Node.js中未导出的异步自调用函数编写单元测试

遇到这种情况确实挺棘手的——匿名自执行函数因为没有对外导出，直接做单元测试会有不少限制。我给你整理了几个可行的方案，从长期维护友好的代码调整方案，到无需修改原代码的hack方案都涵盖了：

方案1：调整原代码结构（最推荐） #

这是最干净、最利于后续维护的方式，把自执行函数里的核心逻辑提取到一个命名函数中，仅在测试环境下导出它，完全不影响生产代码的运行：

// your-module.js
// 把核心业务逻辑提取到命名函数中
const coreAsyncLogic = async () => {
  // 你的异步代码，比如：
  const response = await fetch('/api/data');
  const data = await response.json();
  return data;
};

// 保留原有的自执行逻辑，确保生产环境正常运行
(async () => {
  await coreAsyncLogic();
})();

// 仅在测试环境导出函数，避免污染生产环境的导出
if (process.env.NODE_ENV === 'test') {
  module.exports = { coreAsyncLogic };
}

测试时直接导入这个命名函数即可，想怎么测就怎么测：

// your-module.test.js
const { coreAsyncLogic } = require('./your-module');

test('测试核心异步逻辑的返回值', async () => {
  // 先mock掉fetch这类依赖
  global.fetch = jest.fn().mockResolvedValue({
    json: () => Promise.resolve({ id: 1, name: 'test' })
  });
  
  const result = await coreAsyncLogic();
  expect(result).toEqual({ id: 1, name: 'test' });
  expect(fetch).toHaveBeenCalledWith('/api/data');
});

方案2：用rewire测试自执行函数的副作用（不修改原代码） #

如果你暂时没法修改原代码，rewire其实是能用的，但核心思路是测试自执行函数执行后的副作用（比如修改了模块内的私有变量、调用了外部依赖），而不是直接调用这个匿名函数。

举个例子，假设你的自执行函数会调用fs.writeFile：

// your-module.js
const fs = require('fs').promises;
(async () => {
  await fs.writeFile('output.txt', 'test content');
})();

测试时用rewire替换fs依赖，验证它是否被正确调用：

// your-module.test.js
const rewire = require('rewire');
const mockFs = {
  promises: { writeFile: jest.fn().mockResolvedValue() }
};

test('自执行函数正确调用了文件写入方法', async () => {
  // 用rewire加载模块，并替换内部的fs依赖
  const yourModule = rewire('./your-module');
  yourModule.__set__('fs', mockFs);
  
  // 异步自执行函数不会阻塞模块加载，所以要等它执行完成
  await new Promise(resolve => setImmediate(resolve));
  
  expect(mockFs.promises.writeFile).toHaveBeenCalledWith('output.txt', 'test content');
});

如果自执行函数修改了模块内的私有变量，也可以用rewire.__get__()获取变量来断言：

// your-module.js
let internalCounter = 0;
(async () => {
  internalCounter = await fetchCounterValue();
})();

测试代码：

const yourModule = rewire('./your-module');
await new Promise(resolve => setImmediate(resolve));
expect(yourModule.__get__('internalCounter')).toBe(/* 预期的数值 */);

方案3：用Jest的模块Mock替代rewire #

如果你用Jest作为测试框架，也可以不用rewire，直接利用Jest自带的模块mock能力：

// your-module.test.js
// 先mock掉fs模块
jest.mock('fs', () => ({
  promises: { writeFile: jest.fn().mockResolvedValue() }
}));

const fs = require('fs');
// 加载模块，触发自执行函数
require('./your-module');

test('验证自执行函数的文件写入行为', async () => {
  await new Promise(resolve => setImmediate(resolve));
  expect(fs.promises.writeFile).toHaveBeenCalledTimes(1);
});

关键注意点 #

• 异步自执行函数的执行时机：模块加载时就会启动，但异步操作不会阻塞模块加载，所以测试时一定要等待异步操作完成（用setImmediate、process.nextTick或者直接等待Promise）。
• 如果自执行函数里有未处理的Promise拒绝，测试时可能会抛出未捕获的错误，记得在测试代码里捕获或者提前处理这些错误。

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