如何为特定的 axios 调用模拟 jest.spyOn

Question

如何模拟特定的 axios 调用？ 想象一下 2 次 GET 调用：

await axios.get('api/numbers');
await axios.get('api/letters');

然后这是在测试中进行的：

const mockGet = jest.spyOn(axios, 'get');
mockGet.mockReturnValueOnce(Promise.resolve({ data: 1 }));
mockGet.mockReturnValueOnce(Promise.resolve({ data: 'a' }));

如何根据传递给 axios 的 url 为mockReturnValueOnce创建条件（比如'api/numbers' -> return Promise.resolve({ data: 1 }) ）

Answer 1

到目前为止，还没有jestjs方法来做到这一点。 这是建议：参数化模拟返回值

以下是解决方案：

1. 使用sinon.js

sinon.js支持根据参数模拟返回值，有一个方法叫做stub.withArgs(arg1[, arg2, ...]);

2. 使用jest-when - 用于开玩笑的 when (fn).CalledWith(args).thenReturn(value) lib

3. 如果您坚持使用jestjs ，因为mockReturnValueOnce按顺序返回值（第一次调用、第二次调用等）。 您可以使用以下方式：

index.js ：

import axios from 'axios';

export async function main() {
  const numbersRes = await axios.get('api/numbers');
  const lettersRes = await axios.get('api/letters');
  return { numbersRes, lettersRes };
}

index.test.js ：

import { main } from '.';
import axios from 'axios';

describe('59751925', () => {
  it('should pass', async () => {
    const mockGet = jest.spyOn(axios, 'get');
    mockGet.mockImplementation((url) => {
      switch (url) {
        case 'api/numbers':
          return Promise.resolve({ data: 1 });
        case 'api/letters':
          return Promise.resolve({ data: 'a' });
      }
    });
    const actual = await main();
    expect(actual.numbersRes).toEqual({ data: 1 });
    expect(actual.lettersRes).toEqual({ data: 'a' });
  });
});

单元测试结果：

 PASS  src/stackoverflow/59751925/index.test.js (11.168s)
  59751925
    ✓ should pass (9ms)

Test Suites: 1 passed, 1 total
Tests:       1 passed, 1 total
Snapshots:   0 total
Time:        13.098s