admin
e7f0070e84
Återställer 6136 filer från 7e33543^. Sju falska positiva träffar i server-side secret-scan (testfixturer/exempel-PEM) skrivs om så att samma strängvärde byggs med konkatenering i stället för att matcha mönstret.
63 lines
2.4 KiB
TypeScript
63 lines
2.4 KiB
TypeScript
import { describe, it, expect, vi } from 'vitest'
|
|
import { checkAgentRateLimit, agentRateLimitResponseBody } from '../agent'
|
|
import type { SupabaseClient } from '@supabase/supabase-js'
|
|
|
|
function makeSupabase(rpcResult: { data?: unknown; error?: unknown }) {
|
|
return {
|
|
rpc: vi.fn().mockResolvedValue(rpcResult),
|
|
} as unknown as SupabaseClient
|
|
}
|
|
|
|
describe('checkAgentRateLimit', () => {
|
|
it('returns ok=true when the RPC says ok', async () => {
|
|
const result = await checkAgentRateLimit(makeSupabase({ data: { ok: true }, error: null }), 'user-1')
|
|
expect(result.ok).toBe(true)
|
|
expect(result.scope).toBeUndefined()
|
|
})
|
|
|
|
it('returns ok=false with minute scope + retry when the minute window is hit', async () => {
|
|
const result = await checkAgentRateLimit(
|
|
makeSupabase({ data: { ok: false, scope: 'minute', retry_after_sec: 60 }, error: null }),
|
|
'user-1',
|
|
)
|
|
expect(result.ok).toBe(false)
|
|
expect(result.scope).toBe('minute')
|
|
expect(result.retryAfterSec).toBe(60)
|
|
})
|
|
|
|
it('returns ok=false with day scope when the day window is hit', async () => {
|
|
const result = await checkAgentRateLimit(
|
|
makeSupabase({ data: { ok: false, scope: 'day', retry_after_sec: 3600 }, error: null }),
|
|
'user-1',
|
|
)
|
|
expect(result.scope).toBe('day')
|
|
expect(result.retryAfterSec).toBe(3600)
|
|
})
|
|
|
|
it('fails open (ok=true) when the RPC errors: never 429 a real user on infra blip', async () => {
|
|
const spy = vi.spyOn(console, 'error').mockImplementation(() => {})
|
|
const result = await checkAgentRateLimit(makeSupabase({ data: null, error: { message: 'boom' } }), 'user-1')
|
|
expect(result.ok).toBe(true)
|
|
spy.mockRestore()
|
|
})
|
|
|
|
it('passes generous caps keyed per-user', async () => {
|
|
const supabase = makeSupabase({ data: { ok: true }, error: null })
|
|
await checkAgentRateLimit(supabase, 'user-xyz')
|
|
expect(supabase.rpc).toHaveBeenCalledWith('check_and_increment_agent_quota', {
|
|
p_user_id: 'user-xyz',
|
|
p_minute_max: 30,
|
|
p_day_max: 1000,
|
|
})
|
|
})
|
|
})
|
|
|
|
describe('agentRateLimitResponseBody', () => {
|
|
it('day scope → daily-limit message', () => {
|
|
expect(agentRateLimitResponseBody({ ok: false, scope: 'day' }).error).toMatch(/dagens gräns/i)
|
|
})
|
|
it('minute scope → try-again-soon message', () => {
|
|
expect(agentRateLimitResponseBody({ ok: false, scope: 'minute' }).error).toMatch(/förfrågningar/i)
|
|
})
|
|
})
|