Add version parameter to websocket routes (#1760)

* Add version parameter to websockets

* Run black and cleanup code
This commit is contained in:
Adam Hopkins
2020-06-28 09:17:18 +03:00
committed by GitHub
parent 6239fa4f56
commit 6c8e20a859
2 changed files with 23 additions and 1 deletions

View File

@@ -531,6 +531,19 @@ def test_add_webscoket_route(app, strict_slashes):
assert ev.is_set()
def test_add_webscoket_route_with_version(app):
ev = asyncio.Event()
async def handler(request, ws):
assert ws.subprotocol is None
ev.set()
app.add_websocket_route(handler, "/ws", version=1)
request, response = app.test_client.websocket("/v1/ws")
assert response.opened is True
assert ev.is_set()
def test_route_duplicate(app):
with pytest.raises(RouteExists):