fatsify核心功能示例测试!!!

node_modules/fastify/test/logger/response.test.js

@@ -0,0 +1,183 @@
+'use strict'
+
+const stream = require('node:stream')
+
+const t = require('node:test')
+const split = require('split2')
+const pino = require('pino')
+
+const Fastify = require('../../fastify')
+const { partialDeepStrictEqual } = require('../toolkit')
+const { on } = stream
+
+t.test('response serialization', { timeout: 60000 }, async (t) => {
+  t.plan(4)
+
+  await t.test('Should use serializers from plugin and route', async (t) => {
+    const lines = [
+      { msg: 'incoming request' },
+      { test: 'XHello', test2: 'ZHello' },
+      { msg: 'request completed' }
+    ]
+    t.plan(lines.length + 1)
+
+    const stream = split(JSON.parse)
+
+    const loggerInstance = pino({ level: 'info' }, stream)
+    const fastify = Fastify({
+      loggerInstance
+    })
+    t.after(() => fastify.close())
+
+    fastify.register(context1, {
+      logSerializers: { test: value => 'X' + value }
+    })
+
+    function context1 (instance, opts, done) {
+      instance.get('/', {
+        logSerializers: {
+          test2: value => 'Z' + value
+        }
+      }, (req, reply) => {
+        req.log.info({ test: 'Hello', test2: 'Hello' }) // { test: 'XHello', test2: 'ZHello' }
+        reply.send({ hello: 'world' })
+      })
+      done()
+    }
+
+    await fastify.ready()
+
+    {
+      const response = await fastify.inject({ method: 'GET', url: '/' })
+      const body = await response.json()
+      t.assert.deepStrictEqual(body, { hello: 'world' })
+    }
+
+    for await (const [line] of on(stream, 'data')) {
+      t.assert.ok(partialDeepStrictEqual(line, lines.shift()))
+      if (lines.length === 0) break
+    }
+  })
+
+  await t.test('Should use serializers from instance fastify and route', async (t) => {
+    const lines = [
+      { msg: 'incoming request' },
+      { test: 'XHello', test2: 'ZHello' },
+      { msg: 'request completed' }
+    ]
+    t.plan(lines.length + 1)
+
+    const stream = split(JSON.parse)
+
+    const loggerInstance = pino({
+      level: 'info',
+      serializers: {
+        test: value => 'X' + value,
+        test2: value => 'This should be override - ' + value
+      }
+    }, stream)
+    const fastify = Fastify({
+      loggerInstance
+    })
+    t.after(() => fastify.close())
+
+    fastify.get('/', {
+      logSerializers: {
+        test2: value => 'Z' + value
+      }
+    }, (req, reply) => {
+      req.log.info({ test: 'Hello', test2: 'Hello' }) // { test: 'XHello', test2: 'ZHello' }
+      reply.send({ hello: 'world' })
+    })
+
+    await fastify.ready()
+
+    {
+      const response = await fastify.inject({ method: 'GET', url: '/' })
+      const body = await response.json()
+      t.assert.deepStrictEqual(body, { hello: 'world' })
+    }
+
+    for await (const [line] of on(stream, 'data')) {
+      t.assert.ok(partialDeepStrictEqual(line, lines.shift()))
+      if (lines.length === 0) break
+    }
+  })
+
+  await t.test('Should use serializers inherit from contexts', async (t) => {
+    const lines = [
+      { msg: 'incoming request' },
+      { test: 'XHello', test2: 'YHello', test3: 'ZHello' },
+      { msg: 'request completed' }
+    ]
+    t.plan(lines.length + 1)
+
+    const stream = split(JSON.parse)
+
+    const loggerInstance = pino({
+      level: 'info',
+      serializers: {
+        test: value => 'X' + value
+      }
+    }, stream)
+
+    const fastify = Fastify({ loggerInstance })
+    t.after(() => fastify.close())
+
+    fastify.register(context1, { logSerializers: { test2: value => 'Y' + value } })
+
+    function context1 (instance, opts, done) {
+      instance.get('/', {
+        logSerializers: {
+          test3: value => 'Z' + value
+        }
+      }, (req, reply) => {
+        req.log.info({ test: 'Hello', test2: 'Hello', test3: 'Hello' }) // { test: 'XHello', test2: 'YHello', test3: 'ZHello' }
+        reply.send({ hello: 'world' })
+      })
+      done()
+    }
+
+    await fastify.ready()
+
+    {
+      const response = await fastify.inject({ method: 'GET', url: '/' })
+      const body = await response.json()
+      t.assert.deepStrictEqual(body, { hello: 'world' })
+    }
+
+    for await (const [line] of on(stream, 'data')) {
+      t.assert.ok(partialDeepStrictEqual(line, lines.shift()))
+      if (lines.length === 0) break
+    }
+  })
+
+  await t.test('should serialize request and response', async (t) => {
+    const lines = [
+      { req: { method: 'GET', url: '/500' }, msg: 'incoming request' },
+      { req: { method: 'GET', url: '/500' }, msg: '500 error' },
+      { msg: 'request completed' }
+    ]
+    t.plan(lines.length + 1)
+
+    const stream = split(JSON.parse)
+    const fastify = Fastify({ logger: { level: 'info', stream } })
+    t.after(() => fastify.close())
+
+    fastify.get('/500', (req, reply) => {
+      reply.code(500).send(Error('500 error'))
+    })
+
+    await fastify.ready()
+
+    {
+      const response = await fastify.inject({ method: 'GET', url: '/500' })
+      t.assert.strictEqual(response.statusCode, 500)
+    }
+
+    for await (const [line] of on(stream, 'data')) {
+      t.assert.ok(partialDeepStrictEqual(line, lines.shift()))
+      if (lines.length === 0) break
+    }
+  })
+})