You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
60 lines
1.4 KiB
60 lines
1.4 KiB
import { describe, expect, it } from '@jest/globals'
|
|
import { RateLimitedQueue } from './RateLimitedQueue.js'
|
|
|
|
const delay = ms => new Promise(resolve => setTimeout(resolve, ms))
|
|
|
|
describe('RateLimitedQueue', () => {
|
|
let pending = 0
|
|
function fn () {
|
|
pending++
|
|
return delay(15).then(() => pending--)
|
|
}
|
|
|
|
it('should run at most N promises at the same time', async () => {
|
|
const queue = new RateLimitedQueue(4)
|
|
const fn2 = queue.wrapPromiseFunction(fn)
|
|
|
|
const result = Promise.all([
|
|
fn2(), fn2(), fn2(), fn2(),
|
|
fn2(), fn2(), fn2(), fn2(),
|
|
fn2(), fn2(),
|
|
])
|
|
|
|
expect(pending).toBe(4)
|
|
|
|
await delay(10)
|
|
expect(pending).toBe(4)
|
|
|
|
await result
|
|
expect(pending).toBe(0)
|
|
})
|
|
|
|
it('should accept Infinity as limit', () => {
|
|
const queue = new RateLimitedQueue(Infinity)
|
|
const fn2 = queue.wrapPromiseFunction(fn)
|
|
|
|
const result = Promise.all([
|
|
fn2(), fn2(), fn2(), fn2(),
|
|
fn2(), fn2(), fn2(), fn2(),
|
|
fn2(), fn2(),
|
|
])
|
|
|
|
expect(pending).toBe(10)
|
|
|
|
return result.then(() => {
|
|
expect(pending).toBe(0)
|
|
})
|
|
})
|
|
|
|
it('should accept non-promise function in wrapPromiseFunction()', () => {
|
|
const queue = new RateLimitedQueue(1)
|
|
function syncFn () { return 1 }
|
|
const fn2 = queue.wrapPromiseFunction(syncFn)
|
|
|
|
return Promise.all([
|
|
fn2(), fn2(), fn2(), fn2(),
|
|
fn2(), fn2(), fn2(), fn2(),
|
|
fn2(), fn2(),
|
|
])
|
|
})
|
|
})
|
|
|