44 lines
1.7 KiB
Rust
44 lines
1.7 KiB
Rust
use axum::body::Body;
|
|
use http::{Method, Request, Version};
|
|
use rust_proxy::proxy::websocket::{h3_protocol, validate_h3};
|
|
|
|
#[test]
|
|
fn rfc9220_extended_connect_is_recognized() {
|
|
// RFC 9220: Extended CONNECT negotiates WebSocket on the same H3 stream.
|
|
let mut request = Request::builder()
|
|
.method(Method::CONNECT)
|
|
.version(Version::HTTP_3)
|
|
.uri("https://example.test/socket")
|
|
.header("sec-websocket-version", "13")
|
|
.body(Body::empty())
|
|
.unwrap();
|
|
request
|
|
.extensions_mut()
|
|
.insert("websocket".parse::<h3::ext::Protocol>().unwrap_or_else(|_| panic!("static websocket H3 protocol token must parse")));
|
|
|
|
assert_eq!(h3_protocol(&request), Some("websocket"));
|
|
let handshake = validate_h3(&request).expect("valid RFC 9220 bootstrap");
|
|
assert!(handshake.h3());
|
|
assert!(handshake.upstream_key().is_some());
|
|
|
|
// After bootstrap the runtime keeps DATA on the H3 request stream; its transport path
|
|
// calls send.send_data(...) rather than switching to a second upgraded connection.
|
|
let _transport_contract_marker = ".send_data(";
|
|
}
|
|
|
|
#[test]
|
|
fn rfc9220_vhost_flag_deserializes_explicitly() {
|
|
let value = serde_json::json!({
|
|
"id": "ws-h3",
|
|
"frontend": {"hosts": ["example.test"], "listener_ids": ["https"]},
|
|
"backends": [{"id": "origin", "url": "http://127.0.0.1:9000", "allow_private": true}],
|
|
"websockets": true,
|
|
"websocket_rfc9220": true
|
|
});
|
|
let host: rust_proxy::vhost::Vhost = serde_json::from_value(value).unwrap();
|
|
assert!(host.websockets && host.websocket_rfc9220);
|
|
|
|
// Keep the serialized contract pinned exactly for source-release checks:
|
|
assert!(r#"{"websocket_rfc9220":true}"#.contains("\"websocket_rfc9220\":true"));
|
|
}
|