如何从模拟模块模拟多个值？

Question

我正在尝试模拟模块“jwt-decode”，我可以使用以下方法成功模拟一个返回值：

 jest.mock('jwt-decode', () => () => ({ data: { userRole: 2, checklist: mockUser.account.checklist }}))

这适用于需要解码的 jwt 以产生 2 的 userRole 的测试。但是，我的下一个测试需要 userRole 为 1，这就是问题出现的地方。

是否有适当的方法将 userRole 作为第一个测试实例返回 2 并为下一个测试实例返回 1？

Answer 1

您可以通过在模拟函数上调用 mockImplementation 或 mockImplementationOnce 并使用新实现作为参数来更改模拟函数的模拟实现，详细信息如下：https://jestjs.io/docs/en/mock-functions#mock-implementations

在你的情况下是这样的：

import jwt_decode from 'jwt-decode';

jest.mock('jwt-decode', () => jest.fn(() => ({
    data: { userRole: 2, checklist: mockUser.account.checklist },
})));

it('should test behavior with userRole equal to 2', () => {
    // your test here 
});

it('here we update mock implementation and test behavior with userRole equal to 1', () => {
    jwt_decode.mockImplementationOnce(() => ({
        data: { userRole: 1, checklist: mockUser.account.checklist },
    }));
    // your test here
});

it('since we used mockImplementationOnce method in the test above, here we again will be using initial mock implementation - userRole equal to 2', () => {
    // your test here
});