30 lines
1.1 KiB
TypeScript
30 lines
1.1 KiB
TypeScript
import { describe, it, expect, beforeEach, afterEach, vi } from 'vitest';
|
|
import { createRateLimiter } from '../../src/lib/server/ai/rate-limit';
|
|
|
|
describe('rate-limit', () => {
|
|
beforeEach(() => vi.useFakeTimers());
|
|
afterEach(() => vi.useRealTimers());
|
|
|
|
it('allows first 10 requests, rejects 11th', () => {
|
|
const limiter = createRateLimiter({ windowMs: 60_000, max: 10 });
|
|
for (let i = 0; i < 10; i++) expect(limiter.check('1.2.3.4')).toBe(true);
|
|
expect(limiter.check('1.2.3.4')).toBe(false);
|
|
});
|
|
|
|
it('tracks per-IP independently', () => {
|
|
const limiter = createRateLimiter({ windowMs: 60_000, max: 2 });
|
|
expect(limiter.check('a')).toBe(true);
|
|
expect(limiter.check('a')).toBe(true);
|
|
expect(limiter.check('a')).toBe(false);
|
|
expect(limiter.check('b')).toBe(true);
|
|
});
|
|
|
|
it('resets after window elapses', () => {
|
|
const limiter = createRateLimiter({ windowMs: 1000, max: 1 });
|
|
expect(limiter.check('x')).toBe(true);
|
|
expect(limiter.check('x')).toBe(false);
|
|
vi.advanceTimersByTime(1001);
|
|
expect(limiter.check('x')).toBe(true);
|
|
});
|
|
});
|