您如何在Jest中模拟Mongoose函数?

时间:2018-10-06 19:15:54

标签: mongoose jestjs

您如何模拟诸如find()distinct()count()populate()等猫鼬调用?

我试用了Mockingoose npm软件包,但它似乎仅限于像find()这样的基本调用,尽管它说可以与distinct()一起使用,但它要求您返回一个文档并且不允许您返回一个数组。

1 个答案:

答案 0 :(得分:1)

您可以使用jest.spyOn(),然后使用mockImplementation()模拟第一个调用,例如find()update()。这是findOneAndUpdate()的示例,我们正在检查以确保传递了正确的对象:

// TESTING:
// await Timeline.findOneAndUpdate(query, obj);
//

const Timeline = require("./models/user.timeline");
...
const TimelineFindOneAndUpdateMock = jest.spyOn(Timeline, "findOneAndUpdate");
const TimelineFindOneAndUpdate = jest.fn((query, obj) => {
    expect(obj.sendDateHasPassed).toBeFalsy();
    expect(moment(obj.sendDate).format()).toBe(moment("2018-11-05T23:00:00.000Z").format());
});
TimelineFindOneAndUpdateMock.mockImplementation(TimelineFindOneAndUpdate);

如果要模拟链接的函数,可以让它返回一个对象,该对象带有要调用的下一个链接的函数。这是一个如何模拟链接的distinct()调用的示例。

// TESTING:
// let accountIDs = await Account.find(query).distinct("_id");
//
// WILL RETURN:
// ["124512341234","124512341234","124512341234"]
//

const Account = require("./models/user.account");
...
const AccountFindMock = jest.spyOn(Account, "find");
const AccountFindDistinctResult = ["124512341234","124512341234","124512341234"];
const AccountFindDistinctResult = jest.fn(() => AccountFindDistinctResult);
const AccountFindResult = {
    distinct: AccountFindDistinct
};
const AccountFind = jest.fn(() => AccountFindResult);
AccountFindMock.mockImplementation(AccountFind);

在测试运行之后,如果要检查调用函数的次数,例如调用distinct()的次数,可以添加以下内容:

expect(AccountFindDistinct).toHaveBeenCalledTimes(0);