跳转至主内容
版本:下一个

绕过模块模拟

Jest允许在测试用例中模拟整个模块。这对于你测试被测代码是否正常调用某个模块的函数是很有用的。 然而,你可能只想模拟部分模块,剩下的则使用原生的实现。

考虑为这个 createUser函数编写一个测试用例:

createUser.js
import fetch from 'node-fetch';

export const createUser = async () => {
const response = await fetch('https://website.com/users', {method: 'POST'});
const userId = await response.text();
return userId;
};

上述测试代码模拟 fetch 函数,这样就可以确保它在调用的时候不会发出真实的网络请求。 但是,您还需要使用一个Response(包装在 Promise中) 来模拟fetch的返回值,因为我们的函数使用它来获取已经创建用户的ID。 因此,你可以尝试编写这样的测试:

jest.mock('node-fetch');

import fetch, {Response} from 'node-fetch';
import {createUser} from './createUser';

test('createUser calls fetch with the right args and returns the user id', async () => {
fetch.mockReturnValue(Promise.resolve(new Response('4')));

const userId = await createUser();

expect(fetch).toHaveBeenCalledTimes(1);
expect(fetch).toHaveBeenCalledWith('https://website.com/users', {
method: 'POST',
});
expect(userId).toBe('4');
});

但是,运行该测试你会发现 createUser 函数执行失败,并抛出错误: TypeError: response.text is not a function。 这是因为从 node-fetch导入的 Response 类已经被模拟了(由测试文件顶部的jest.mock调用) ,所以它不再具备原有的行为,即不存在 text 函数。

为了解决这样的问题,Jest 提供了 jest.requireActual。 要使上述测试用例工作,请对测试文件中导入的内容做如下更改:

// 修改前
jest.mock('node-fetch');
import fetch, {Response} from 'node-fetch';
// 修改后
jest.mock('node-fetch');
import fetch from 'node-fetch';
const {Response} = jest.requireActual('node-fetch');

测试文件将从 node-fetch 导入真实的 Response 对象,而不是模拟的对象。 意味着测试用例现在将正确通过。