67 lines
2.0 KiB
JavaScript
67 lines
2.0 KiB
JavaScript
|
import makeDeferred from './makeDeferred';
|
||
|
|
import Queue from './Queue';
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Utils
|
||
|
|
*/
|
||
|
|
|
||
|
|
function timeout(delay) {
|
||
|
|
const { resolve, promise } = makeDeferred();
|
||
|
|
setTimeout(() => void resolve(Date.now()), delay);
|
||
|
|
return promise;
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Tests
|
||
|
|
*/
|
||
|
|
|
||
|
|
describe('Queue', () => {
|
||
|
|
it('should bind functions to the queue', async () => {
|
||
|
|
const queue = new Queue(2);
|
||
|
|
const mockedTimeout = jest.fn(timeout);
|
||
|
|
const timer = queue.bind(mockedTimeout);
|
||
|
|
const start = Date.now();
|
||
|
|
timer(120).then(now => {
|
||
|
|
const elapsed = now - start;
|
||
|
|
expect(elapsed >= 120 && elapsed < 240).toBe(true);
|
||
|
|
});
|
||
|
|
const end = await timer(120);
|
||
|
|
expect(end - start > 240).toBe(true);
|
||
|
|
expect(mockedTimeout).toBeCalledTimes(2);
|
||
|
|
});
|
||
|
|
it('should prevent task execution when queue limit is reached', async () => {
|
||
|
|
const queue = new Queue(1);
|
||
|
|
const mockedTimeout = jest.fn(timeout);
|
||
|
|
const timer = queue.bind(mockedTimeout);
|
||
|
|
const start = Date.now();
|
||
|
|
const promise = timer(120).then(time => time - start);
|
||
|
|
try {
|
||
|
|
await timer(120);
|
||
|
|
} catch (e) {
|
||
|
|
expect(Date.now() - start < 120).toBe(true);
|
||
|
|
expect(e.message).toBe('Queue limit reached');
|
||
|
|
}
|
||
|
|
const elapsed = await promise;
|
||
|
|
expect(elapsed >= 120 && elapsed < 240).toBe(true);
|
||
|
|
expect(mockedTimeout).toBeCalledTimes(1);
|
||
|
|
});
|
||
|
|
it('should safely bind tasks to the queue', async () => {
|
||
|
|
const queue = new Queue(1);
|
||
|
|
const mockedErrorHandler = jest.fn();
|
||
|
|
const mockedTimeout = jest.fn(timeout);
|
||
|
|
const timer = queue.bindSafe(mockedTimeout, mockedErrorHandler);
|
||
|
|
const start = Date.now();
|
||
|
|
const promise = timer(120).then(time => time - start);
|
||
|
|
await timer(120);
|
||
|
|
expect(Date.now() - start < 120).toBe(true);
|
||
|
|
expect(mockedErrorHandler).toBeCalledTimes(1);
|
||
|
|
expect(mockedErrorHandler).nthCalledWith(
|
||
|
|
1,
|
||
|
|
expect.objectContaining({ message: 'Queue limit reached' })
|
||
|
|
);
|
||
|
|
const elapsed = await promise;
|
||
|
|
expect(elapsed >= 120 && elapsed < 240).toBe(true);
|
||
|
|
expect(mockedTimeout).toBeCalledTimes(1);
|
||
|
|
});
|
||
|
|
});
|