feat(ai): simpler In-Memory-Ratelimiter pro IP
This commit is contained in:
29
tests/unit/rate-limit.test.ts
Normal file
29
tests/unit/rate-limit.test.ts
Normal file
@@ -0,0 +1,29 @@
|
||||
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);
|
||||
});
|
||||
});
|
||||
Reference in New Issue
Block a user