overleaf/libraries/fetch-utils/test/unit/helpers/TestServer.js

const express = require('express')
const bodyParser = require('body-parser')
const { EventEmitter } = require('events')
const http = require('http')
const https = require('https')
const { promisify } = require('util')

class TestServer {
  constructor() {
    this.app = express()
    this.events = new EventEmitter()

    this.app.use(bodyParser.json())
    this.app.use((req, res, next) => {
      this.lastReq = req
      next()
    })

    // Plain text endpoints

    this.app.get('/hello', (req, res) => {
      res.send('hello')
    })

    this.largePayload = 'x'.repeat(16 * 1024 * 1024)
    this.app.get('/large', (req, res) => {
      res.send(this.largePayload)
    })

    this.app.get('/204', (req, res) => {
      res.status(204).end()
    })

    this.app.get('/empty', (req, res) => {
      res.end()
    })

    this.app.get('/500', (req, res) => {
      res.sendStatus(500)
    })

    this.app.post('/sink', (req, res) => {
      this.events.emit('request-received')
      req.on('data', () => {})
      req.on('end', () => {
        res.status(204).end()
      })
    })

    // JSON endpoints

    this.app.get('/json/hello', (req, res) => {
      res.json({ msg: 'hello' })
    })

    this.app.post('/json/add', (req, res) => {
      const { a, b } = req.body
      res.json({ sum: a + b })
    })

    this.app.get('/json/500', (req, res) => {
      res.status(500).json({ error: 'Internal server error' })
    })

    this.app.get('/json/basic-auth', (req, res) => {
      const expectedAuth =
        'Basic ' + Buffer.from('user:pass').toString('base64')
      if (req.headers.authorization === expectedAuth) {
        res.json({ key: 'verysecret' })
      } else {
        res.status(401).json({ error: 'unauthorized' })
      }
    })

    this.app.post('/json/ignore-request', (req, res) => {
      res.json({ msg: 'hello' })
    })

    // Never returns

    this.app.get('/hang', (req, res) => {})
    this.app.post('/hang', (req, res) => {})

    // Redirect

    this.app.get('/redirect/1', (req, res) => {
      res.redirect('/redirect/2')
    })
    this.app.get('/redirect/2', (req, res) => {
      res.send('body after redirect')
    })
    this.app.get('/redirect/empty-location', (req, res) => {
      res.sendStatus(302)
    })
  }

  start(port, httpsPort, httpsOptions) {
    const startHttp = new Promise((resolve, reject) => {
      this.server = http.createServer(this.app).listen(port, err => {
        if (err) {
          reject(err)
        } else {
          resolve()
        }
      })
    })
    const startHttps = new Promise((resolve, reject) => {
      this.https_server = https
        .createServer(httpsOptions, this.app)
        .listen(httpsPort, err => {
          if (err) {
            reject(err)
          } else {
            resolve()
          }
        })
    })
    return Promise.all([startHttp, startHttps])
  }

  stop() {
    const stopHttp = promisify(this.server.close).bind(this.server)
    const stopHttps = promisify(this.https_server.close).bind(this.https_server)
    this.server.closeAllConnections()
    this.https_server.closeAllConnections()
    return Promise.all([stopHttp(), stopHttps()])
  }
}

module.exports = { TestServer }