PeerTube/shared/extra-utils/mock-servers/mock-plugin-blocklist.ts

import express, { Request, Response } from 'express'
import { Server } from 'http'
import { randomInt } from '@shared/core-utils'
import { terminateServer } from './utils'

type BlocklistResponse = {
  data: {
    value: string
    action?: 'add' | 'remove'
    updatedAt?: string
  }[]
}

export class MockBlocklist {
  private body: BlocklistResponse
  private server: Server

  initialize () {
    return new Promise<number>(res => {
      const app = express()

      app.get('/blocklist', (req: Request, res: Response) => {
        return res.json(this.body)
      })

      const port = 42201 + randomInt(1, 100)
      this.server = app.listen(port, () => res(port))
    })
  }

  replace (body: BlocklistResponse) {
    this.body = body
  }

  terminate () {
    return terminateServer(this.server)
  }
}