使用 Jest 测试框架调用 fetch(url) 时，如何根据 url 模拟变量响应？

Question

我有一个如下所示的模块：

计算平均值.js

const fetch = require('node-fetch')
const stats = require('stats-lite')
const BASE_URL = 'https://www.example.com/api'

const calculateAverage = async(numApiCalls) => {
  const importantData = []

  for (let i = 0; i < numApiCalls; i++) {
      const url = `${BASE_URL}/${i}` // will make requests to https://www.example.com/api/0, https://www.example.com/api/1 and so on....
      const res = await fetch(url)
      const jsonRes = await res.json()

      importantData.push(jsonRes.importantData)
  }

  return stats.mean(importantData)
}

module.exports = calculateAverage

我尝试按照以下方式测试它，但我显然离解决方案还很远：

计算平均值.test.js

const calculateAverage = require('../calculate-average')

jest.mock(
    'node-fetch',
    () => {
        return jest.fn(() => {})
    }
)

test('Should calculate stats for liquidation open interest delatas', async() => {
    const stats = await calculateAverage(100) // Should make 100 API calls.

    console.log(stats)
})

我需要做的是以下内容：

1. 能够为每个 API 调用指定自定义的不同响应。例如，我应该能够指定对https://www.example.com/api/0returns的调用、对returns{ importantData: 0 }的调用等等......https://www.example.com/api/1{ importantData: 1 }
2. url如果向我未指定响应的请求发出，则会提供默认响应。例如，如果对 做出响应，则发送https://www.example.com/api/101默认响应。{ importantData: 1000 }

我最好只使用 Jest 来执行此操作，而不依赖于mock-fetch和等模块jest-mock-fetch。但是，如果不使用的解决方案太复杂，那么我很乐意使用它们。如果不需要，只是不想创建不必要的依赖项。

Answer 1

你当然可以！您可以使用模拟函数mockResolvedValueOnce方法返回特定调用的结果并mockResolvedValue返回默认结果。

jest.mock('node-fetch', () => {
    const generateResponse = (value) => {
        return { json: () => ({ importantData: value }) };
    };

    return jest
        .fn()
        .mockResolvedValue(generateResponse(1000)) // default response
        .mockResolvedValueOnce(generateResponse(0)) // response for first call
        .mockResolvedValueOnce(generateResponse(1)) // response for second call
        .mockResolvedValueOnce(generateResponse(2)); // response for third call
});

请注意，我们返回一个具有该json属性的对象，以便在您调用时返回 jsonres.json()数据calculate-average.js。

如果您想根据url参数返回特定响应，则必须在返回的模拟函数中模拟所需的行为node-fetch。以下示例将模拟返回值，以便对于计数器大于 100 的 URL，它将返回 1000。否则，它将返回与 中存在的相同值url：

jest.mock('node-fetch', () => {
    return jest.fn((url) => {
        // Get and parse the URL parameter.
        const value = parseInt(url.split('/').slice(-1)[0], 10);

        return Promise.resolve({
            json: () => ({ importantData: value > 100 ? 1000 : value })
        });
    });
});