34 lines
885 B
JavaScript
34 lines
885 B
JavaScript
/**
|
|
* A mock version of a commonUtils `backOff` to test multiple
|
|
* retries.
|
|
*
|
|
* Usage:
|
|
*
|
|
* ```
|
|
* import * as commonUtils from '~/lib/utils/common_utils';
|
|
* import { backoffMockImplementation } from '../../helpers/backoff_helper';
|
|
*
|
|
* beforeEach(() => {
|
|
* // ...
|
|
* jest.spyOn(commonUtils, 'backOff').mockImplementation(backoffMockImplementation);
|
|
* });
|
|
* ```
|
|
*
|
|
* @param {Function} callback
|
|
*/
|
|
export const backoffMockImplementation = (callback) => {
|
|
const q = new Promise((resolve, reject) => {
|
|
const stop = (arg) => (arg instanceof Error ? reject(arg) : resolve(arg));
|
|
const next = () => callback(next, stop);
|
|
// Define a timeout based on a mock timer
|
|
setTimeout(() => {
|
|
callback(next, stop);
|
|
});
|
|
});
|
|
// Run all resolved promises in chain
|
|
jest.runOnlyPendingTimers();
|
|
return q;
|
|
};
|
|
|
|
export default { backoffMockImplementation };
|