如何注入多个模拟承诺以对 JS 中的 fetch 进行单元测试？

Question

我正在尝试为我的 JS 项目中的 api 调用设置单元测试。 能够为单个 API 调用设置它，使用以下格式

describe('Token refresh success', () => {
  beforeAll(() => {
    global.fetch = () =>
      Promise.resolve({
        json: () => Promise.resolve(mockTokenCreateSuccess),
      })
  })

  afterAll(() => {
    global.fetch = unmockedFetch
  })

  test('testTokenRefreshSuccess', async () => {
    const tokenData = await c.tokenRefresh();
    expect(tokenData.access_token).toEqual('SYoHdm4yw');
    expect(tokenData.refresh_token).toEqual('QxJ3yEgX4NThbTE66u7lshWTpQkRBilq');
  });
})

现在这种格式效果很好，我可以通过注入一个 promise 来创建单独的测试。

现在，有一个案例我想对特定的 API 调用进行两次单元测试。 我需要第一次注入失败响应，第二次注入成功响应。

我尝试了以下方法，但没有奏效：

describe('Token refresh trigger as expected on create token fail', () => {
  beforeAll(() => {
    global.fetch = () => [
      Promise.resolve({
        json: () => Promise.reject(mockError(400)),
      }),
      Promise.resolve({
        json: () => Promise.resolve(mockTokenCreateSuccess),
      })
    ] 
  })

  afterAll(() => {
    global.fetch = unmockedFetch
  })

  test('testTokenRefreshTriggerOnTokenCreateFail', async () => {
    const tokenData = await c.tokenRefresh();
    expect(tokenData.access_token).toEqual('jkhjk');
    expect(tokenData.refresh_token).toEqual('dfdfdf');
  });
})

如果出现 400 错误，我的 tokenRefresh() function 应该被调用 2 次。

async tokenRefresh(retryCount = 0) {
        console.log('tokenRefresh');

        const request = this.getTokenRefreshRequest();
        try {
            const response = await this.fetchData(request);
            console.log('token refresh success', response);
            return { access_token, refresh_token };
        } catch (err) {
            if ((err.status == 400 || err.status == 401) && retryCount < 2) {
                await this.tokenRefresh(retryCount++);
            } else {
                throw new Error(`unable to refresh a token from API ${err.status}`);
            }
        }
    };

我可以通过使用单个 promise 来验证

  Promise.resolve({
    json: () => Promise.reject(mockError(400)),
  })

tokenRefresh() 按预期再次被调用，但第二次我无法弄清楚如何通过成功 promise，目前它第二次也失败了。

Answer 1

我能够找到一个有效的组合

beforeAll(() => {
    global.fetch = fetchMock.mockRejectOnce(customError) // makes sure first api call fails
    .mockResponseOnce(JSON.stringify(successResponse)) // makes sure second api call succeeds
  })

这使我的测试通过