#![recursion_limit = "256"]
#![cfg(feature = "rustls")]
use std::sync::{Arc, atomic::AtomicUsize, atomic::Ordering};
use ntex::client::Client;
use ntex::http::{self, HttpService, Version, test::server as test_server};
use ntex::web::{self, App, HttpResponse};
use ntex::{SharedCfg, service};
mod rustls_utils;
#[ntex::test]
async fn test_connection_reuse_h2() {
let num = Arc::new(AtomicUsize::new(0));
let num2 = num.clone();
let srv = test_server(async move |_| {
let num2 = num2.clone();
service(async move |io| {
num2.fetch_add(1, Ordering::Relaxed);
Ok(io)
})
.and_then(http::rustls(
rustls_utils::tls_acceptor(),
http::ALPN_PROTO_H2,
HttpService::h2(
App::new().service(web::resource("/").route(web::to(async || HttpResponse::Ok()))),
),
))
});
let mut config = rustls_utils::tls_connector();
config.alpn_protocols = vec![b"h2".to_vec(), b"http/1.1".to_vec()];
let client = Client::builder().rustls(config).build(SharedCfg::default());
let response = client.get(srv.surl("/")).send().await.unwrap();
assert!(response.status().is_success());
let response = client.post(srv.surl("/")).send().await.unwrap();
assert!(response.status().is_success());
assert_eq!(response.version(), Version::HTTP_2);
assert_eq!(num.load(Ordering::Relaxed), 1);
}