1
0
Fork 0
peertube/server/tests/shared/mock-servers/mock-429.ts
2021-12-17 12:24:03 +01:00

33 lines
721 B
TypeScript

import express from 'express'
import { Server } from 'http'
import { getPort, randomListen, terminateServer } from './shared'
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)
}
}