Node.js中如何Stub/Mock对象所有方法？含Mongoose模型测试场景

当然可以！完全不用挨个Mock Mongoose模型的方法，用Sinon或者专门的工具库就能批量搞定，帮你省超多重复代码。下面给你几个实用的方案：

方案1：使用sinon-mongoose库（最省心） #

这个库是专门为Mongoose和Sinon适配的，能轻松Stub模型的查询方法，甚至完美支持链式调用（比如find().where().sort()这种场景）。

步骤：

• 先安装依赖：npm install sinon-mongoose --save-dev
• 示例代码：

const sinon = require('sinon');
const sinonMongoose = require('sinon-mongoose');
const User = require('../models/User'); // 你的Mongoose模型

// 先Stub模型的find方法，返回预设的模拟数据
const userStub = sinonMongoose.stub(User)
  .withArgs('find')
  .resolves([{ name: 'Test User', email: 'test@example.com' }]);

// 链式Stub其他方法，比如findOne、where
userStub.withArgs('findOne').resolves({ name: 'Single User' });
userStub.withArgs('where').returnsThis(); // 链式调用需要返回this维持调用链

不管是直接调用User.find()，还是复杂的链式查询User.where('name').equals('Test')，都能被正确拦截并返回你想要的结果。

方案2：用Sinon直接批量Stub模型原型 #

Mongoose模型的核心方法大多挂载在原型上（比如Model.prototype.find），所以可以遍历原型方法批量Stub：

const sinon = require('sinon');
const User = require('../models/User');

// 定义你想返回的默认模拟数据
const mockResult = { 
  name: 'Mocked User', 
  toJSON: () => ({ name: 'Mocked User' }) 
};

// 遍历模型原型的所有方法，批量Stub
Object.getOwnPropertyNames(User.prototype).forEach(methodName => {
  // 跳过构造函数和非函数属性
  if (methodName !== 'constructor' && typeof User.prototype[methodName] === 'function') {
    // 区分异步/同步方法：查询类方法返回Promise，同步方法直接返回模拟数据
    if (['find', 'findOne', 'findById', 'where'].includes(methodName)) {
      sinon.stub(User.prototype, methodName).resolves(mockResult);
    } else {
      sinon.stub(User.prototype, methodName).returns(mockResult);
    }
  }
});

注意：这种方式要注意区分异步方法（比如find返回Promise）和同步方法（比如toJSON直接返回对象），避免测试报错。

方案3：创建Mock模型类替代原模型 #

如果不想修改原模型的原型，可以直接写一个Mock类，覆盖所有需要的方法：

const mockResult = { 
  name: 'Mocked User', 
  toJSON: () => ({ name: 'Mocked User' }) 
};

class MockUser {
  static find() { return Promise.resolve(mockResult); }
  static findOne() { return Promise.resolve(mockResult); }
  static where() { return this; } // 支持链式调用
  static sort() { return Promise.resolve(mockResult); }
  // 实例方法也直接返回模拟数据
  toJSON() { return mockResult; }
}

// 在测试中用MockUser替代原User模型（以Jest为例）
jest.mock('../models/User', () => MockUser);

这种方式更灵活，你可以完全控制每个方法的行为，而且不会污染原模型的原型。

额外注意事项 #

• 链式调用处理：像where、sort这类方法需要返回this才能维持链式调用，所以Stub时要用returnsThis()（Sinon方式）或者返回类本身（Mock类方式）。
• 清理Stub：测试结束后记得用sinon.restore()恢复原方法，避免影响其他测试用例。
• 按需覆盖：如果某些方法需要特殊返回值，可以在批量Stub之后单独修改，比如：

// 批量Stub后，单独修改findOne的返回值
sinon.stub(User, 'findOne').resolves({ name: 'Special User' });

这样就能轻松搞定Mongoose模型方法的批量Stub，不用再挨个写重复的Mock代码啦！

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