使用retryWhen进行rxjs单元测试

import { of, throwError, timer } from 'rxjs';
import { mergeMap, retryWhen, take } from 'rxjs/operators';

// 要测试的函数
function fetchData() {
  return of('data').pipe(
    mergeMap(() => {
      // 模拟请求失败
      return throwError('Error');
    }),
    retryWhen(errors => {
      // 重试3次，每次间隔1秒
      return errors.pipe(
        mergeMap((error, index) => {
          if (index < 3) {
            return timer(1000);
          }
          return throwError(error);
        })
      );
    })
  );
}

// 单元测试
describe('fetchData', () => {
  it('should retry 3 times', done => {
    const spy = jest.spyOn(console, 'log');
    fetchData().pipe(take(1)).subscribe({
      next: data => {
        expect(data).toBe('data');
        expect(spy).toHaveBeenCalledTimes(4); // 包括第一次请求和3次重试
        done();
      },
      error: error => {
        done.fail(error);
      }
    });
  });
});