test: have websockets failures report *which* one

Author: pajod

tests/test_web_websocket.py

@@ -215,6 +215,12 @@ def test_can_prepare_unknown_protocol(make_request: _RequestMaker) -> None:
     assert WebSocketReady(True, None) == ws.can_prepare(req)
 
 
+def test_can_prepare_invalid_method(make_request: _RequestMaker) -> None:
+    req = make_request("POST", "/")
+    ws = web.WebSocketResponse()
+    assert WebSocketReady(False, None) == ws.can_prepare(req)
+
+
 def test_can_prepare_without_upgrade(make_request: _RequestMaker) -> None:
     req = make_request("GET", "/", headers=CIMultiDict({}))
     ws = web.WebSocketResponse()
@@ -369,11 +375,11 @@ async def test_close_idempotent(make_request: _RequestMaker) -> None:
     assert close_code == 0
 
 
-async def test_prepare_post_method_ok(make_request: _RequestMaker) -> None:
+async def test_prepare_invalid_method(make_request: _RequestMaker) -> None:
     req = make_request("POST", "/")
     ws = web.WebSocketResponse()
-    await ws.prepare(req)
-    assert ws.prepared
+    with pytest.raises(web.HTTPMethodNotAllowed):
+        await ws.prepare(req)
 
 
 async def test_prepare_without_upgrade(make_request: _RequestMaker) -> None:

tests/test_websocket_handshake.py

@@ -40,29 +40,60 @@ def gen_ws_headers(
     return hdrs, key
 
 
+async def test_not_get() -> None:
+    ws = web.WebSocketResponse()
+    req = make_mocked_request("POST", "/")
+    with pytest.raises(web.HTTPMethodNotAllowed):
+        await ws.prepare(req)
+
+
+async def test_inappropriate_method() -> None:
+    ws = web.WebSocketResponse()
+    req = make_mocked_request(
+        "HEAD",
+        "/",
+        headers={
+            "Upgrade": "websocket",
+            "Expect": "100-continue",
+            "Expect": "100-continue",
+        },
+    )
+    with pytest.raises(web.HTTPMethodNotAllowed) as ctx:
+        await ws.prepare(req)
+    assert ctx.value.method == "HEAD"
+    assert ctx.value.allowed_methods == {"GET"}
+    assert ctx.value.status == 405
+
+
 async def test_no_upgrade() -> None:
     ws = web.WebSocketResponse()
     req = make_mocked_request("GET", "/")
-    with pytest.raises(web.HTTPBadRequest):
+    with pytest.raises(web.HTTPBadRequest) as ctx:
         await ws.prepare(req)
+    assert "UPGRADE" in ctx.value.text
+    assert ctx.value.status == 400
 
 
 async def test_no_connection() -> None:
     ws = web.WebSocketResponse()
     req = make_mocked_request(
         "GET", "/", headers={"Upgrade": "websocket", "Connection": "keep-alive"}
     )
-    with pytest.raises(web.HTTPBadRequest):
+    with pytest.raises(web.HTTPBadRequest) as ctx:
         await ws.prepare(req)
+    assert "CONNECTION" in ctx.value.text
+    assert ctx.value.status == 400
 
 
 async def test_protocol_version_unset() -> None:
     ws = web.WebSocketResponse()
     req = make_mocked_request(
         "GET", "/", headers={"Upgrade": "websocket", "Connection": "upgrade"}
     )
-    with pytest.raises(web.HTTPBadRequest):
+    with pytest.raises(web.HTTPBadRequest) as ctx:
         await ws.prepare(req)
+    assert "version" in ctx.value.text
+    assert ctx.value.status == 400
 
 
 async def test_protocol_version_not_supported() -> None:
@@ -76,8 +107,10 @@ async def test_protocol_version_not_supported() -> None:
             "Sec-Websocket-Version": "1",
         },
     )
-    with pytest.raises(web.HTTPBadRequest):
+    with pytest.raises(web.HTTPBadRequest) as ctx:
         await ws.prepare(req)
+    assert "version" in ctx.value.text
+    assert ctx.value.status == 400
 
 
 async def test_protocol_key_not_present() -> None:
@@ -91,8 +124,10 @@ async def test_protocol_key_not_present() -> None:
             "Sec-Websocket-Version": "13",
         },
     )
-    with pytest.raises(web.HTTPBadRequest):
+    with pytest.raises(web.HTTPBadRequest) as ctx:
         await ws.prepare(req)
+    assert "Handshake" in ctx.value.text
+    assert ctx.value.status == 400
 
 
 async def test_protocol_key_invalid() -> None:
@@ -107,8 +142,10 @@ async def test_protocol_key_invalid() -> None:
             "Sec-Websocket-Key": "123",
         },
     )
-    with pytest.raises(web.HTTPBadRequest):
+    with pytest.raises(web.HTTPBadRequest) as ctx:
         await ws.prepare(req)
+    assert "Handshake" in ctx.value.text
+    assert ctx.value.status == 400
 
 
 async def test_protocol_key_bad_size() -> None:
@@ -125,8 +162,10 @@ async def test_protocol_key_bad_size() -> None:
             "Sec-Websocket-Key": val,
         },
     )
-    with pytest.raises(web.HTTPBadRequest):
+    with pytest.raises(web.HTTPBadRequest) as ctx:
         await ws.prepare(req)
+    assert "Handshake" in ctx.value.text
+    assert ctx.value.status == 400
 
 
 async def test_handshake_ok() -> None: