Question

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

calculate-average.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

我尝试按照以下方式进行测试，但显然与解决方案相去甚远：

calculate-average.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)
})

我需要做的是以下事情：

能够为每个API调用指定自定义变化的响应。例如，我应该能够指定对https://www.example.com/api/0的调用返回{ importantData: 0 }，对https://www.example.com/api/1的调用返回{ importantData: 1 }，依此类推...
如果对我未指定响应的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属性的对象，以便当您在res.json()中调用calculate-average.js时它返回json数据。

如果要基于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 })
        });
    });
});

使用Jest测试框架调用fetch（url）时，如何基于url模拟变量响应？

1 个答案: