fix: use correct websocket middleware assignment

Author: jyecusch

nitric/faas.py

@@ -733,7 +733,7 @@ def bucket_notification(
         self.__bucket_notification_handler = compose_middleware(*handlers)
         return self
 
-    def websocket(self, *handlers: WebsocketMiddleware) -> FunctionServer:
+    def websocket(self, *handlers: WebsocketHandler | WebsocketMiddleware) -> FunctionServer:
         """
         Register one or more Websocket Trigger Handlers or Middleware.
 

nitric/resources/websockets.py

@@ -92,7 +92,7 @@ def decorator(func: WebsocketHandler) -> None:
                     event_type=event_type,
                 )
             )
-            self._server._websocket_handler(func)
+            self._server.websocket(func)
             return Nitric._register_worker(self._server)  # type: ignore
 
         return decorator