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

node_modules/fastify/test/upgrade.test.js

@@ -0,0 +1,52 @@
+'use strict'
+
+const { describe, test } = require('node:test')
+const Fastify = require('..')
+const { connect } = require('node:net')
+const { once } = require('node:events')
+const dns = require('node:dns').promises
+
+describe('upgrade to both servers', async () => {
+  const localAddresses = await dns.lookup('localhost', { all: true })
+  const skip = localAddresses.length === 1 && 'requires both IPv4 and IPv6'
+
+  await test('upgrade IPv4 and IPv6', { skip }, async t => {
+    t.plan(2)
+
+    const fastify = Fastify()
+    fastify.server.on('upgrade', (req, socket, head) => {
+      t.assert.ok(`upgrade event ${JSON.stringify(socket.address())}`)
+      socket.end()
+    })
+
+    fastify.get('/', (req, res) => {
+      res.send()
+    })
+
+    await fastify.listen()
+    t.after(() => fastify.close())
+
+    {
+      const clientIPv4 = connect(fastify.server.address().port, '127.0.0.1')
+      clientIPv4.write('GET / HTTP/1.1\r\n')
+      clientIPv4.write('Upgrade: websocket\r\n')
+      clientIPv4.write('Connection: Upgrade\r\n')
+      clientIPv4.write('Sec-WebSocket-Key: x3JJHMbDL1EzLkh9GBhXDw==\r\n')
+      clientIPv4.write('Sec-WebSocket-Protocol: com.xxx.service.v1\r\n')
+      clientIPv4.write('Sec-WebSocket-Version: 13\r\n\r\n')
+      clientIPv4.write('\r\n\r\n')
+      await once(clientIPv4, 'close')
+    }
+
+    {
+      const clientIPv6 = connect(fastify.server.address().port, '::1')
+      clientIPv6.write('GET / HTTP/1.1\r\n')
+      clientIPv6.write('Upgrade: websocket\r\n')
+      clientIPv6.write('Connection: Upgrade\r\n')
+      clientIPv6.write('Sec-WebSocket-Key: x3JJHMbDL1EzLkh9GBhXDw==\r\n')
+      clientIPv6.write('Sec-WebSocket-Protocol: com.xxx.service.v1\r\n')
+      clientIPv6.write('Sec-WebSocket-Version: 13\r\n\r\n')
+      await once(clientIPv6, 'close')
+    }
+  })
+})