actix/actix-web
1
0
Fork 0
mirror of https://github.com/fafhrd91/actix-web synced 2025-01-31 19:10:07 +01:00
Code Issues Releases Wiki Activity

misc: add unit test for expected behaviour jpeg

Browse Source
This commit is contained in:
William R. Arellano 2023-03-07 21:11:41 -05:00
parent 68adcf657a
commit c96844ab14
1 changed files with 28 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
actix-web/src/middleware/compress.rs
View File

@ -17,7 +17,7 @@ use pin_project_lite::pin_project;
use crate::{ use crate::{
body::{EitherBody, MessageBody}, body::{EitherBody, MessageBody},
http::{ http::{
header::{self, AcceptEncoding, Encoding, HeaderValue}, header::{self, AcceptEncoding, ContentType, Encoding, HeaderValue},
StatusCode, StatusCode,
}, },
service::{ServiceRequest, ServiceResponse}, service::{ServiceRequest, ServiceResponse},
@ -324,4 +324,31 @@ mod tests {
assert!(vary_headers.contains(&HeaderValue::from_static("x-test"))); assert!(vary_headers.contains(&HeaderValue::from_static("x-test")));
assert!(vary_headers.contains(&HeaderValue::from_static("accept-encoding"))); assert!(vary_headers.contains(&HeaderValue::from_static("accept-encoding")));
} }
#[actix_rt::test]
async fn prevents_compression_jpeg() {
const D: &str = "test image";
const DATA: &str = const_str::repeat!(D, 100);
let app = test::init_service({
App::new().wrap(Compress::default()).route(
"/image",
web::get().to(move || {
let mut builder = HttpResponse::Ok();
builder.body(DATA);
builder.insert_header(ContentType::jpeg());
builder
}),
)
})
.await;
let req = test::TestRequest::default()
.uri("/image")
.insert_header((header::ACCEPT_ENCODING, "gzip"))
.to_request();
let res = test::call_service(&app, req).await;
assert_eq!(res.status(), StatusCode::OK);
assert_eq!(res.headers().get(header::CONTENT_TYPE).unwrap(), "gzip");
let bytes = test::read_body(res).await;
assert_eq!(bytes, DATA.as_bytes());
}
} }