Jest单元测试：如何按需Mock依赖函数funcB以测试funcA

问题1：测试文件中是否必须导入funcB才能进行Mock？或是需要Mock funcB所在的整个模块并单独指定其实现？ #

不需要在测试文件里单独导入funcB，直接Mock它所在的整个模块即可。以Jest为例，你只需要在测试文件顶部调用jest.mock('./FileB')，就能替换整个模块的导出内容，包括funcB。之后你可以通过require('./FileB').funcB获取到Mock后的函数，再指定它的返回值或实现逻辑，完全不需要单独把funcB导入到测试文件里。

问题2：4个测试用例中，3个需要Mock funcB，2个需要使用其真实实现，最佳实现方案是什么？ #

推荐用默认Mock整个模块，在需要真实实现的用例中临时恢复的方案，具体步骤如下：

1. 先在测试文件顶部默认Mock模块 #

// 测试文件 test/FileA.test.js
const { funcA } = require('../FileA');
const { funcB } = require('../FileB');

// 默认Mock整个FileB模块，funcB会被替换成jest.fn()
jest.mock('../FileB');

2. 编写需要Mock funcB的测试用例 #

直接给Mock后的funcB指定返回值或实现即可：

test('Mock funcB返回固定值', () => {
  // 指定Mock返回值
  funcB.mockReturnValue('模拟的测试数据');
  
  funcA();
  
  // 验证funcB被调用
  expect(funcB).toHaveBeenCalled();
});

test('Mock funcB抛出错误', () => {
  funcB.mockImplementation(() => {
    throw new Error('模拟错误');
  });
  
  expect(() => funcA()).toThrow('模拟错误');
});

// 第三个Mock用例同理...

3. 编写需要使用真实funcB的测试用例 #

用jest.requireActual()获取模块的真实导出，然后临时替换Mock函数的实现：

test('使用真实funcB的逻辑', () => {
  // 获取真实的funcB
  const realFuncB = jest.requireActual('../FileB').funcB;
  // 把Mock的funcB替换成真实实现
  funcB.mockImplementation(realFuncB);
  
  // 执行测试，验证真实逻辑的结果
  funcA();
  // 这里可以根据真实funcB的行为编写断言
  
  // 测试完成后重置Mock，避免影响其他用例
  funcB.mockReset();
});

test('另一个使用真实funcB的用例', () => {
  const realFuncB = jest.requireActual('../FileB').funcB;
  funcB.mockImplementation(realFuncB);
  
  // 测试逻辑...
  
  funcB.mockReset();
});

这种方案的好处是：

• 大部分用例（需要Mock的）不需要额外配置，直接用默认Mock即可
• 需要真实实现的用例可以精准控制，且测试后重置Mock不会污染其他用例
• 代码结构清晰，维护成本低

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