1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
|
pub(super) mod future;
pub(super) mod layer;
pub(super) mod service;
#[cfg(test)]
mod tests {
use super::service::RequestDecompression;
use crate::decompression::DecompressionBody;
use crate::test_helpers::Body;
use flate2::{write::GzEncoder, Compression};
use http::{header, Request, Response, StatusCode};
use http_body_util::BodyExt;
use std::{convert::Infallible, io::Write};
use tower::{service_fn, Service, ServiceExt};
#[tokio::test]
async fn decompress_accepted_encoding() {
let req = request_gzip();
let mut svc = RequestDecompression::new(service_fn(assert_request_is_decompressed));
let _ = svc.ready().await.unwrap().call(req).await.unwrap();
}
#[tokio::test]
async fn support_unencoded_body() {
let req = Request::builder().body(Body::from("Hello?")).unwrap();
let mut svc = RequestDecompression::new(service_fn(assert_request_is_decompressed));
let _ = svc.ready().await.unwrap().call(req).await.unwrap();
}
#[tokio::test]
async fn unaccepted_content_encoding_returns_unsupported_media_type() {
let req = request_gzip();
let mut svc = RequestDecompression::new(service_fn(should_not_be_called)).gzip(false);
let res = svc.ready().await.unwrap().call(req).await.unwrap();
assert_eq!(StatusCode::UNSUPPORTED_MEDIA_TYPE, res.status());
}
#[tokio::test]
async fn pass_through_unsupported_encoding_when_enabled() {
let req = request_gzip();
let mut svc = RequestDecompression::new(service_fn(assert_request_is_passed_through))
.pass_through_unaccepted(true)
.gzip(false);
let _ = svc.ready().await.unwrap().call(req).await.unwrap();
}
async fn assert_request_is_decompressed(
req: Request<DecompressionBody<Body>>,
) -> Result<Response<Body>, Infallible> {
let (parts, mut body) = req.into_parts();
let body = read_body(&mut body).await;
assert_eq!(body, b"Hello?");
assert!(!parts.headers.contains_key(header::CONTENT_ENCODING));
Ok(Response::new(Body::from("Hello, World!")))
}
async fn assert_request_is_passed_through(
req: Request<DecompressionBody<Body>>,
) -> Result<Response<Body>, Infallible> {
let (parts, mut body) = req.into_parts();
let body = read_body(&mut body).await;
assert_ne!(body, b"Hello?");
assert!(parts.headers.contains_key(header::CONTENT_ENCODING));
Ok(Response::new(Body::empty()))
}
async fn should_not_be_called(
_: Request<DecompressionBody<Body>>,
) -> Result<Response<Body>, Infallible> {
panic!("Inner service should not be called");
}
fn request_gzip() -> Request<Body> {
let mut encoder = GzEncoder::new(Vec::new(), Compression::default());
encoder.write_all(b"Hello?").unwrap();
let body = encoder.finish().unwrap();
Request::builder()
.header(header::CONTENT_ENCODING, "gzip")
.body(Body::from(body))
.unwrap()
}
async fn read_body(body: &mut DecompressionBody<Body>) -> Vec<u8> {
body.collect().await.unwrap().to_bytes().to_vec()
}
}
|
