Function _build_websocket_request

python/vercel-runtime/tests/test_runtime.py:312–331

(
    port: int,
    path: str = "/ws",
    headers: dict[str, str] | None = None,
)

def _build_websocket_request(
    port: int,
    path: str = "/ws",
    headers: dict[str, str] | None = None,
) -> bytes:
    key = base64.b64encode(os.urandom(16)).decode("ascii")
    request_headers = {
        "Host": f"127.0.0.1:{port}",
        "Upgrade": "websocket",
        "Connection": "Upgrade",
        "Sec-WebSocket-Key": key,
        "Sec-WebSocket-Version": "13",
        **(headers or {}),
    }

    request_lines = [f"GET {path} HTTP/1.1"]
    request_lines.extend(
        f"{name}: {value}" for name, value in request_headers.items()
    )
    return ("\r\n".join([*request_lines, "", ""])).encode("ascii")


class _WebSocketClient: