node.js - 单元测试: How to mock mongoose code when using populate?

Question

我有一个带有静态函数的 Mongoose 模型，该函数通过 ID 查找员工文档并填充引用的 manager 和 interviewer 字段。

employeeSchema.statics.findAndPopulateById = function(id) {
  return this.findById(id)
    .populate("manager", "firstname lastname email")
    .populate("interviewer", "email")
    .then(employee => {
      if (!employee) {
        throw new errors.NotFoundError();
      }
      return employee;
    });
}

我知道当这个函数不包含填充链时如何测试它，但是填充部分让我陷入了困境。

具体来说，我正在尝试测试在找不到匹配记录时是否会引发 NotFoundError 异常，但我很困惑如何模拟 findById 方法以便 .populate() 可以在返回值上调用。

如果我在没有 .populate() 链的情况下测试此方法，我会执行类似的操作

let findByIdMock = sandbox.stub(Employee, 'findById');
findByIdMock.resolves(null);

return expect(Employee.findAndPopulateById('123')).to.be.rejectedWith(errors.NotFoundError);

但是，当涉及到 populate 时，这种方法当然不起作用。我认为我应该返回一个模拟查询或类似的东西，但我还需要能够在该模拟上再次调用 populate 或将其解析为 promise 。

如何为这段代码编写测试？我的功能应该采用不同的结构吗？

Answer 1

好吧，这比我预期的要容易，我只需要添加一个对 .exec() 的调用我的最后.populate()之间和.then()使下面的测试工作。

it("should throw not found exception if employee doesn't exist", () => {
  const mockQuery = {
    exec: sinon.stub().resolves(null)
  }
  mockQuery.populate = sinon.stub().returns(mockQuery);

  let findById = sandbox.stub(Employee, 'findById');
  findById.withArgs(mockId).returns(mockQuery);

  return expect(Employee.findAndPopulateById(mockId)).to.be.rejectedWith(errors.NotFoundError);
});