53 lines
1.8 KiB
JavaScript
53 lines
1.8 KiB
JavaScript
'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')
|
|
}
|
|
})
|
|
})
|