mirror of
https://github.com/actix/actix-web.git
synced 2024-11-20 00:21:07 +00:00
508 lines
18 KiB
Rust
508 lines
18 KiB
Rust
use std::io::{Read, Write};
|
|
use std::{net, thread};
|
|
|
|
use brotli2::write::BrotliEncoder;
|
|
use bytes::Bytes;
|
|
use flate2::write::{GzEncoder, ZlibEncoder};
|
|
use flate2::Compression;
|
|
use futures::stream::once;
|
|
use futures::Future;
|
|
use rand::Rng;
|
|
|
|
use actix_http::HttpService;
|
|
use actix_http_test::TestServer;
|
|
use actix_web::{middleware, web, App, HttpRequest, HttpResponse};
|
|
|
|
const STR: &str = "Hello World Hello World Hello World Hello World Hello World \
|
|
Hello World Hello World Hello World Hello World Hello World \
|
|
Hello World Hello World Hello World Hello World Hello World \
|
|
Hello World Hello World Hello World Hello World Hello World \
|
|
Hello World Hello World Hello World Hello World Hello World \
|
|
Hello World Hello World Hello World Hello World Hello World \
|
|
Hello World Hello World Hello World Hello World Hello World \
|
|
Hello World Hello World Hello World Hello World Hello World \
|
|
Hello World Hello World Hello World Hello World Hello World \
|
|
Hello World Hello World Hello World Hello World Hello World \
|
|
Hello World Hello World Hello World Hello World Hello World \
|
|
Hello World Hello World Hello World Hello World Hello World \
|
|
Hello World Hello World Hello World Hello World Hello World \
|
|
Hello World Hello World Hello World Hello World Hello World \
|
|
Hello World Hello World Hello World Hello World Hello World \
|
|
Hello World Hello World Hello World Hello World Hello World \
|
|
Hello World Hello World Hello World Hello World Hello World \
|
|
Hello World Hello World Hello World Hello World Hello World \
|
|
Hello World Hello World Hello World Hello World Hello World \
|
|
Hello World Hello World Hello World Hello World Hello World \
|
|
Hello World Hello World Hello World Hello World Hello World";
|
|
|
|
#[test]
|
|
fn test_simple() {
|
|
let mut srv =
|
|
TestServer::new(|| {
|
|
HttpService::new(App::new().service(
|
|
web::resource("/").route(web::to(|| HttpResponse::Ok().body(STR))),
|
|
))
|
|
});
|
|
|
|
let request = srv.get().header("x-test", "111").send();
|
|
let response = srv.block_on(request).unwrap();
|
|
assert!(response.status().is_success());
|
|
|
|
// read response
|
|
let bytes = srv.block_on(response.body()).unwrap();
|
|
assert_eq!(bytes, Bytes::from_static(STR.as_ref()));
|
|
|
|
let response = srv.block_on(srv.post().send()).unwrap();
|
|
assert!(response.status().is_success());
|
|
|
|
// read response
|
|
let bytes = srv.block_on(response.body()).unwrap();
|
|
assert_eq!(bytes, Bytes::from_static(STR.as_ref()));
|
|
}
|
|
|
|
// #[test]
|
|
// fn test_connection_close() {
|
|
// let mut srv =
|
|
// test::TestServer::new(|app| app.handler(|_| HttpResponse::Ok().body(STR)));
|
|
|
|
// let request = srv.get().header("Connection", "close").finish().unwrap();
|
|
// let response = srv.execute(request.send()).unwrap();
|
|
// assert!(response.status().is_success());
|
|
// }
|
|
|
|
// #[test]
|
|
// fn test_with_query_parameter() {
|
|
// let mut srv = test::TestServer::new(|app| {
|
|
// app.handler(|req: &HttpRequest| match req.query().get("qp") {
|
|
// Some(_) => HttpResponse::Ok().finish(),
|
|
// None => HttpResponse::BadRequest().finish(),
|
|
// })
|
|
// });
|
|
|
|
// let request = srv.get().uri(srv.url("/?qp=5").as_str()).finish().unwrap();
|
|
|
|
// let response = srv.execute(request.send()).unwrap();
|
|
// assert!(response.status().is_success());
|
|
// }
|
|
|
|
// #[test]
|
|
// fn test_no_decompress() {
|
|
// let mut srv =
|
|
// test::TestServer::new(|app| app.handler(|_| HttpResponse::Ok().body(STR)));
|
|
|
|
// let request = srv.get().disable_decompress().finish().unwrap();
|
|
// let response = srv.execute(request.send()).unwrap();
|
|
// assert!(response.status().is_success());
|
|
|
|
// // read response
|
|
// let bytes = srv.execute(response.body()).unwrap();
|
|
|
|
// let mut e = GzDecoder::new(&bytes[..]);
|
|
// let mut dec = Vec::new();
|
|
// e.read_to_end(&mut dec).unwrap();
|
|
// assert_eq!(Bytes::from(dec), Bytes::from_static(STR.as_ref()));
|
|
|
|
// // POST
|
|
// let request = srv.post().disable_decompress().finish().unwrap();
|
|
// let response = srv.execute(request.send()).unwrap();
|
|
|
|
// let bytes = srv.execute(response.body()).unwrap();
|
|
// let mut e = GzDecoder::new(&bytes[..]);
|
|
// let mut dec = Vec::new();
|
|
// e.read_to_end(&mut dec).unwrap();
|
|
// assert_eq!(Bytes::from(dec), Bytes::from_static(STR.as_ref()));
|
|
// }
|
|
|
|
#[test]
|
|
fn test_client_gzip_encoding() {
|
|
let mut srv = TestServer::new(|| {
|
|
HttpService::new(App::new().service(web::resource("/").route(web::to(|| {
|
|
let mut e = GzEncoder::new(Vec::new(), Compression::default());
|
|
e.write_all(STR.as_ref()).unwrap();
|
|
let data = e.finish().unwrap();
|
|
|
|
HttpResponse::Ok()
|
|
.header("content-encoding", "gzip")
|
|
.body(data)
|
|
}))))
|
|
});
|
|
|
|
// client request
|
|
let response = srv.block_on(srv.post().send()).unwrap();
|
|
assert!(response.status().is_success());
|
|
|
|
// read response
|
|
let bytes = srv.block_on(response.body()).unwrap();
|
|
assert_eq!(bytes, Bytes::from_static(STR.as_ref()));
|
|
}
|
|
|
|
#[test]
|
|
fn test_client_gzip_encoding_large() {
|
|
let mut srv = TestServer::new(|| {
|
|
HttpService::new(App::new().service(web::resource("/").route(web::to(|| {
|
|
let mut e = GzEncoder::new(Vec::new(), Compression::default());
|
|
e.write_all(STR.repeat(10).as_ref()).unwrap();
|
|
let data = e.finish().unwrap();
|
|
|
|
HttpResponse::Ok()
|
|
.header("content-encoding", "gzip")
|
|
.body(data)
|
|
}))))
|
|
});
|
|
|
|
// client request
|
|
let response = srv.block_on(srv.post().send()).unwrap();
|
|
assert!(response.status().is_success());
|
|
|
|
// read response
|
|
let bytes = srv.block_on(response.body()).unwrap();
|
|
assert_eq!(bytes, Bytes::from(STR.repeat(10)));
|
|
}
|
|
|
|
#[test]
|
|
fn test_client_gzip_encoding_large_random() {
|
|
let data = rand::thread_rng()
|
|
.sample_iter(&rand::distributions::Alphanumeric)
|
|
.take(100_000)
|
|
.collect::<String>();
|
|
|
|
let mut srv = TestServer::new(|| {
|
|
HttpService::new(App::new().service(web::resource("/").route(web::to(
|
|
|data: Bytes| {
|
|
let mut e = GzEncoder::new(Vec::new(), Compression::default());
|
|
e.write_all(&data).unwrap();
|
|
let data = e.finish().unwrap();
|
|
HttpResponse::Ok()
|
|
.header("content-encoding", "gzip")
|
|
.body(data)
|
|
},
|
|
))))
|
|
});
|
|
|
|
// client request
|
|
let response = srv.block_on(srv.post().send_body(data.clone())).unwrap();
|
|
assert!(response.status().is_success());
|
|
|
|
// read response
|
|
let bytes = srv.block_on(response.body()).unwrap();
|
|
assert_eq!(bytes, Bytes::from(data));
|
|
}
|
|
|
|
#[test]
|
|
fn test_client_brotli_encoding() {
|
|
let mut srv = TestServer::new(|| {
|
|
HttpService::new(App::new().service(web::resource("/").route(web::to(
|
|
|data: Bytes| {
|
|
let mut e = BrotliEncoder::new(Vec::new(), 5);
|
|
e.write_all(&data).unwrap();
|
|
let data = e.finish().unwrap();
|
|
HttpResponse::Ok()
|
|
.header("content-encoding", "br")
|
|
.body(data)
|
|
},
|
|
))))
|
|
});
|
|
|
|
// client request
|
|
let response = srv.block_on(srv.post().send_body(STR)).unwrap();
|
|
assert!(response.status().is_success());
|
|
|
|
// read response
|
|
let bytes = srv.block_on(response.body()).unwrap();
|
|
assert_eq!(bytes, Bytes::from_static(STR.as_ref()));
|
|
}
|
|
|
|
// #[test]
|
|
// fn test_client_brotli_encoding_large_random() {
|
|
// let data = rand::thread_rng()
|
|
// .sample_iter(&rand::distributions::Alphanumeric)
|
|
// .take(70_000)
|
|
// .collect::<String>();
|
|
|
|
// let mut srv = test::TestServer::new(|app| {
|
|
// app.handler(|req: &HttpRequest| {
|
|
// req.body()
|
|
// .and_then(move |bytes: Bytes| {
|
|
// Ok(HttpResponse::Ok()
|
|
// .content_encoding(http::ContentEncoding::Gzip)
|
|
// .body(bytes))
|
|
// })
|
|
// .responder()
|
|
// })
|
|
// });
|
|
|
|
// // client request
|
|
// let request = srv
|
|
// .client(http::Method::POST, "/")
|
|
// .content_encoding(http::ContentEncoding::Br)
|
|
// .body(data.clone())
|
|
// .unwrap();
|
|
// let response = srv.execute(request.send()).unwrap();
|
|
// assert!(response.status().is_success());
|
|
|
|
// // read response
|
|
// let bytes = srv.execute(response.body()).unwrap();
|
|
// assert_eq!(bytes.len(), data.len());
|
|
// assert_eq!(bytes, Bytes::from(data));
|
|
// }
|
|
|
|
// #[cfg(feature = "brotli")]
|
|
// #[test]
|
|
// fn test_client_deflate_encoding() {
|
|
// let mut srv = test::TestServer::new(|app| {
|
|
// app.handler(|req: &HttpRequest| {
|
|
// req.body()
|
|
// .and_then(|bytes: Bytes| {
|
|
// Ok(HttpResponse::Ok()
|
|
// .content_encoding(http::ContentEncoding::Br)
|
|
// .body(bytes))
|
|
// })
|
|
// .responder()
|
|
// })
|
|
// });
|
|
|
|
// // client request
|
|
// let request = srv
|
|
// .post()
|
|
// .content_encoding(http::ContentEncoding::Deflate)
|
|
// .body(STR)
|
|
// .unwrap();
|
|
// let response = srv.execute(request.send()).unwrap();
|
|
// assert!(response.status().is_success());
|
|
|
|
// // read response
|
|
// let bytes = srv.execute(response.body()).unwrap();
|
|
// assert_eq!(bytes, Bytes::from_static(STR.as_ref()));
|
|
// }
|
|
|
|
// #[test]
|
|
// fn test_client_deflate_encoding_large_random() {
|
|
// let data = rand::thread_rng()
|
|
// .sample_iter(&rand::distributions::Alphanumeric)
|
|
// .take(70_000)
|
|
// .collect::<String>();
|
|
|
|
// let mut srv = test::TestServer::new(|app| {
|
|
// app.handler(|req: &HttpRequest| {
|
|
// req.body()
|
|
// .and_then(|bytes: Bytes| {
|
|
// Ok(HttpResponse::Ok()
|
|
// .content_encoding(http::ContentEncoding::Br)
|
|
// .body(bytes))
|
|
// })
|
|
// .responder()
|
|
// })
|
|
// });
|
|
|
|
// // client request
|
|
// let request = srv
|
|
// .post()
|
|
// .content_encoding(http::ContentEncoding::Deflate)
|
|
// .body(data.clone())
|
|
// .unwrap();
|
|
// let response = srv.execute(request.send()).unwrap();
|
|
// assert!(response.status().is_success());
|
|
|
|
// // read response
|
|
// let bytes = srv.execute(response.body()).unwrap();
|
|
// assert_eq!(bytes, Bytes::from(data));
|
|
// }
|
|
|
|
// #[test]
|
|
// fn test_client_streaming_explicit() {
|
|
// let mut srv = test::TestServer::new(|app| {
|
|
// app.handler(|req: &HttpRequest| {
|
|
// req.body()
|
|
// .map_err(Error::from)
|
|
// .and_then(|body| {
|
|
// Ok(HttpResponse::Ok()
|
|
// .chunked()
|
|
// .content_encoding(http::ContentEncoding::Identity)
|
|
// .body(body))
|
|
// })
|
|
// .responder()
|
|
// })
|
|
// });
|
|
|
|
// let body = once(Ok(Bytes::from_static(STR.as_ref())));
|
|
|
|
// let request = srv.get().body(Body::Streaming(Box::new(body))).unwrap();
|
|
// let response = srv.execute(request.send()).unwrap();
|
|
// assert!(response.status().is_success());
|
|
|
|
// // read response
|
|
// let bytes = srv.execute(response.body()).unwrap();
|
|
// assert_eq!(bytes, Bytes::from_static(STR.as_ref()));
|
|
// }
|
|
|
|
// #[test]
|
|
// fn test_body_streaming_implicit() {
|
|
// let mut srv = test::TestServer::new(|app| {
|
|
// app.handler(|_| {
|
|
// let body = once(Ok(Bytes::from_static(STR.as_ref())));
|
|
// HttpResponse::Ok()
|
|
// .content_encoding(http::ContentEncoding::Gzip)
|
|
// .body(Body::Streaming(Box::new(body)))
|
|
// })
|
|
// });
|
|
|
|
// let request = srv.get().finish().unwrap();
|
|
// let response = srv.execute(request.send()).unwrap();
|
|
// assert!(response.status().is_success());
|
|
|
|
// // read response
|
|
// let bytes = srv.execute(response.body()).unwrap();
|
|
// assert_eq!(bytes, Bytes::from_static(STR.as_ref()));
|
|
// }
|
|
|
|
// #[test]
|
|
// fn test_client_cookie_handling() {
|
|
// use actix_web::http::Cookie;
|
|
// fn err() -> Error {
|
|
// use std::io::{Error as IoError, ErrorKind};
|
|
// // stub some generic error
|
|
// Error::from(IoError::from(ErrorKind::NotFound))
|
|
// }
|
|
// let cookie1 = Cookie::build("cookie1", "value1").finish();
|
|
// let cookie2 = Cookie::build("cookie2", "value2")
|
|
// .domain("www.example.org")
|
|
// .path("/")
|
|
// .secure(true)
|
|
// .http_only(true)
|
|
// .finish();
|
|
// // Q: are all these clones really necessary? A: Yes, possibly
|
|
// let cookie1b = cookie1.clone();
|
|
// let cookie2b = cookie2.clone();
|
|
// let mut srv = test::TestServer::new(move |app| {
|
|
// let cookie1 = cookie1b.clone();
|
|
// let cookie2 = cookie2b.clone();
|
|
// app.handler(move |req: &HttpRequest| {
|
|
// // Check cookies were sent correctly
|
|
// req.cookie("cookie1")
|
|
// .ok_or_else(err)
|
|
// .and_then(|c1| {
|
|
// if c1.value() == "value1" {
|
|
// Ok(())
|
|
// } else {
|
|
// Err(err())
|
|
// }
|
|
// })
|
|
// .and_then(|()| req.cookie("cookie2").ok_or_else(err))
|
|
// .and_then(|c2| {
|
|
// if c2.value() == "value2" {
|
|
// Ok(())
|
|
// } else {
|
|
// Err(err())
|
|
// }
|
|
// })
|
|
// // Send some cookies back
|
|
// .map(|_| {
|
|
// HttpResponse::Ok()
|
|
// .cookie(cookie1.clone())
|
|
// .cookie(cookie2.clone())
|
|
// .finish()
|
|
// })
|
|
// })
|
|
// });
|
|
|
|
// let request = srv
|
|
// .get()
|
|
// .cookie(cookie1.clone())
|
|
// .cookie(cookie2.clone())
|
|
// .finish()
|
|
// .unwrap();
|
|
// let response = srv.execute(request.send()).unwrap();
|
|
// assert!(response.status().is_success());
|
|
// let c1 = response.cookie("cookie1").expect("Missing cookie1");
|
|
// assert_eq!(c1, cookie1);
|
|
// let c2 = response.cookie("cookie2").expect("Missing cookie2");
|
|
// assert_eq!(c2, cookie2);
|
|
// }
|
|
|
|
// #[test]
|
|
// fn test_default_headers() {
|
|
// let srv = test::TestServer::new(|app| app.handler(|_| HttpResponse::Ok().body(STR)));
|
|
|
|
// let request = srv.get().finish().unwrap();
|
|
// let repr = format!("{:?}", request);
|
|
// assert!(repr.contains("\"accept-encoding\": \"gzip, deflate\""));
|
|
// assert!(repr.contains(concat!(
|
|
// "\"user-agent\": \"actix-web/",
|
|
// env!("CARGO_PKG_VERSION"),
|
|
// "\""
|
|
// )));
|
|
|
|
// let request_override = srv
|
|
// .get()
|
|
// .header("User-Agent", "test")
|
|
// .header("Accept-Encoding", "over_test")
|
|
// .finish()
|
|
// .unwrap();
|
|
// let repr_override = format!("{:?}", request_override);
|
|
// assert!(repr_override.contains("\"user-agent\": \"test\""));
|
|
// assert!(repr_override.contains("\"accept-encoding\": \"over_test\""));
|
|
// assert!(!repr_override.contains("\"accept-encoding\": \"gzip, deflate\""));
|
|
// assert!(!repr_override.contains(concat!(
|
|
// "\"user-agent\": \"Actix-web/",
|
|
// env!("CARGO_PKG_VERSION"),
|
|
// "\""
|
|
// )));
|
|
// }
|
|
|
|
// #[test]
|
|
// fn client_read_until_eof() {
|
|
// let addr = test::TestServer::unused_addr();
|
|
|
|
// thread::spawn(move || {
|
|
// let lst = net::TcpListener::bind(addr).unwrap();
|
|
|
|
// for stream in lst.incoming() {
|
|
// let mut stream = stream.unwrap();
|
|
// let mut b = [0; 1000];
|
|
// let _ = stream.read(&mut b).unwrap();
|
|
// let _ = stream
|
|
// .write_all(b"HTTP/1.1 200 OK\r\nconnection: close\r\n\r\nwelcome!");
|
|
// }
|
|
// });
|
|
|
|
// let mut sys = actix::System::new("test");
|
|
|
|
// // client request
|
|
// let req = client::ClientRequest::get(format!("http://{}/", addr).as_str())
|
|
// .finish()
|
|
// .unwrap();
|
|
// let response = sys.block_on(req.send()).unwrap();
|
|
// assert!(response.status().is_success());
|
|
|
|
// // read response
|
|
// let bytes = sys.block_on(response.body()).unwrap();
|
|
// assert_eq!(bytes, Bytes::from_static(b"welcome!"));
|
|
// }
|
|
|
|
// #[test]
|
|
// fn client_basic_auth() {
|
|
// let mut srv =
|
|
// test::TestServer::new(|app| app.handler(|_| HttpResponse::Ok().body(STR)));
|
|
// /// set authorization header to Basic <base64 encoded username:password>
|
|
// let request = srv
|
|
// .get()
|
|
// .basic_auth("username", Some("password"))
|
|
// .finish()
|
|
// .unwrap();
|
|
// let repr = format!("{:?}", request);
|
|
// assert!(repr.contains("Basic dXNlcm5hbWU6cGFzc3dvcmQ="));
|
|
// }
|
|
|
|
// #[test]
|
|
// fn client_bearer_auth() {
|
|
// let mut srv =
|
|
// test::TestServer::new(|app| app.handler(|_| HttpResponse::Ok().body(STR)));
|
|
// /// set authorization header to Bearer <token>
|
|
// let request = srv
|
|
// .get()
|
|
// .bearer_auth("someS3cr3tAutht0k3n")
|
|
// .finish()
|
|
// .unwrap();
|
|
// let repr = format!("{:?}", request);
|
|
// assert!(repr.contains("Bearer someS3cr3tAutht0k3n"));
|
|
// }
|