Node.js单元测试中如何Mock变量？以awsMock.service为例

既然awsMock是你传入业务方法的参数，最直接的方式就是在测试时构造包含模拟service属性的awsMock实例，完全不需要复杂的Stub操作。以下是主流测试框架下的具体实现：

用Jest实现 #

基础Mock service 属性 #

先构造符合结构的模拟对象，直接传入业务方法即可：

// 你的业务方法示例
function handleLambda(awsMock) {
  let abc = awsMock.service.lambda;
  // 后续业务逻辑
  return abc;
}

// 测试用例
test('mock awsMock.service.lambda 成功', () => {
  // 构造模拟的service对象
  const mockService = {
    lambda: '模拟的Lambda配置' // 可以是字符串、对象、甚至模拟函数
  };
  // 组装成完整的awsMock模拟对象
  const mockAws = {
    service: mockService
  };

  const result = handleLambda(mockAws);
  // 断言结果符合预期
  expect(result).toBe('模拟的Lambda配置');
});

后续扩展Mock方法 #

如果需要给awsMock添加模拟方法，直接在对象上定义Jest模拟函数即可：

const mockAws = {
  service: mockService,
  invokeLambda: jest.fn().mockReturnValue({ StatusCode: 200 })
};

// 调用业务方法后，可验证方法调用
expect(mockAws.invokeLambda).toHaveBeenCalledTimes(1);

用Sinon实现 #

基础Mock + 方法Stub #

先构造基础结构，再用Sinon Stub需要的方法：

const sinon = require('sinon');

// 业务方法示例
function handleLambda(awsMock) {
  let abc = awsMock.service.lambda;
  const invokeResult = awsMock.invokeLambda();
  return { abc, invokeResult };
}

// 测试用例
it('mock service属性与方法成功', () => {
  const mockService = { lambda: '模拟的Lambda实例' };
  // 先给awsMock定义空方法，方便后续Stub
  const mockAws = {
    service: mockService,
    invokeLambda: () => {}
  };

  // Stub invokeLambda方法，指定返回值
  const invokeStub = sinon.stub(mockAws, 'invokeLambda').returns('执行成功');

  const result = handleLambda(mockAws);
  
  // 断言验证
  expect(result.abc).toBe('模拟的Lambda实例');
  expect(result.invokeResult).toBe('执行成功');
  sinon.assert.calledOnce(invokeStub);
});

之前Stub失败的常见原因 #

• 你可能试图Stub一个未提前定义属性的对象：比如awsMock一开始没有service属性，直接Stub awsMock.service会报错。解决办法是先给awsMock初始化service属性（哪怕是空对象），再Stub内部的lambda，或者直接构造完整结构的模拟对象。
• 如果awsMock是模块导出的单例而非传入参数，才需要用框架的模块Mock能力，但你明确是传入方法的对象，所以重点还是构造传入的模拟实例。

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