javascript - 模拟单元测试的 express-rate-limit
问题描述
我已经在我的应用程序中实现了“express-rate-limit”作为特定端点的中间件。当涉及到单元测试时,我在尝试对这个中间件进行存根以适当地控制响应时遇到了麻烦。
例如,我有一个每 15 分钟允许 3 个请求的路由,但是,我有 10 个针对该路由的单元测试。前 3 个测试按预期通过,接下来的 7 个测试返回“429 Too Many Requests”响应。
似乎强烈建议使用“express-rate-limit”库进行速率限制,但是,我找不到有关如何在测试环境中使用它的任何信息。
下面显示了简化的实现尝试。
ratelimit.js
exports.createUser = rateLimit({
windowMs: 15 * 60 * 1000, // 15 minute
max: 3,
message: 'Too many requests',
statusCode: 429
})
路由.js
const rateLimits = require('./ratelimit')
let incorrect = function (res, msg) {
res.status(401)
res.send({
status: 'err',
payload: {
msg: msg
}
})
}
router.post('/create', rateLimits.createUser, (req, res) => {
if (!req.body.email) return incorrect(res, 'Email not provided')
userController
.createUser(req.body.email)
.then(user => {
if (!user) incorrect(res, 'User not created')
else correct(res)
})
.catch(() => incorrect(res, 'Internal Error'))
})
测试.js
这应该将速率限制存根以仅允许发出单个请求,这意味着第二次测试应该失败。然而,第二个测试将通过,因为这从未被调用(根据“真实”实现,速率限制仍为 3 )
const rateLimits = require('./ratelimit')
const userController = require('./usercontroller')
let server
let limiter
describe('Users', () => {
before(() => {
limiter = sinon.stub(rateLimits, 'createUser').callsFake(() => rateLimit({
windowMs: 15 * 60 * 1000, // 15 minute
max: 1,
message: 'Too many requests',
statusCode: 429
}))
server = require('../app')
})
it('should return an error for invalid email', done => {
chai.request(server)
.post('/users/create')
.send({
email: 'notanemail'
})
.end((err, res) => {
res.should.have.status(401)
res.body.status.should.equal('err')
done()
})
})
it('should return an error for missing parameter', done => {
chai.request(server)
.post('/users/create')
.send({})
.end((err, res) => {
res.should.have.status(401)
res.body.status.should.equal('err')
done()
})
})
})
这个问题Stubbing Out Middleware强调了应用程序本身应该在 stubbing 执行后初始化,以确保正确加载。这似乎不起作用,在文件顶部创建导入的标准 .implementation 也不起作用。
或者,出于测试目的,我尝试在执行时将一些文本记录到控制台,以确保调用存根函数。该函数似乎永远不会被调用,因为日志永远不会打印到控制台。
limiter = sinon.stub(rateLimits, 'createUser').callsFake(() => console.log("I was executed"))
我尝试的另一个失败的替代方法是直接调用express-rate-limit模块
const rateLimit = require('express-rate-limit')
limiter = sinon.stub(rateLimit.prototype, 'constructor').callsFake(() => rateLimit({
windowMs: 15 * 60 * 1000, // 15 minute
max: 1,
message: 'Too many requests',
statusCode: 429
}))
解决方案
这个问题已经 4 个月了,还没有收到回复,所以我会尽我所能解释它,因为我已经完成并让它工作了。
所以说你正在使用 express-rate-limit 像这样:
中间件.js
const rateLimit = require('express-rate-limit')
const limiter = rateLimit({
windowMs: 60 * 60 * 1000 * 24, // 24 hour window
max: 5, // start blocking after 5 requests
message:
'Too many requests sent from this IP, please try again after an hour'
})
module.exports = {
limiter
}
应用程序.js
const middleware = require('./middleware')
router.post('/my-rate-limited-api', middleware.limiter, async (req, res) => {
try {
const info = await doSomething('my args', req.body)
res.json(info)
} catch (err) {
res.status(err.status || 400).send(`error: ${err.message}`)
}
})
为了能够测试这条路线内的东西,你需要存根限制器。
在您的测试文件中,确保在 req.xml 中包含一个 IP。在你的回复中,存根
req = {
method: 'POST',
ip: '12.12.123.123',
url: '/my-rate-limited-api',
body: [{ id: '123' }]
}
和
res.headers = {}
和
res.setHeader = (x, y) => { res.headers[x] = y }
beforeEach(function () {
const middleware = require('./middleware')
this.sinon.stub(middleware, 'limiter').callsFake(function (req, res, next) {
return next()
})
res = new Promise(function (resolve, reject) {
doneResolve = this.resolve = resolve
doneReject = this.reject = reject
})
res.headers = {}
res.setHeader = (x, y) => { res.headers[x] = y }
req = {
method: 'POST',
ip: '12.12.123.123',
url: '/my-rate-limited-api',
body: [{ id: '123' }]
}
next = this.sinon.spy(function createDeferredNext () {
let resolve
const promise = new Promise(function () {
resolve = arguments[0]
})
const fn = (arg) => resolve(arg)
fn.then = promise.then.bind(promise)
return fn
})
})
从那里,您将能够随心所欲地测试您的功能。
it('can test without rate limits', function () {
router(req, res, next)
return res
.then(ret => {
expect(res.statusCode).to.eql(200)
})
})
推荐阅读
- c# - 将表单转换为高质量的图像
- ssas - 多个维度的度量的 TOTAL 的计算成员
- entity-framework - Entity Framework TPH 检测鉴别器列
- .net - .net mvc 特殊字符输出
- libtorrent - 慢速磁盘的 Libtorrent 设置
- c++ - 普通循环未使用 gcc 4.8.5 自动矢量化
- node.js - 将 Angular CLI 升级到最新版本?
- java - 如何在 Eclipse 中为 SQL Server JDBC 驱动程序配置集成身份验证?
- php - 使用 preg_match() 并替换为将 URL 转换为文本中的锚标记
- r - 基于前两列合并两个数据帧,在R中以不同方式排列相同的信息