57 lines
1.4 KiB
JavaScript
57 lines
1.4 KiB
JavaScript
const AsyncLimiter = require('../../src/util/async-limiter');
|
|
const {test} = require('node:test');
|
|
const assert = require('node:assert');
|
|
|
|
test('Runs callback', async () => {
|
|
/* eslint-disable-next-line require-await */
|
|
const callback = async (a, b) => a + b;
|
|
|
|
const limiter = new AsyncLimiter(callback, 2);
|
|
|
|
assert.deepEqual(await Promise.all([
|
|
limiter.do(1, 2),
|
|
limiter.do(3, 4),
|
|
limiter.do(5, 6),
|
|
limiter.do(7, 8),
|
|
limiter.do(9, 10)
|
|
]), [
|
|
3,
|
|
7,
|
|
11,
|
|
15,
|
|
19
|
|
]);
|
|
});
|
|
|
|
test('Errors', async () => {
|
|
const errorObject = new Error('Testing testing 123'); // want to get the same *exact* object back
|
|
const callback = () => Promise.reject(errorObject);
|
|
const limiter = new AsyncLimiter(callback, 10);
|
|
await assert.rejects(limiter.do(), e => e === errorObject);
|
|
});
|
|
|
|
test('Limit and queue', async () => {
|
|
const calls = [];
|
|
const callback = () => new Promise(resolve => {
|
|
calls.push({
|
|
resolve
|
|
});
|
|
});
|
|
|
|
const limiter = new AsyncLimiter(callback, 5);
|
|
|
|
for (let i = 0; i < 12; i++) {
|
|
limiter.do();
|
|
}
|
|
|
|
assert.strictEqual(calls.length, 5);
|
|
|
|
calls.forEach(i => i.resolve());
|
|
await Promise.resolve();
|
|
assert.strictEqual(calls.length, 10);
|
|
|
|
calls.forEach(i => i.resolve());
|
|
await Promise.resolve();
|
|
assert.strictEqual(calls.length, 12);
|
|
});
|