def test_handshake_accept_http1() -> None:
handshake = Handshake(
[
(b"connection", b"upgrade, keep-alive"),
(b"sec-websocket-version", b"13"),
(b"upgrade", b"websocket"),
(b"sec-websocket-key", b"UnQ3lpJAH6j2PslA993iKQ=="),
],
"1.1",
)
status_code, headers, _ = handshake.accept(None)
assert status_code == 101
assert headers == [
(b"sec-websocket-accept", b"1BpNk/3ah1huDGgcuMJBcjcMbEA="),
(b"upgrade", b"WebSocket"),
(b"connection", b"Upgrade"),
]
def test_handshake_accept_http2() -> None:
handshake = Handshake([(b"sec-websocket-version", b"13")], "2")
status_code, headers, _ = handshake.accept(None)
assert status_code == 200
assert headers == []
def test_handshake_validity(headers: List[Tuple[bytes, bytes]],
http_version: str, valid: bool) -> None:
handshake = Handshake(headers, http_version)
assert handshake.is_valid() is valid