绕过模块模拟
Jest允许在测试用例中模拟整个模块。这对于你测试被测代码是否正常调用某个模块的函数是很有用的。 然而,你可能只想模拟部分模块,剩下的则使用原生的实现。
考虑为这个 createUser
函数编写一个测试用例:
createUser.js
import fetch from 'node-fetch';
export const createUser = async () => {
const response = await fetch('http://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 调用后的返回值
fetch.mockReturnValue(Promise.resolve(new Response('4')));
const userId = await createUser();
expect(fetch).toHaveBeenCalledTimes(1);
expect(fetch).toHaveBeenCalledWith('http://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
对象,而不是模拟的对象。 意味着测试用例现在将正确通过。