strfry-policies/src/policies/rate-limit-policy.test.ts
2023-03-28 12:30:53 -05:00

23 lines
955 B
TypeScript

import { assertEquals } from '../deps.ts';
import { buildEvent, buildInputMessage } from '../test.ts';
import rateLimitPolicy from './rate-limit-policy.ts';
Deno.test({
name: 'blocks events from IPs that are publishing events too quickly',
fn: async () => {
const opts = { max: 4, databaseUrl: undefined };
const msg = buildInputMessage({ sourceType: 'IP4', sourceInfo: '1.1.1.1', event: buildEvent() });
assertEquals((await rateLimitPolicy(msg, opts)).action, 'accept');
assertEquals((await rateLimitPolicy(msg, opts)).action, 'accept');
assertEquals((await rateLimitPolicy(msg, opts)).action, 'accept');
assertEquals((await rateLimitPolicy(msg, opts)).action, 'accept');
assertEquals((await rateLimitPolicy(msg, opts)).action, 'reject');
assertEquals((await rateLimitPolicy(msg, opts)).action, 'reject');
assertEquals((await rateLimitPolicy(msg, opts)).action, 'reject');
},
sanitizeResources: false,
});