mirror of https://github.com/fafhrd91/actix-web
114 lines
3.9 KiB
Rust
114 lines
3.9 KiB
Rust
#![cfg(feature = "openssl")]
|
|
use actix_http::HttpService;
|
|
use actix_http_test::test_server;
|
|
use actix_service::{map_config, ServiceFactory};
|
|
use actix_web::http::Version;
|
|
use actix_web::{dev::AppConfig, test, web, App, HttpResponse};
|
|
use awc::http::StatusCode;
|
|
use bytes::Bytes;
|
|
use open_ssl::ssl::{SslAcceptor, SslConnector, SslFiletype, SslMethod, SslVerifyMode};
|
|
|
|
fn ssl_acceptor() -> SslAcceptor {
|
|
// load ssl keys
|
|
let mut builder = SslAcceptor::mozilla_intermediate(SslMethod::tls()).unwrap();
|
|
builder
|
|
.set_private_key_file("../tests/key.pem", SslFiletype::PEM)
|
|
.unwrap();
|
|
builder
|
|
.set_certificate_chain_file("../tests/cert.pem")
|
|
.unwrap();
|
|
builder.set_alpn_select_callback(|_, protos| {
|
|
const H2: &[u8] = b"\x02h2";
|
|
if protos.windows(3).any(|window| window == H2) {
|
|
Ok(b"h2")
|
|
} else {
|
|
Err(open_ssl::ssl::AlpnError::NOACK)
|
|
}
|
|
});
|
|
builder.set_alpn_protos(b"\x02h2").unwrap();
|
|
builder.build()
|
|
}
|
|
|
|
#[actix_rt::test]
|
|
async fn test_connection_window_size() {
|
|
let srv = test_server(move || {
|
|
HttpService::build()
|
|
.h2(map_config(
|
|
App::new().service(web::resource("/").route(web::to(HttpResponse::Ok))),
|
|
|_| AppConfig::default(),
|
|
))
|
|
.openssl(ssl_acceptor())
|
|
.map_err(|_| ())
|
|
})
|
|
.await;
|
|
|
|
// disable ssl verification
|
|
let mut builder = SslConnector::builder(SslMethod::tls()).unwrap();
|
|
builder.set_verify(SslVerifyMode::NONE);
|
|
let _ = builder
|
|
.set_alpn_protos(b"\x02h2\x08http/1.1")
|
|
.map_err(|e| log::error!("Can not set alpn protocol: {:?}", e));
|
|
|
|
let client = awc::Client::builder()
|
|
.connector(awc::Connector::new().ssl(builder.build()).finish())
|
|
.initial_window_size(100)
|
|
.initial_connection_window_size(100)
|
|
.finish();
|
|
|
|
let request = client.get(srv.surl("/")).send();
|
|
let response = request.await.unwrap();
|
|
assert!(response.status().is_success());
|
|
assert_eq!(response.version(), Version::HTTP_2);
|
|
}
|
|
|
|
#[actix_rt::test]
|
|
async fn test_follow_redirects() {
|
|
let srv = test::start(|| {
|
|
App::new()
|
|
.service(web::resource("/do/redirect").route(web::to(|| {
|
|
HttpResponse::TemporaryRedirect()
|
|
.header("Location", "get")
|
|
.finish()
|
|
})))
|
|
.service(web::resource("/do/get").route(web::to(HttpResponse::Ok)))
|
|
});
|
|
|
|
let client = awc::Client::builder().finish();
|
|
|
|
let request = client.get(srv.url("/do/redirect")).send();
|
|
let response = request.await.unwrap();
|
|
assert!(response.status().is_success());
|
|
assert_eq!(response.status(), StatusCode::OK);
|
|
}
|
|
|
|
#[actix_rt::test]
|
|
async fn test_max_redirects() {
|
|
let srv = test::start(|| {
|
|
App::new()
|
|
.service(web::resource("/first-redirect").route(web::to(|| {
|
|
HttpResponse::TemporaryRedirect()
|
|
.header("Location", "/second-redirect")
|
|
.body("first")
|
|
})))
|
|
.service(web::resource("/second-redirect").route(web::to(|| {
|
|
HttpResponse::TemporaryRedirect()
|
|
.header("Location", "/third-redirect")
|
|
.body("second")
|
|
})))
|
|
.service(web::resource("/third-redirect").route(web::to(|| {
|
|
HttpResponse::TemporaryRedirect()
|
|
.header("Location", "/the-content")
|
|
.body("third")
|
|
})))
|
|
.service(web::resource("/the-content").route(web::to(HttpResponse::Ok)))
|
|
});
|
|
|
|
let client = awc::Client::builder().max_redirects(2).finish();
|
|
|
|
let request = client.get(srv.url("/first-redirect")).send();
|
|
let mut response = request.await.unwrap();
|
|
assert!(response.status().is_redirection());
|
|
assert_eq!(response.status(), StatusCode::TEMPORARY_REDIRECT);
|
|
assert_eq!(response.body().await.unwrap(), Bytes::from("third"));
|
|
}
|