到目前为止,还没有方法jestjs
可以做到这一点。这是建议:参数化模拟返回值
以下是解决方案:
- 利用
sinon.js
sinon.js
支持根据参数模拟返回值,有一个方法名为stub.withArgs(arg1[, arg2, ...]);
使用jest-when - A when(fn).calledWith(args).thenReturn(value) lib 开玩笑
如果您坚持使用jestjs
, since会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