你如何在Jest中嘲弄猫鼬的功能？

Question

你怎么嘲笑猫鼬的叫声，比如find()，distinct()，count()，populate()等等？

我尝试了Mockingoose包，但它似乎仅限于基本的调用，比如find()，虽然它说它适用于distinct()，但它要求您返回一个文档，并且不允许您返回数组。

Answer 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);