PeerTube/shared/extra-utils/mock-servers/mock-429.ts

34 lines
720 B
TypeScript
Raw Normal View History

2021-11-16 11:17:52 +01:00
import express from 'express'
import { Server } from 'http'
import { getPort, randomListen, terminateServer } from './utils'
export class Mock429 {
private server: Server
private responseSent = false
async initialize () {
const app = express()
app.get('/', (req: express.Request, res: express.Response, next: express.NextFunction) => {
if (!this.responseSent) {
this.responseSent = true
// Retry after 5 seconds
res.header('retry-after', '2')
return res.sendStatus(429)
}
return res.sendStatus(200)
})
this.server = await randomListen(app)
return getPort(this.server)
}
terminate () {
return terminateServer(this.server)
}
}