绕过模块模拟

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

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

createUser.js

  1. import fetch from 'node-fetch';
  2. export const createUser = async () => {
  3. const response = await fetch('https://website.com/users', {method: 'POST'});
  4. const userId = await response.text();
  5. return userId;
  6. };

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

  1. jest.mock('node-fetch');
  2. import fetch, {Response} from 'node-fetch';
  3. import {createUser} from './createUser';
  4. test('createUser calls fetch with the right args and returns the user id', async () => {
  5. fetch.mockReturnValue(Promise.resolve(new Response('4')));
  6. const userId = await createUser();
  7. expect(fetch).toHaveBeenCalledTimes(1);
  8. expect(fetch).toHaveBeenCalledWith('https://website.com/users', {
  9. method: 'POST',
  10. });
  11. expect(userId).toBe('4');
  12. });

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

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

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

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