fastify
Version:
Fast and low overhead web framework, for Node.js
486 lines (429 loc) • 14.7 kB
JavaScript
'use strict'
const { test } = require('node:test')
const Fastify = require('../..')
const loggerUtils = require('../../lib/logger-factory')
const { createLogController, LogController } = require('../../lib/logger-factory')
const { serializers } = require('../../lib/logger-pino')
test('time resolution', t => {
t.plan(2)
t.assert.strictEqual(typeof loggerUtils.now, 'function')
t.assert.strictEqual(typeof loggerUtils.now(), 'number')
})
test('The logger should add a unique id for every request', (t, done) => {
const ids = []
const fastify = Fastify()
fastify.get('/', (req, reply) => {
t.assert.ok(req.id)
reply.send({ id: req.id })
})
fastify.listen({ port: 0 }, err => {
t.assert.ifError(err)
const queue = new Queue()
for (let i = 0; i < 10; i++) {
queue.add(checkId)
}
queue.add(() => {
fastify.close()
done()
})
})
function checkId (done) {
fastify.inject({
method: 'GET',
url: 'http://localhost:' + fastify.server.address().port
}, (err, res) => {
t.assert.ifError(err)
const payload = JSON.parse(res.payload)
t.assert.ok(ids.indexOf(payload.id) === -1, 'the id should not be duplicated')
ids.push(payload.id)
done()
})
}
})
test('The logger should not reuse request id header for req.id', (t, done) => {
const fastify = Fastify()
fastify.get('/', (req, reply) => {
t.assert.ok(req.id)
reply.send({ id: req.id })
})
fastify.listen({ port: 0 }, err => {
t.assert.ifError(err)
fastify.inject({
method: 'GET',
url: 'http://localhost:' + fastify.server.address().port,
headers: {
'Request-Id': 'request-id-1'
}
}, (err, res) => {
t.assert.ifError(err)
const payload = JSON.parse(res.payload)
t.assert.ok(payload.id !== 'request-id-1', 'the request id from the header should not be returned with default configuration')
t.assert.ok(payload.id === 'req-1') // first request id when using the default configuration
fastify.close()
done()
})
})
})
test('The logger should reuse request id header for req.id if requestIdHeader is set', (t, done) => {
const fastify = Fastify({
requestIdHeader: 'request-id'
})
fastify.get('/', (req, reply) => {
t.assert.ok(req.id)
reply.send({ id: req.id })
})
fastify.listen({ port: 0 }, err => {
t.assert.ifError(err)
fastify.inject({
method: 'GET',
url: 'http://localhost:' + fastify.server.address().port,
headers: {
'Request-Id': 'request-id-1'
}
}, (err, res) => {
t.assert.ifError(err)
const payload = JSON.parse(res.payload)
t.assert.ok(payload.id === 'request-id-1', 'the request id from the header should be returned')
fastify.close()
done()
})
})
})
function Queue () {
this.q = []
this.running = false
}
Queue.prototype.add = function add (job) {
this.q.push(job)
if (!this.running) this.run()
}
Queue.prototype.run = function run () {
this.running = true
const job = this.q.shift()
job(() => {
if (this.q.length) {
this.run()
} else {
this.running = false
}
})
}
test('The logger should error if both stream and file destination are given', t => {
t.plan(2)
const stream = require('node:stream').Writable
try {
Fastify({
logger: {
level: 'info',
stream,
file: '/test'
}
})
} catch (err) {
t.assert.strictEqual(err.code, 'FST_ERR_LOG_INVALID_DESTINATION')
t.assert.strictEqual(err.message, 'Cannot specify both logger.stream and logger.file options')
}
})
test('LogController defaults', t => {
t.plan(3)
const dispatcher = new LogController()
t.assert.strictEqual(dispatcher.disableRequestLogging, false)
t.assert.strictEqual(dispatcher.requestIdLogLabel, 'reqId')
t.assert.strictEqual(dispatcher.isLogDisabled({}), false)
})
test('LogController with custom options', t => {
t.plan(2)
const dispatcher = new LogController({
disableRequestLogging: true,
requestIdLogLabel: 'traceId'
})
t.assert.strictEqual(dispatcher.disableRequestLogging, true)
t.assert.strictEqual(dispatcher.requestIdLogLabel, 'traceId')
})
test('LogController with function disableRequestLogging', t => {
t.plan(2)
const dispatcher = new LogController({
disableRequestLogging: (req) => req.skip === true
})
t.assert.strictEqual(dispatcher.isLogDisabled({ skip: true }), true)
t.assert.strictEqual(dispatcher.isLogDisabled({ skip: false }), false)
})
test('requestCompleted should not log when logging is disabled (boolean)', t => {
t.plan(1)
const logController = new LogController({ disableRequestLogging: true })
const log = {
error: () => { t.assert.fail('error should not be called') },
info: () => { t.assert.fail('info should not be called') }
}
const request = {}
const reply = { request, log }
logController.requestCompleted(new Error('test'), request, reply)
t.assert.ok(true, 'logger was not called')
})
test('requestCompleted should not log when logging is disabled (function)', t => {
t.plan(1)
const logController = new LogController({ disableRequestLogging: () => true })
const log = {
error: () => { t.assert.fail('error should not be called') },
info: () => { t.assert.fail('info should not be called') }
}
const request = {}
const reply = { request, log }
logController.requestCompleted(new Error('test'), request, reply)
t.assert.ok(true, 'logger was not called')
})
test('requestCompleted should log error when err is present', t => {
t.plan(1)
const logController = new LogController({ disableRequestLogging: false })
const err = new Error('test')
const log = {
error: (data, msg) => {
t.assert.strictEqual(msg, 'request errored')
}
}
const request = {}
const reply = { request, log, elapsedTime: 42 }
logController.requestCompleted(err, request, reply)
})
test('requestCompleted should log info when no error', t => {
t.plan(1)
const logController = new LogController({ disableRequestLogging: false })
const log = {
info: (data, msg) => {
t.assert.strictEqual(msg, 'request completed')
}
}
const request = {}
const reply = { request, log, elapsedTime: 42 }
logController.requestCompleted(null, request, reply)
})
test('defaultErrorLog should not log when logging is disabled (boolean)', t => {
t.plan(1)
const logController = new LogController({ disableRequestLogging: true })
const log = {
info: () => { t.assert.fail('info should not be called') },
error: () => { t.assert.fail('error should not be called') }
}
const request = {}
const reply = { request, log, statusCode: 404 }
logController.defaultErrorLog(new Error('not found'), request, reply)
t.assert.ok(true, 'logger was not called')
})
test('defaultErrorLog should not log when logging is disabled (function)', t => {
t.plan(1)
const logController = new LogController({ disableRequestLogging: () => true })
const log = {
info: () => { t.assert.fail('info should not be called') },
error: () => { t.assert.fail('error should not be called') }
}
const request = {}
const reply = { request, log, statusCode: 404 }
logController.defaultErrorLog(new Error('not found'), request, reply)
t.assert.ok(true, 'logger was not called')
})
test('defaultErrorLog should log info for 4xx errors', t => {
t.plan(1)
const logController = new LogController({ disableRequestLogging: false })
const err = new Error('not found')
const log = {
info: (data, msg) => {
t.assert.strictEqual(msg, 'not found')
}
}
const request = {}
const reply = { request, log, statusCode: 404 }
logController.defaultErrorLog(err, request, reply)
})
test('defaultErrorLog should log error for 5xx errors', t => {
t.plan(1)
const logController = new LogController({ disableRequestLogging: false })
const err = new Error('internal error')
const log = {
error: (data, msg) => {
t.assert.strictEqual(msg, 'internal error')
}
}
const request = {}
const reply = { request, log, statusCode: 500 }
logController.defaultErrorLog(err, request, reply)
})
test('writeHeadError should not log when logging is disabled (boolean)', t => {
t.plan(1)
const logController = new LogController({ disableRequestLogging: true })
const log = {
warn: () => { t.assert.fail('warn should not be called') }
}
const request = {}
const reply = { request, log }
logController.writeHeadError(new Error('write head failed'), request, reply)
t.assert.ok(true, 'logger was not called')
})
test('writeHeadError should not log when logging is disabled (function)', t => {
t.plan(1)
const logController = new LogController({ disableRequestLogging: () => true })
const log = {
warn: () => { t.assert.fail('warn should not be called') }
}
const request = {}
const reply = { request, log }
logController.writeHeadError(new Error('write head failed'), request, reply)
t.assert.ok(true, 'logger was not called')
})
test('writeHeadError should log warn with error message', t => {
t.plan(2)
const logController = new LogController({ disableRequestLogging: false })
const error = new Error('write head failed')
const request = {}
const reply = {
request,
log: {
warn: (data, msg) => {
t.assert.strictEqual(msg, 'write head failed')
t.assert.strictEqual(data.err, error)
}
}
}
logController.writeHeadError(error, request, reply)
})
test('serializerError should not log when logging is disabled (boolean)', t => {
t.plan(1)
const logController = new LogController({ disableRequestLogging: true })
const log = {
error: () => { t.assert.fail('error should not be called') }
}
const request = {}
const reply = { request, log }
logController.serializerError(new Error('serializer failed'), request, reply, { statusCode: 500 })
t.assert.ok(true, 'logger was not called')
})
test('serializerError should not log when logging is disabled (function)', t => {
t.plan(1)
const logController = new LogController({ disableRequestLogging: () => true })
const log = {
error: () => { t.assert.fail('error should not be called') }
}
const request = {}
const reply = { request, log }
logController.serializerError(new Error('serializer failed'), request, reply, { statusCode: 500 })
t.assert.ok(true, 'logger was not called')
})
test('serializerError should log error with status code', t => {
t.plan(2)
const logController = new LogController({ disableRequestLogging: false })
const err = new Error('serializer failed')
const request = {}
const reply = {
request,
log: {
error: (data, msg) => {
t.assert.strictEqual(msg, 'The serializer for the given status code failed')
t.assert.strictEqual(data.statusCode, 500)
}
}
}
logController.serializerError(err, request, reply, { statusCode: 500 })
})
test('createLogController should use LogController instance directly', t => {
t.plan(2)
const custom = new LogController({ disableRequestLogging: true, requestIdLogLabel: 'traceId' })
const dispatcher = createLogController({ logController: custom })
t.assert.strictEqual(dispatcher, custom)
t.assert.strictEqual(dispatcher.requestIdLogLabel, 'traceId')
})
test('createLogController should create default when no instance provided', t => {
t.plan(2)
const dispatcher = createLogController({ disableRequestLogging: false, requestIdLogLabel: 'reqId' })
t.assert.ok(dispatcher instanceof LogController)
t.assert.strictEqual(dispatcher.requestIdLogLabel, 'reqId')
})
test('createLogController should throw on invalid type', t => {
t.plan(1)
t.assert.throws(() => {
createLogController({ disableRequestLogging: false, logController: 'bad' })
}, { code: 'FST_ERR_LOG_INVALID_LOG_CONTROLLER' })
})
test('createLogController should throw when plain object is provided', t => {
t.plan(1)
t.assert.throws(() => {
createLogController({ disableRequestLogging: false, logController: { incomingRequest: () => { } } })
}, { code: 'FST_ERR_LOG_INVALID_LOG_CONTROLLER' })
})
test('createLogController should accept null', t => {
t.plan(1)
const logController = createLogController({ disableRequestLogging: false, logController: null })
t.assert.ok(logController)
})
test('LogController subclass should work', t => {
t.plan(2)
class MyDispatcher extends LogController {
constructor () {
super({ requestIdLogLabel: 'traceId' })
}
incomingRequest (request, reply, metadata) {
t.assert.ok(true, 'custom incomingRequest called')
}
}
const dispatcher = new MyDispatcher()
t.assert.strictEqual(dispatcher.requestIdLogLabel, 'traceId')
dispatcher.incomingRequest({})
})
test('LogController subclass keeps defaults for non-overridden methods', t => {
t.plan(1)
class MyDispatcher extends LogController {
incomingRequest () { }
}
const dispatcher = new MyDispatcher()
const log = {
info: (data, msg) => {
t.assert.strictEqual(msg, 'request completed')
}
}
const request = {}
const reply = { request, log, elapsedTime: 42 }
dispatcher.requestCompleted(null, request, reply)
})
test('serviceUnavailable should log with logger and receive server', t => {
t.plan(2)
const dispatcher = new LogController()
const fakeLogger = {
info: (data, msg) => {
t.assert.deepStrictEqual(data, { res: { statusCode: 503 } })
t.assert.strictEqual(msg, 'request aborted - refusing to accept new requests as server is closing')
}
}
const fakeServer = { name: 'test-server' }
dispatcher.serviceUnavailable(fakeLogger, fakeServer)
})
test('serviceUnavailable subclass can use server', t => {
t.plan(1)
const fakeServer = { name: 'test-server' }
class MyDispatcher extends LogController {
serviceUnavailable (logger, server) {
t.assert.strictEqual(server, fakeServer)
}
}
const dispatcher = new MyDispatcher()
dispatcher.serviceUnavailable({}, fakeServer)
})
test('LogController is exported from fastify', t => {
t.plan(2)
t.assert.ok(Fastify.LogController)
t.assert.strictEqual(Fastify.LogController, LogController)
})
test('The serializer prevent fails if the request socket is undefined', t => {
t.plan(1)
const serialized = serializers.req({
method: 'GET',
url: '/',
socket: undefined,
headers: {}
})
t.assert.deepStrictEqual(serialized, {
method: 'GET',
url: '/',
version: undefined,
host: undefined,
remoteAddress: undefined,
remotePort: undefined
})
})