Reputation: 2416
I have the following method and I need to write some unit tests for it, but I cannot mock the response, I've tried using the TestScheduler but it was not successful, any help would be appreciated. I'm using jest.
protected waitForResponse(id: number, requestId: string) {
return this.service.getData(id, requestId)
.pipe(
mergeMap((resp: ResponseModel) => {
if (resp.status !== 'WAITING') {
return of(resp);
}
return throwError(resp);
}),
retryWhen(errors => errors.pipe(
concatMap((e, i) =>
iif(
() => i > 11,
// max number of 11 attempts has been reached
throwError(new HttpErrorResponse({status: HttpStatusCode.TOO_MANY_REQUESTS})),
// Otherwise try again in 5 secs
of(e).pipe(delay(5000))
)
)
)
)
);
}
Upvotes: 2
Views: 1922
Reputation: 2416
I've managed to get a solution by using the following function:
const mockFunction = (times) => {
let count = 0;
return Observable.create((observer) => {
if (count++ > times) {
observer.next(latestData);
} else {
observer.next(waitingData);
}
});
};
And them I've used jest spy on to mock the return value:
jest.spyOn(service, 'getData').mockReturnValue(mockFunction(4));
This will return 4 waiting responses and them the latest one.
jest.spyOn(service, 'getData').mockReturnValue(mockFunction(20));
Any number above 11 will result in exceeding my max number of attempts
Upvotes: 4