Jest 绕过模块模拟

2021-09-18 20:27 更新

Jest 允许在测试中模拟整个模块,这对于测试你的代码是否正确地从该模块调用函数非常有用。但是,有时你可能希望在测试文件中使用部分模拟模块,在这种情况下,希望访问原始实现,而不是模拟版本。

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

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

你的测试将要模拟​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('http://website.com/users', {
  9. method: 'POST',
  10. });
  11. expect(userId).toBe('4');
  12. });

但是,如果运行该测试,你会发现该​createUser​函数会失败,并抛出错误:​TypeError: response.text is not a function​。这是因为​Response从中导入的类​node-fetch​已被模拟(由于​jest.mock​测试文件顶部的调用),因此它不再按应有的方式运行。

为了解决此类问题,Jest 提供了​jest.requireActual​帮助程序。要使上述测试工作,请对测试文件中的导入进行以下更改:

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

这允许你的测试文件从node-fetch​导入实际​Response​对象​,而不是模拟版本。这意味着测试现在将正确通过。


以上内容是否对您有帮助:
在线笔记
App下载
App下载

扫描二维码

下载编程狮App

公众号
微信公众号

编程狮公众号