req-id-gen-factory.test.js — fastify Source File

Source Code

'use strict'

const { test } = require('node:test')
const { reqIdGenFactory } = require('../../lib/req-id-gen-factory')

test('should create incremental ids deterministically', t => {
  t.plan(1)
  const reqIdGen = reqIdGenFactory()

  for (let i = 1; i < 1e4; ++i) {
    if (reqIdGen() !== 'req-' + i.toString(36)) {
      t.assert.fail()
      break
    }
  }
  t.assert.ok(true)
})

test('should have prefix "req-"', t => {
  t.plan(1)
  const reqIdGen = reqIdGenFactory()

  t.assert.ok(reqIdGen().startsWith('req-'))
})

test('different id generator functions should have separate internal counters', t => {
  t.plan(5)
  const reqIdGenA = reqIdGenFactory()
  const reqIdGenB = reqIdGenFactory()

  t.assert.strictEqual(reqIdGenA(), 'req-1')
  t.assert.strictEqual(reqIdGenA(), 'req-2')
  t.assert.strictEqual(reqIdGenB(), 'req-1')
  t.assert.strictEqual(reqIdGenA(), 'req-3')
  t.assert.strictEqual(reqIdGenB(), 'req-2')
})

test('should start counting with 1', t => {
  t.plan(1)
  const reqIdGen = reqIdGenFactory()

  t.assert.strictEqual(reqIdGen(), 'req-1')
})

test('should handle requestIdHeader and return provided id in header', t => {
  t.plan(1)

  const reqIdGen = reqIdGenFactory('id')

  t.assert.strictEqual(reqIdGen({ headers: { id: '1337' } }), '1337')
})

test('should handle requestIdHeader and fallback if id is not provided in header', t => {
  t.plan(1)

  const reqIdGen = reqIdGenFactory('id')

  t.assert.strictEqual(reqIdGen({ headers: { notId: '1337' } }), 'req-1')
})

// ... (74 more lines)