1
0
Fork 0
mirror of https://github.com/actix/actix-web.git synced 2024-06-02 21:39:26 +00:00
actix-web/tests/test_server.rs

893 lines
28 KiB
Rust
Raw Normal View History

2020-05-18 02:47:20 +00:00
use std::future::Future;
2018-05-29 17:31:37 +00:00
use std::io::{Read, Write};
2019-12-20 07:50:07 +00:00
use std::pin::Pin;
use std::task::{Context, Poll};
2019-03-02 06:51:32 +00:00
use actix_http::http::header::{
2019-03-26 22:14:32 +00:00
ContentEncoding, ACCEPT_ENCODING, CONTENT_ENCODING, CONTENT_LENGTH,
TRANSFER_ENCODING,
2019-03-02 06:51:32 +00:00
};
2019-12-20 07:50:07 +00:00
use brotli2::write::{BrotliDecoder, BrotliEncoder};
2019-03-02 06:51:32 +00:00
use bytes::Bytes;
2018-02-19 21:18:18 +00:00
use flate2::read::GzDecoder;
2019-03-26 22:14:32 +00:00
use flate2::write::{GzEncoder, ZlibDecoder, ZlibEncoder};
use flate2::Compression;
2020-05-18 02:47:20 +00:00
use futures_util::ready;
2019-03-02 06:51:32 +00:00
use rand::{distributions::Alphanumeric, Rng};
2019-12-18 03:30:14 +00:00
use actix_web::dev::BodyEncoding;
use actix_web::middleware::Compress;
use actix_web::{dev, test, web, App, Error, HttpResponse};
2018-02-19 21:18:18 +00:00
2018-04-13 23:02:01 +00:00
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";
2018-01-11 05:47:30 +00:00
2019-12-20 07:50:07 +00:00
struct TestBody {
data: Bytes,
chunk_size: usize,
delay: actix_rt::time::Delay,
}
impl TestBody {
fn new(data: Bytes, chunk_size: usize) -> Self {
TestBody {
data,
chunk_size,
delay: actix_rt::time::delay_for(std::time::Duration::from_millis(10)),
}
}
}
2020-05-18 02:47:20 +00:00
impl futures_core::stream::Stream for TestBody {
2019-12-20 07:50:07 +00:00
type Item = Result<Bytes, Error>;
fn poll_next(
mut self: Pin<&mut Self>,
cx: &mut Context<'_>,
) -> Poll<Option<Self::Item>> {
ready!(Pin::new(&mut self.delay).poll(cx));
self.delay = actix_rt::time::delay_for(std::time::Duration::from_millis(10));
let chunk_size = std::cmp::min(self.chunk_size, self.data.len());
let chunk = self.data.split_to(chunk_size);
if chunk.is_empty() {
Poll::Ready(None)
} else {
Poll::Ready(Some(Ok(chunk)))
}
}
}
2019-11-26 05:25:50 +00:00
#[actix_rt::test]
async fn test_body() {
2019-12-12 16:28:47 +00:00
let srv = test::start(|| {
App::new()
.service(web::resource("/").route(web::to(|| HttpResponse::Ok().body(STR))))
2019-11-26 05:25:50 +00:00
});
let mut response = srv.get("/").send().await.unwrap();
assert!(response.status().is_success());
// read response
let bytes = response.body().await.unwrap();
assert_eq!(bytes, Bytes::from_static(STR.as_ref()));
2018-01-11 06:42:26 +00:00
}
2019-11-26 05:25:50 +00:00
#[actix_rt::test]
async fn test_body_gzip() {
2019-12-12 16:28:47 +00:00
let srv = test::start_with(test::config().h1(), || {
App::new()
.wrap(Compress::new(ContentEncoding::Gzip))
.service(web::resource("/").route(web::to(|| HttpResponse::Ok().body(STR))))
2019-11-26 05:25:50 +00:00
});
let mut response = srv
.get("/")
.no_decompress()
.header(ACCEPT_ENCODING, "gzip")
.send()
.await
.unwrap();
assert!(response.status().is_success());
// read response
let bytes = response.body().await.unwrap();
// decode
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()));
2019-04-18 00:48:25 +00:00
}
2019-11-26 05:25:50 +00:00
#[actix_rt::test]
async fn test_body_gzip2() {
2019-12-12 16:28:47 +00:00
let srv = test::start_with(test::config().h1(), || {
App::new()
.wrap(Compress::new(ContentEncoding::Gzip))
.service(web::resource("/").route(web::to(|| {
HttpResponse::Ok().body(STR).into_body::<dev::Body>()
})))
2019-11-26 05:25:50 +00:00
});
let mut response = srv
.get("/")
.no_decompress()
.header(ACCEPT_ENCODING, "gzip")
.send()
.await
.unwrap();
assert!(response.status().is_success());
// read response
let bytes = response.body().await.unwrap();
// decode
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()));
2018-01-11 05:47:30 +00:00
}
2019-11-26 05:25:50 +00:00
#[actix_rt::test]
async fn test_body_encoding_override() {
2019-12-12 16:28:47 +00:00
let srv = test::start_with(test::config().h1(), || {
App::new()
.wrap(Compress::new(ContentEncoding::Gzip))
.service(web::resource("/").route(web::to(|| {
HttpResponse::Ok()
.encoding(ContentEncoding::Deflate)
.body(STR)
})))
.service(web::resource("/raw").route(web::to(|| {
let body = actix_web::dev::Body::Bytes(STR.into());
let mut response =
HttpResponse::with_body(actix_web::http::StatusCode::OK, body);
response.encoding(ContentEncoding::Deflate);
response
})))
2019-11-26 05:25:50 +00:00
});
// Builder
let mut response = srv
.get("/")
.no_decompress()
.header(ACCEPT_ENCODING, "deflate")
.send()
.await
.unwrap();
assert!(response.status().is_success());
// read response
let bytes = response.body().await.unwrap();
// decode
let mut e = ZlibDecoder::new(Vec::new());
e.write_all(bytes.as_ref()).unwrap();
let dec = e.finish().unwrap();
assert_eq!(Bytes::from(dec), Bytes::from_static(STR.as_ref()));
// Raw Response
let mut response = srv
.request(actix_web::http::Method::GET, srv.url("/raw"))
.no_decompress()
.header(ACCEPT_ENCODING, "deflate")
.send()
.await
.unwrap();
assert!(response.status().is_success());
// read response
let bytes = response.body().await.unwrap();
// decode
let mut e = ZlibDecoder::new(Vec::new());
e.write_all(bytes.as_ref()).unwrap();
let dec = e.finish().unwrap();
assert_eq!(Bytes::from(dec), Bytes::from_static(STR.as_ref()));
}
2019-11-26 05:25:50 +00:00
#[actix_rt::test]
async fn test_body_gzip_large() {
let data = STR.repeat(10);
let srv_data = data.clone();
2019-12-12 16:28:47 +00:00
let srv = test::start_with(test::config().h1(), move || {
2019-11-26 05:25:50 +00:00
let data = srv_data.clone();
2019-12-12 16:28:47 +00:00
App::new()
.wrap(Compress::new(ContentEncoding::Gzip))
.service(
web::resource("/")
.route(web::to(move || HttpResponse::Ok().body(data.clone()))),
)
2019-11-26 05:25:50 +00:00
});
let mut response = srv
.get("/")
.no_decompress()
.header(ACCEPT_ENCODING, "gzip")
.send()
.await
.unwrap();
assert!(response.status().is_success());
// read response
let bytes = response.body().await.unwrap();
// decode
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(data));
}
2019-11-26 05:25:50 +00:00
#[actix_rt::test]
async fn test_body_gzip_large_random() {
let data = rand::thread_rng()
.sample_iter(&Alphanumeric)
.take(70_000)
.collect::<String>();
let srv_data = data.clone();
2019-12-12 16:28:47 +00:00
let srv = test::start_with(test::config().h1(), move || {
2019-11-26 05:25:50 +00:00
let data = srv_data.clone();
2019-12-12 16:28:47 +00:00
App::new()
.wrap(Compress::new(ContentEncoding::Gzip))
.service(
web::resource("/")
.route(web::to(move || HttpResponse::Ok().body(data.clone()))),
)
2019-11-26 05:25:50 +00:00
});
let mut response = srv
.get("/")
.no_decompress()
.header(ACCEPT_ENCODING, "gzip")
.send()
.await
.unwrap();
assert!(response.status().is_success());
// read response
let bytes = response.body().await.unwrap();
// decode
let mut e = GzDecoder::new(&bytes[..]);
let mut dec = Vec::new();
e.read_to_end(&mut dec).unwrap();
assert_eq!(dec.len(), data.len());
assert_eq!(Bytes::from(dec), Bytes::from(data));
}
2019-11-26 05:25:50 +00:00
#[actix_rt::test]
async fn test_body_chunked_implicit() {
2019-12-12 16:28:47 +00:00
let srv = test::start_with(test::config().h1(), || {
App::new()
.wrap(Compress::new(ContentEncoding::Gzip))
.service(web::resource("/").route(web::get().to(move || {
HttpResponse::Ok()
2019-12-20 07:50:07 +00:00
.streaming(TestBody::new(Bytes::from_static(STR.as_ref()), 24))
2019-12-12 16:28:47 +00:00
})))
2019-11-26 05:25:50 +00:00
});
let mut response = srv
.get("/")
.no_decompress()
.header(ACCEPT_ENCODING, "gzip")
.send()
.await
.unwrap();
assert!(response.status().is_success());
assert_eq!(
response.headers().get(TRANSFER_ENCODING).unwrap(),
&b"chunked"[..]
);
// read response
let bytes = response.body().await.unwrap();
// decode
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()));
2018-01-11 23:26:46 +00:00
}
2019-11-26 05:25:50 +00:00
#[actix_rt::test]
async fn test_body_br_streaming() {
2019-12-12 16:28:47 +00:00
let srv = test::start_with(test::config().h1(), || {
App::new().wrap(Compress::new(ContentEncoding::Br)).service(
web::resource("/").route(web::to(move || {
HttpResponse::Ok()
2019-12-20 07:50:07 +00:00
.streaming(TestBody::new(Bytes::from_static(STR.as_ref()), 24))
2019-12-12 16:28:47 +00:00
})),
)
2019-11-26 05:25:50 +00:00
});
let mut response = srv
.get("/")
.header(ACCEPT_ENCODING, "br")
.no_decompress()
.send()
.await
.unwrap();
assert!(response.status().is_success());
// read response
let bytes = response.body().await.unwrap();
2019-12-20 07:50:07 +00:00
println!("TEST: {:?}", bytes.len());
2019-11-26 05:25:50 +00:00
// decode br
2019-12-20 07:50:07 +00:00
let mut e = BrotliDecoder::new(Vec::with_capacity(2048));
2019-11-26 05:25:50 +00:00
e.write_all(bytes.as_ref()).unwrap();
2019-12-20 07:50:07 +00:00
let dec = e.finish().unwrap();
println!("T: {:?}", Bytes::copy_from_slice(&dec));
2019-11-26 05:25:50 +00:00
assert_eq!(Bytes::from(dec), Bytes::from_static(STR.as_ref()));
2018-01-12 00:22:27 +00:00
}
2018-01-21 00:12:38 +00:00
2019-11-26 05:25:50 +00:00
#[actix_rt::test]
async fn test_head_binary() {
2019-12-12 16:28:47 +00:00
let srv = test::start_with(test::config().h1(), || {
2020-05-21 08:56:53 +00:00
App::new().service(
web::resource("/")
.route(web::head().to(move || HttpResponse::Ok().body(STR))),
)
2019-11-26 05:25:50 +00:00
});
let mut response = srv.head("/").send().await.unwrap();
assert!(response.status().is_success());
{
let len = response.headers().get(CONTENT_LENGTH).unwrap();
assert_eq!(format!("{}", STR.len()), len.to_str().unwrap());
}
// read response
let bytes = response.body().await.unwrap();
assert!(bytes.is_empty());
2018-01-21 00:12:38 +00:00
}
2019-11-26 05:25:50 +00:00
#[actix_rt::test]
async fn test_no_chunking() {
2019-12-12 16:28:47 +00:00
let srv = test::start_with(test::config().h1(), || {
App::new().service(web::resource("/").route(web::to(move || {
HttpResponse::Ok()
.no_chunking(STR.len() as u64)
2019-12-20 07:50:07 +00:00
.streaming(TestBody::new(Bytes::from_static(STR.as_ref()), 24))
2019-12-12 16:28:47 +00:00
})))
2019-11-26 05:25:50 +00:00
});
let mut response = srv.get("/").send().await.unwrap();
assert!(response.status().is_success());
assert!(!response.headers().contains_key(TRANSFER_ENCODING));
// read response
let bytes = response.body().await.unwrap();
assert_eq!(bytes, Bytes::from_static(STR.as_ref()));
2018-01-12 00:22:27 +00:00
}
2019-11-26 05:25:50 +00:00
#[actix_rt::test]
async fn test_body_deflate() {
2019-12-12 16:28:47 +00:00
let srv = test::start_with(test::config().h1(), || {
App::new()
.wrap(Compress::new(ContentEncoding::Deflate))
.service(
web::resource("/").route(web::to(move || HttpResponse::Ok().body(STR))),
)
2019-11-26 05:25:50 +00:00
});
// client request
let mut response = srv
.get("/")
.header(ACCEPT_ENCODING, "deflate")
.no_decompress()
.send()
.await
.unwrap();
assert!(response.status().is_success());
// read response
let bytes = response.body().await.unwrap();
let mut e = ZlibDecoder::new(Vec::new());
e.write_all(bytes.as_ref()).unwrap();
let dec = e.finish().unwrap();
assert_eq!(Bytes::from(dec), Bytes::from_static(STR.as_ref()));
2018-01-11 05:47:30 +00:00
}
2019-11-26 05:25:50 +00:00
#[actix_rt::test]
async fn test_body_brotli() {
2019-12-12 16:28:47 +00:00
let srv = test::start_with(test::config().h1(), || {
App::new().wrap(Compress::new(ContentEncoding::Br)).service(
web::resource("/").route(web::to(move || HttpResponse::Ok().body(STR))),
)
2019-11-26 05:25:50 +00:00
});
// client request
let mut response = srv
.get("/")
.header(ACCEPT_ENCODING, "br")
.no_decompress()
.send()
.await
.unwrap();
assert!(response.status().is_success());
// read response
let bytes = response.body().await.unwrap();
// decode brotli
2019-12-20 07:50:07 +00:00
let mut e = BrotliDecoder::new(Vec::with_capacity(2048));
2019-11-26 05:25:50 +00:00
e.write_all(bytes.as_ref()).unwrap();
2019-12-20 07:50:07 +00:00
let dec = e.finish().unwrap();
2019-11-26 05:25:50 +00:00
assert_eq!(Bytes::from(dec), Bytes::from_static(STR.as_ref()));
2018-01-11 05:47:30 +00:00
}
2019-11-26 05:25:50 +00:00
#[actix_rt::test]
async fn test_encoding() {
2019-12-12 16:28:47 +00:00
let srv = test::start_with(test::config().h1(), || {
App::new().wrap(Compress::default()).service(
web::resource("/")
.route(web::to(move |body: Bytes| HttpResponse::Ok().body(body))),
)
2019-11-26 05:25:50 +00:00
});
// client request
let mut e = GzEncoder::new(Vec::new(), Compression::default());
e.write_all(STR.as_ref()).unwrap();
let enc = e.finish().unwrap();
let request = srv
.post("/")
.header(CONTENT_ENCODING, "gzip")
.send_body(enc.clone());
let mut response = request.await.unwrap();
assert!(response.status().is_success());
// read response
let bytes = response.body().await.unwrap();
assert_eq!(bytes, Bytes::from_static(STR.as_ref()));
}
2019-11-26 05:25:50 +00:00
#[actix_rt::test]
async fn test_gzip_encoding() {
2019-12-12 16:28:47 +00:00
let srv = test::start_with(test::config().h1(), || {
App::new().service(
web::resource("/")
.route(web::to(move |body: Bytes| HttpResponse::Ok().body(body))),
)
2019-11-26 05:25:50 +00:00
});
// client request
let mut e = GzEncoder::new(Vec::new(), Compression::default());
e.write_all(STR.as_ref()).unwrap();
let enc = e.finish().unwrap();
let request = srv
.post("/")
.header(CONTENT_ENCODING, "gzip")
.send_body(enc.clone());
let mut response = request.await.unwrap();
assert!(response.status().is_success());
// read response
let bytes = response.body().await.unwrap();
assert_eq!(bytes, Bytes::from_static(STR.as_ref()));
2019-03-26 22:14:32 +00:00
}
2019-03-02 06:51:32 +00:00
2019-11-26 05:25:50 +00:00
#[actix_rt::test]
async fn test_gzip_encoding_large() {
let data = STR.repeat(10);
2019-12-12 16:28:47 +00:00
let srv = test::start_with(test::config().h1(), || {
App::new().service(
web::resource("/")
.route(web::to(move |body: Bytes| HttpResponse::Ok().body(body))),
)
2019-11-26 05:25:50 +00:00
});
// client request
let mut e = GzEncoder::new(Vec::new(), Compression::default());
e.write_all(data.as_ref()).unwrap();
let enc = e.finish().unwrap();
let request = srv
.post("/")
.header(CONTENT_ENCODING, "gzip")
.send_body(enc.clone());
let mut response = request.await.unwrap();
assert!(response.status().is_success());
// read response
let bytes = response.body().await.unwrap();
assert_eq!(bytes, Bytes::from(data));
2019-03-26 22:14:32 +00:00
}
2019-03-02 06:51:32 +00:00
2019-11-26 05:25:50 +00:00
#[actix_rt::test]
async fn test_reading_gzip_encoding_large_random() {
let data = rand::thread_rng()
.sample_iter(&Alphanumeric)
.take(60_000)
.collect::<String>();
2019-12-12 16:28:47 +00:00
let srv = test::start_with(test::config().h1(), || {
App::new().service(
web::resource("/")
.route(web::to(move |body: Bytes| HttpResponse::Ok().body(body))),
)
2019-11-26 05:25:50 +00:00
});
// client request
let mut e = GzEncoder::new(Vec::new(), Compression::default());
e.write_all(data.as_ref()).unwrap();
let enc = e.finish().unwrap();
let request = srv
.post("/")
.header(CONTENT_ENCODING, "gzip")
.send_body(enc.clone());
let mut response = request.await.unwrap();
assert!(response.status().is_success());
// read response
let bytes = response.body().await.unwrap();
assert_eq!(bytes.len(), data.len());
assert_eq!(bytes, Bytes::from(data));
2019-03-26 22:14:32 +00:00
}
2019-03-02 06:51:32 +00:00
2019-11-26 05:25:50 +00:00
#[actix_rt::test]
async fn test_reading_deflate_encoding() {
2019-12-12 16:28:47 +00:00
let srv = test::start_with(test::config().h1(), || {
App::new().service(
web::resource("/")
.route(web::to(move |body: Bytes| HttpResponse::Ok().body(body))),
)
2019-11-26 05:25:50 +00:00
});
let mut e = ZlibEncoder::new(Vec::new(), Compression::default());
e.write_all(STR.as_ref()).unwrap();
let enc = e.finish().unwrap();
// client request
let request = srv
.post("/")
.header(CONTENT_ENCODING, "deflate")
.send_body(enc.clone());
let mut response = request.await.unwrap();
assert!(response.status().is_success());
// read response
let bytes = response.body().await.unwrap();
assert_eq!(bytes, Bytes::from_static(STR.as_ref()));
2019-03-26 22:14:32 +00:00
}
2019-03-02 06:51:32 +00:00
2019-11-26 05:25:50 +00:00
#[actix_rt::test]
async fn test_reading_deflate_encoding_large() {
let data = STR.repeat(10);
2019-12-12 16:28:47 +00:00
let srv = test::start_with(test::config().h1(), || {
App::new().service(
web::resource("/")
.route(web::to(move |body: Bytes| HttpResponse::Ok().body(body))),
)
2019-11-26 05:25:50 +00:00
});
let mut e = ZlibEncoder::new(Vec::new(), Compression::default());
e.write_all(data.as_ref()).unwrap();
let enc = e.finish().unwrap();
// client request
let request = srv
.post("/")
.header(CONTENT_ENCODING, "deflate")
.send_body(enc.clone());
let mut response = request.await.unwrap();
assert!(response.status().is_success());
// read response
let bytes = response.body().await.unwrap();
assert_eq!(bytes, Bytes::from(data));
2019-03-26 22:14:32 +00:00
}
2019-11-26 05:25:50 +00:00
#[actix_rt::test]
async fn test_reading_deflate_encoding_large_random() {
let data = rand::thread_rng()
.sample_iter(&Alphanumeric)
.take(160_000)
.collect::<String>();
2019-12-12 16:28:47 +00:00
let srv = test::start_with(test::config().h1(), || {
App::new().service(
web::resource("/")
.route(web::to(move |body: Bytes| HttpResponse::Ok().body(body))),
)
2019-11-26 05:25:50 +00:00
});
let mut e = ZlibEncoder::new(Vec::new(), Compression::default());
e.write_all(data.as_ref()).unwrap();
let enc = e.finish().unwrap();
// client request
let request = srv
.post("/")
.header(CONTENT_ENCODING, "deflate")
.send_body(enc.clone());
let mut response = request.await.unwrap();
assert!(response.status().is_success());
// read response
let bytes = response.body().await.unwrap();
assert_eq!(bytes.len(), data.len());
assert_eq!(bytes, Bytes::from(data));
2019-03-26 22:14:32 +00:00
}
2019-11-26 05:25:50 +00:00
#[actix_rt::test]
async fn test_brotli_encoding() {
2019-12-12 16:28:47 +00:00
let srv = test::start_with(test::config().h1(), || {
App::new().service(
web::resource("/")
.route(web::to(move |body: Bytes| HttpResponse::Ok().body(body))),
)
2019-11-26 05:25:50 +00:00
});
2019-12-20 07:50:07 +00:00
let mut e = BrotliEncoder::new(Vec::new(), 5);
2019-11-26 05:25:50 +00:00
e.write_all(STR.as_ref()).unwrap();
2019-12-20 07:50:07 +00:00
let enc = e.finish().unwrap();
2019-11-26 05:25:50 +00:00
// client request
let request = srv
.post("/")
.header(CONTENT_ENCODING, "br")
.send_body(enc.clone());
let mut response = request.await.unwrap();
assert!(response.status().is_success());
// read response
let bytes = response.body().await.unwrap();
assert_eq!(bytes, Bytes::from_static(STR.as_ref()));
2019-03-26 22:14:32 +00:00
}
2019-11-26 05:25:50 +00:00
#[actix_rt::test]
async fn test_brotli_encoding_large() {
2019-12-20 07:50:07 +00:00
let data = rand::thread_rng()
.sample_iter(&Alphanumeric)
.take(320_000)
.collect::<String>();
2019-12-12 16:28:47 +00:00
let srv = test::start_with(test::config().h1(), || {
App::new().service(
web::resource("/")
.app_data(web::PayloadConfig::new(320_000))
2019-12-20 07:50:07 +00:00
.route(web::to(move |body: Bytes| {
HttpResponse::Ok().streaming(TestBody::new(body, 10240))
})),
2019-12-12 16:28:47 +00:00
)
2019-11-26 05:25:50 +00:00
});
2019-12-20 07:50:07 +00:00
let mut e = BrotliEncoder::new(Vec::new(), 5);
2019-11-26 05:25:50 +00:00
e.write_all(data.as_ref()).unwrap();
2019-12-20 07:50:07 +00:00
let enc = e.finish().unwrap();
2019-11-26 05:25:50 +00:00
// client request
let request = srv
.post("/")
.header(CONTENT_ENCODING, "br")
.send_body(enc.clone());
let mut response = request.await.unwrap();
assert!(response.status().is_success());
// read response
2019-12-20 07:50:07 +00:00
let bytes = response.body().limit(320_000).await.unwrap();
2019-11-26 05:25:50 +00:00
assert_eq!(bytes, Bytes::from(data));
2019-03-26 22:14:32 +00:00
}
2019-03-02 06:51:32 +00:00
2019-12-12 16:28:47 +00:00
#[cfg(feature = "openssl")]
#[actix_rt::test]
async fn test_brotli_encoding_large_openssl() {
// load ssl keys
use open_ssl::ssl::{SslAcceptor, SslFiletype, SslMethod};
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();
2019-03-02 06:51:32 +00:00
2019-12-12 16:28:47 +00:00
let data = STR.repeat(10);
let srv = test::start_with(test::config().openssl(builder.build()), move || {
App::new().service(web::resource("/").route(web::to(|bytes: Bytes| {
HttpResponse::Ok()
2019-12-18 03:30:14 +00:00
.encoding(actix_web::http::ContentEncoding::Identity)
2019-12-12 16:28:47 +00:00
.body(bytes)
})))
});
2019-03-02 06:51:32 +00:00
2019-12-12 16:28:47 +00:00
// body
2019-12-20 07:50:07 +00:00
let mut e = BrotliEncoder::new(Vec::new(), 3);
2019-12-12 16:28:47 +00:00
e.write_all(data.as_ref()).unwrap();
2019-12-20 07:50:07 +00:00
let enc = e.finish().unwrap();
2019-03-02 06:51:32 +00:00
2019-12-12 16:28:47 +00:00
// client request
let mut response = srv
.post("/")
2019-12-18 03:30:14 +00:00
.header(actix_web::http::header::CONTENT_ENCODING, "br")
2019-12-12 16:28:47 +00:00
.send_body(enc)
.await
.unwrap();
assert!(response.status().is_success());
2019-03-02 06:51:32 +00:00
2019-12-12 16:28:47 +00:00
// read response
let bytes = response.body().await.unwrap();
assert_eq!(bytes, Bytes::from(data));
}
2019-03-02 06:51:32 +00:00
#[cfg(all(feature = "rustls", feature = "openssl"))]
2019-11-26 05:25:50 +00:00
#[actix_rt::test]
2019-12-12 16:28:47 +00:00
async fn test_reading_deflate_encoding_large_random_rustls() {
2019-11-26 05:25:50 +00:00
use rust_tls::internal::pemfile::{certs, pkcs8_private_keys};
use rust_tls::{NoClientAuth, ServerConfig};
use std::fs::File;
use std::io::BufReader;
let data = rand::thread_rng()
.sample_iter(&Alphanumeric)
.take(160_000)
.collect::<String>();
2019-12-12 16:28:47 +00:00
// load ssl keys
let mut config = ServerConfig::new(NoClientAuth::new());
let cert_file = &mut BufReader::new(File::open("tests/cert.pem").unwrap());
let key_file = &mut BufReader::new(File::open("tests/key.pem").unwrap());
let cert_chain = certs(cert_file).unwrap();
let mut keys = pkcs8_private_keys(key_file).unwrap();
config.set_single_cert(cert_chain, keys.remove(0)).unwrap();
let srv = test::start_with(test::config().rustls(config), || {
App::new().service(web::resource("/").route(web::to(|bytes: Bytes| {
HttpResponse::Ok()
2019-12-18 03:30:14 +00:00
.encoding(actix_web::http::ContentEncoding::Identity)
2019-12-12 16:28:47 +00:00
.body(bytes)
})))
2019-11-26 05:25:50 +00:00
});
// encode data
let mut e = ZlibEncoder::new(Vec::new(), Compression::default());
e.write_all(data.as_ref()).unwrap();
let enc = e.finish().unwrap();
// client request
2019-12-12 16:28:47 +00:00
let req = srv
.post("/")
2019-12-18 03:30:14 +00:00
.header(actix_web::http::header::CONTENT_ENCODING, "deflate")
2019-12-20 07:50:07 +00:00
.send_stream(TestBody::new(Bytes::from(enc), 1024));
2019-11-26 05:25:50 +00:00
let mut response = req.await.unwrap();
assert!(response.status().is_success());
// read response
let bytes = response.body().await.unwrap();
assert_eq!(bytes.len(), data.len());
assert_eq!(bytes, Bytes::from(data));
2019-03-29 23:28:19 +00:00
}
2019-03-02 06:51:32 +00:00
// #[test]
// fn test_server_cookies() {
// use actix_web::http;
2019-11-20 17:33:22 +00:00
// let srv = test::TestServer::with_factory(|| {
2019-03-02 06:51:32 +00:00
// App::new().resource("/", |r| {
// r.f(|_| {
// HttpResponse::Ok()
// .cookie(
// http::CookieBuilder::new("first", "first_value")
// .http_only(true)
// .finish(),
// )
// .cookie(http::Cookie::new("second", "first_value"))
// .cookie(http::Cookie::new("second", "second_value"))
// .finish()
// })
// })
// });
// let first_cookie = http::CookieBuilder::new("first", "first_value")
// .http_only(true)
// .finish();
// let second_cookie = http::Cookie::new("second", "second_value");
2019-04-02 20:35:01 +00:00
// let request = srv.get("/").finish().unwrap();
2019-03-02 06:51:32 +00:00
// let response = srv.execute(request.send()).unwrap();
// assert!(response.status().is_success());
// let cookies = response.cookies().expect("To have cookies");
// assert_eq!(cookies.len(), 2);
// if cookies[0] == first_cookie {
// assert_eq!(cookies[1], second_cookie);
// } else {
// assert_eq!(cookies[0], second_cookie);
// assert_eq!(cookies[1], first_cookie);
// }
// let first_cookie = first_cookie.to_string();
// let second_cookie = second_cookie.to_string();
// //Check that we have exactly two instances of raw cookie headers
// let cookies = response
// .headers()
// .get_all(http::header::SET_COOKIE)
// .iter()
// .map(|header| header.to_str().expect("To str").to_string())
// .collect::<Vec<_>>();
// assert_eq!(cookies.len(), 2);
// if cookies[0] == first_cookie {
// assert_eq!(cookies[1], second_cookie);
// } else {
// assert_eq!(cookies[0], second_cookie);
// assert_eq!(cookies[1], first_cookie);
// }
// }
2019-12-12 16:28:47 +00:00
#[actix_rt::test]
async fn test_slow_request() {
use std::net;
2019-03-02 06:51:32 +00:00
2019-12-12 16:28:47 +00:00
let srv = test::start_with(test::config().client_timeout(200), || {
App::new().service(web::resource("/").route(web::to(HttpResponse::Ok)))
2019-12-12 16:28:47 +00:00
});
2019-03-02 06:51:32 +00:00
2019-12-12 16:28:47 +00:00
let mut stream = net::TcpStream::connect(srv.addr()).unwrap();
let mut data = String::new();
let _ = stream.read_to_string(&mut data);
assert!(data.starts_with("HTTP/1.1 408 Request Timeout"));
2019-03-02 06:51:32 +00:00
2019-12-12 16:28:47 +00:00
let mut stream = net::TcpStream::connect(srv.addr()).unwrap();
let _ = stream.write_all(b"GET /test/tests/test HTTP/1.1\r\n");
let mut data = String::new();
let _ = stream.read_to_string(&mut data);
assert!(data.starts_with("HTTP/1.1 408 Request Timeout"));
}
2019-03-02 06:51:32 +00:00
2019-12-12 16:28:47 +00:00
// #[cfg(feature = "openssl")]
// #[actix_rt::test]
// async fn test_ssl_handshake_timeout() {
// use open_ssl::ssl::{SslAcceptor, SslFiletype, SslMethod};
2019-03-02 06:51:32 +00:00
// use std::net;
// // 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();
2019-12-12 16:28:47 +00:00
// let srv = test::start_with(test::config().openssl(builder.build()), || {
// App::new().service(web::resource("/").route(web::to(|| HttpResponse::Ok())))
2019-03-02 06:51:32 +00:00
// });
2019-12-12 16:28:47 +00:00
// let mut stream = net::TcpStream::connect(srv.addr()).unwrap();
2019-03-02 06:51:32 +00:00
// let mut data = String::new();
// let _ = stream.read_to_string(&mut data);
// assert!(data.is_empty());
// }