limiter.test.ts

packages/shared/test/limiter.test.ts

1. import { describe, expect, it } from 'vitest'
2. import { TokenBucket } from '../src/limiter.js'
4. describe('TokenBucket', () => {
5. it('allows a burst up to capacity then paces', async () => {
6. let now = 0
7. const slept: number[] = []
8. const bucket = new TokenBucket(
9. 3,
10. 3,
11. () => now,
12. async (ms) => {
13. slept.push(ms)
14. now += ms
15. },
16. )
17. for (let i = 0; i < 3; i++) await bucket.acquire()
18. expect(slept).toHaveLength(0)
19. await bucket.acquire()
20. expect(slept).toHaveLength(1)
21. })
23. it('rejects nonsensical configuration', () => {
24. expect(() => new TokenBucket(0, 10)).toThrow()
25. expect(() => new TokenBucket(10, 0)).toThrow()
26. })
27. })