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

node_modules/fastify/test/issue-4959.test.js

@@ -0,0 +1,93 @@
+'use strict'
+
+const { test } = require('node:test')
+const http = require('node:http')
+const Fastify = require('../fastify')
+const { setTimeout } = require('node:timers')
+
+/*
+* Ensure that a socket error during the request does not cause the
+* onSend hook to be called multiple times.
+*
+* @see https://github.com/fastify/fastify/issues/4959
+*/
+function runBadClientCall (reqOptions, payload) {
+  let innerResolve, innerReject
+  const promise = new Promise((resolve, reject) => {
+    innerResolve = resolve
+    innerReject = reject
+  })
+
+  const postData = JSON.stringify(payload)
+
+  const req = http.request({
+    ...reqOptions,
+    headers: {
+      'Content-Type': 'application/json',
+      'Content-Length': Buffer.byteLength(postData)
+    }
+  }, () => {
+    innerReject(new Error('Request should have failed'))
+  })
+
+  // Kill the socket immediately (before sending data)
+  req.on('socket', (socket) => {
+    socket.on('connect', () => {
+      setTimeout(() => { socket.destroy() }, 0)
+    })
+  })
+  req.on('error', innerResolve)
+  req.write(postData)
+  req.end()
+
+  return promise
+}
+
+test('should handle a socket error', async (t) => {
+  t.plan(4)
+  const fastify = Fastify()
+
+  function shouldNotHappen () {
+    t.assert.fail('This should not happen')
+  }
+  process.on('unhandledRejection', shouldNotHappen)
+
+  t.after(() => {
+    fastify.close()
+    process.removeListener('unhandledRejection', shouldNotHappen)
+  })
+
+  fastify.addHook('onRequest', async (request, reply) => {
+    t.assert.ok('onRequest hook called')
+  })
+
+  fastify.addHook('onSend', async (request, reply, payload) => {
+    if (request.onSendCalled) {
+      t.assert.fail('onSend hook called more than once')
+      return
+    }
+
+    t.assert.ok('onSend hook called')
+    request.onSendCalled = true
+
+    // Introduce a delay
+    await new Promise(resolve => setTimeout(resolve, 5))
+    return payload
+  })
+
+  // The handler must be async to trigger the error
+  fastify.put('/', async (request, reply) => {
+    t.assert.ok('PUT handler called')
+    return reply.send({ hello: 'world' })
+  })
+
+  await fastify.listen({ port: 0 })
+
+  const err = await runBadClientCall({
+    hostname: 'localhost',
+    port: fastify.server.address().port,
+    path: '/',
+    method: 'PUT'
+  }, { test: 'me' })
+  t.assert.equal(err.code, 'ECONNRESET')
+})