NextApiResponse and NextApiRequest mocks

nextapimock.ts

import { IncomingMessage } from 'http'
import {
  NextApiRequestCookies,
  NextApiRequestQuery,
} from 'next/dist/next-server/server/api-utils'
import { Socket } from 'net'
import { ServerResponse } from 'http'
import { NextApiRequest, NextApiResponse } from 'next'
import { Env } from 'next/dist/lib/load-env-config'

export type NextApiRequestOptions = Partial<NextApiRequestMock>
export class NextApiRequestMock extends IncomingMessage
  implements NextApiRequest {
  public query: NextApiRequestQuery = {}
  public cookies: NextApiRequestCookies = {}
  public env: Env = {}
  public body: any

  constructor(options?: NextApiRequestOptions) {
    super(new Socket())

    if (options) {
      this.method = options.method
      this.body = options.body
      this.query = options.query || {}
      this.headers = options.headers || {}
      this.env = options.env || {}
    }
  }
}

export class NextApiResponseMock extends ServerResponse
  implements NextApiResponse {
  public body: any
  public jsonBody: any

  send(body: any) {
    this.body = body
  }

  json(jsonBody: any) {
    this.jsonBody = jsonBody
  }

  status(statusCode: number) {
    this.statusCode = statusCode
    return this
  }

  setPreviewData(
    data: object | string,
    options?: {
      maxAge?: number
    }
  ) {
    return this
  }

  clearPreviewData() {
    return this
  }
}