Fix done is undefined error (#200)

Signed-off-by: Matteo Collina <[email protected]>

Author: mcollina

index.js

@@ -93,7 +93,7 @@ function fastifyWebsocket (fastify, opts, next) {
     done()
   })
 
-  fastify.addHook('onResponse', (request, reply, error, done) => {
+  fastify.addHook('onResponse', (request, reply, done) => {
     if (request.ws) {
       request.raw[kWs].destroy()
     }

test/hooks.js

@@ -5,6 +5,7 @@ const net = require('net')
 const Fastify = require('fastify')
 const fastifyWebsocket = require('..')
 const WebSocket = require('ws')
+const split = require('split2')
 
 test('Should run onRequest, preValidation, preHandler hooks', t => {
   t.plan(7)
@@ -335,8 +336,9 @@ test('Should not hijack reply for an normal request to a websocket route that is
 })
 
 test('Should not hijack reply for an WS request to a WS route that gets sent a normal HTTP response in a hook', t => {
-  t.plan(2)
-  const fastify = Fastify()
+  t.plan(6)
+  const stream = split(JSON.parse)
+  const fastify = Fastify({ logger: { stream } })
 
   fastify.register(fastifyWebsocket)
   fastify.register(async function (fastify) {
@@ -348,6 +350,10 @@ test('Should not hijack reply for an WS request to a WS route that gets sent a n
     })
   })
 
+  stream.on('data', (chunk) => {
+    t.ok(chunk.level < 50)
+  })
+
   fastify.listen({ port: 0 }, err => {
     t.error(err)