fixed failing test

Author: eadwinCode

ellar/core/routing/websocket/route.py

@@ -34,13 +34,6 @@ def disconnect(self, func: t.Callable[[WebSocket, int], None]) -> t.Callable:
         self._handlers_kwargs.update(on_disconnect=func)
         return func
 
-    def custom_handler(self, name: str) -> t.Callable:
-        def _wrap(func: t.Callable) -> t.Callable:
-            self._handlers_kwargs.update({name: func})
-            return func
-
-        return _wrap
-
 
 class WebsocketRouteOperation(
     WebSocketOperationMixin, WebsocketRouteOperationBase, StarletteWebSocketRoute

tests/test_middleware/__init__.py

@@ -13,7 +13,8 @@
     WebSocket as EllarWebSocket,
 )
 from ellar.core.context import IExecutionContext
-from ellar.core.middleware import Middleware, SessionMiddleware
+from ellar.core.middleware import Middleware
+from ellar.core.middleware.sessions import SessionMiddleware
 from ellar.core.routing import ModuleRouter
 from ellar.helper.importer import get_class_import