feat(schema): set hide true for websockets (#336)

* feat(schema): set hide true for websockets

Signed-off-by: Roberto Bianchi <[email protected]>

* update

Signed-off-by: Roberto Bianchi <[email protected]>

---------

Signed-off-by: Roberto Bianchi <[email protected]>

Author: rozzilla

index.js

@@ -162,6 +162,11 @@ function fastifyWebsocket (fastify, opts, next) {
       isWebsocketRoute = true
 
       if (routeOptions.websocket) {
+        if (!routeOptions.schema) {
+          routeOptions.schema = {}
+        }
+        routeOptions.schema.hide = true
+
         wsHandler = routeOptions.handler
         handler = function (_, reply) {
           reply.code(404).send()

test/hooks.test.js

@@ -8,15 +8,18 @@ const WebSocket = require('ws')
 const split = require('split2')
 
 test('Should run onRequest, preValidation, preHandler hooks', (t, end) => {
-  t.plan(7)
+  t.plan(8)
   const fastify = Fastify()
 
   t.after(() => fastify.close())
 
   fastify.register(fastifyWebsocket)
 
   fastify.register(async function (fastify) {
-    fastify.addHook('onRequest', async () => t.assert.ok('called', 'onRequest'))
+    fastify.addHook('onRequest', async ({ routeOptions: { schema: { hide } } }) => {
+      t.assert.ok('called', 'onRequest')
+      t.assert.strictEqual(hide, true, 'schema hide property should be set to true when route option is websocket')
+    })
     fastify.addHook('preParsing', async () => t.assert.ok('called', 'preParsing'))
     fastify.addHook('preValidation', async () => t.assert.ok('called', 'preValidation'))
     fastify.addHook('preHandler', async () => t.assert.ok('called', 'preHandler'))