fastify/test/types/serverFactory.tst.ts

Fast and low overhead web framework, for Node.js

import * as http from 'node:http'
import { expect } from 'tstyche'
import fastify, { FastifyServerFactory } from '../../fastify.js'

// Custom Server
type CustomType = void
interface CustomIncomingMessage extends http.IncomingMessage {
  fakeMethod?: () => CustomType;
}

interface CustomServerResponse extends http.ServerResponse {
  fakeMethod?: () => CustomType;
}

const serverFactory: FastifyServerFactory<http.Server> = (handler, opts) => {
  const server = http.createServer((req: CustomIncomingMessage, res: CustomServerResponse) => {
    req.fakeMethod = () => {}
    res.fakeMethod = () => {}

    handler(req, res)
  })

  return server
}

// The request and reply objects should have the fakeMethods available (even though they may be undefined)
const customServer = fastify<http.Server, CustomIncomingMessage, CustomServerResponse>({ serverFactory })

customServer.get('/', function (request, reply) {
  if (request.raw.fakeMethod) {
    expect(request.raw.fakeMethod()).type.toBe<CustomType>()
  }

  if (reply.raw.fakeMethod) {
    expect(reply.raw.fakeMethod()).type.toBe<CustomType>()
  }
})