wraps websocket handler

sanic/app.py

@@ -6,7 +6,7 @@
 
 from asyncio import CancelledError, Protocol, ensure_future, get_event_loop
 from collections import defaultdict, deque
-from functools import partial
+from functools import partial, wraps
 from inspect import getmodulename, isawaitable, signature, stack
 from socket import socket
 from ssl import Purpose, SSLContext, create_default_context
@@ -494,6 +494,7 @@ def response(handler):
             else:
                 routes = []
 
+            @wraps(handler)
             async def websocket_handler(request, *args, **kwargs):
                 request.app = self
                 if not getattr(handler, "__blueprintname__", False):

tests/test_url_for_strict_slashes.py

@@ -0,0 +1,32 @@
+import asyncio
+
+from sanic.blueprints import Blueprint
+
+
+def test_websocket_bp_route_wraps(app):
+    """Tests that blueprint websocket route is wrapped."""
+    event = asyncio.Event()
+    bp = Blueprint("test_bp", url_prefix="/bp")
+
+    @bp.websocket("/route")
+    async def test_route(request, ws):
+        event.set()
+
+    @bp.websocket("/route2")
+    async def test_route2(request, ws):
+        event.set()
+
+    app.blueprint(bp)
+
+    uri = app.url_for("test_bp.test_route")
+    assert uri == "/bp/route"
+    request, response = app.test_client.websocket(uri)
+    assert response.opened is True
+    assert event.is_set()
+
+    event.clear()
+    uri2 = app.url_for("test_bp.test_route2")
+    assert uri2 == "/bp/route2"
+    request, response = app.test_client.websocket(uri2)
+    assert response.opened is True
+    assert event.is_set()