50 lines
1.6 KiB
JavaScript
50 lines
1.6 KiB
JavaScript
import { describe, it, before, after, beforeEach } from 'node:test';
|
|
import assert from 'node:assert/strict';
|
|
import Koa from 'koa';
|
|
import bodyParser from 'koa-bodyparser';
|
|
import supertest from 'supertest';
|
|
import { rateLimit } from '../middleware/ratelimit.js';
|
|
|
|
describe('rateLimit 中间件', () => {
|
|
let app, agent;
|
|
|
|
before(() => {
|
|
app = new Koa();
|
|
app.use(bodyParser());
|
|
app.use(rateLimit(3, 60_000));
|
|
app.use(async (ctx) => {
|
|
ctx.body = { ok: true };
|
|
});
|
|
agent = supertest(app.callback());
|
|
});
|
|
|
|
it('前3次请求应正常通过', async () => {
|
|
for (let i = 0; i < 3; i++) {
|
|
const res = await agent.post('/test').send({});
|
|
assert.equal(res.status, 200);
|
|
assert.equal(res.body.ok, true);
|
|
}
|
|
});
|
|
|
|
it('第4次请求应被限流', async () => {
|
|
const res = await agent.post('/test').send({});
|
|
assert.equal(res.status, 200); // HTTP 200，业务码 429
|
|
assert.equal(res.body.code, 429);
|
|
assert.ok(res.body.msg.includes('请求过于频繁'));
|
|
assert.ok(res.headers['retry-after']);
|
|
});
|
|
|
|
it('限流响应头应包含 X-RateLimit 信息', async () => {
|
|
// 新建一个不受之前计数影响的中间件（不同限流阈值 + 不同路径）
|
|
const freshApp = new Koa();
|
|
freshApp.use(bodyParser());
|
|
freshApp.use(rateLimit(100, 60_000));
|
|
freshApp.use(async (ctx) => { ctx.body = { ok: true }; });
|
|
const freshAgent = supertest(freshApp.callback());
|
|
|
|
const res = await freshAgent.post('/headercheck').send({});
|
|
assert.equal(res.headers['x-ratelimit-limit'], '100');
|
|
assert.equal(res.headers['x-ratelimit-remaining'], '99');
|
|
});
|
|
});
|