FilesExpand file tree

main

/

rate-limit.test.ts

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

21 lines (19 loc) · 965 Bytes

main

/

rate-limit.test.ts

Copy path

Top

File metadata and controls

• Code
• Blame

21 lines (19 loc) · 965 Bytes

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

import { test } from "node:test";

import assert from "node:assert/strict";

import { createLimiter, clientIp } from "@/lib/rate-limit";

test("a limiter blocks after max hits within the window", () => {

const limiter = createLimiter(2, 60_000);

assert.equal(limiter.isLimited("a"), false);

limiter.record("a");

limiter.record("a");

assert.equal(limiter.isLimited("a"), true);

assert.equal(limiter.isLimited("b"), false);

limiter.reset("a");

assert.equal(limiter.isLimited("a"), false);

});

test("clientIp prefers the Cloudflare header", () => {

const req = new Request("http://localhost/", { headers: { "cf-connecting-ip": "203.0.113.1", "x-forwarded-for": "198.51.100.1, 10.0.0.1" } });

assert.equal(clientIp(req), "203.0.113.1");

assert.equal(clientIp(new Request("http://localhost/", { headers: { "x-forwarded-for": "198.51.100.1, 10.0.0.1" } })), "198.51.100.1");

assert.equal(clientIp(new Request("http://localhost/")), "unknown");

});