This commit is contained in:
Mateusz Gruszczyński committed 2026-10-05 09:30:21 +02:00
1 parent 1dea7f55c8
commit c819f0a335
151 files changed
+7610 -4264

No files matched your search

+37 -24
View File
@@ -15,11 +15,7 @@ mod tests {
Some(4 * 1024 * 1024)
);
assert_eq!(
transform_memory_required_bytes(
2 * 1024 * 1024,
Some("gzip"),
2 * 1024 * 1024
),
transform_memory_required_bytes(2 * 1024 * 1024, Some("gzip"), 2 * 1024 * 1024),
Some(6 * 1024 * 1024)
);
assert_eq!(
@@ -113,7 +109,9 @@ mod tests {
response
.headers_mut()
.insert("content-length", HeaderValue::from_static("128"));
assert!(apply(response, "/data.json", &settings, false).await.is_err());
assert!(apply(response, "/data.json", &settings, false)
.await
.is_err());
}
#[test]
@@ -143,10 +141,7 @@ mod tests {
assert_eq!(view.total.cache_hits, 1);
assert_eq!(view.total.saved_bytes, 30);
assert_eq!(view.total.by_kind.get("css"), Some(&1));
assert_eq!(
view.total.reasons.get("response_no_transform"),
Some(&1)
);
assert_eq!(view.total.reasons.get("response_no_transform"), Some(&1));
assert_eq!(view.scopes.len(), 1);
assert_eq!(view.scopes[0].vhost, "app");
}
@@ -185,8 +180,13 @@ mod tests {
response
.headers_mut()
.insert("vary", HeaderValue::from_static("Accept-Encoding"));
let response = apply(response, "/data.json", &settings, false).await.unwrap();
assert!(response.headers()["etag"].to_str().unwrap().starts_with("\"rustproxy-"));
let response = apply(response, "/data.json", &settings, false)
.await
.unwrap();
assert!(response.headers()["etag"]
.to_str()
.unwrap()
.starts_with("\"rustproxy-"));
assert!(!response.headers().contains_key("vary"));
assert!(response.extensions().get::<CanonicalIdentity>().is_some());
let validators = response.extensions().get::<OriginValidators>().unwrap();
@@ -213,11 +213,12 @@ mod tests {
response
.headers_mut()
.insert("etag", HeaderValue::from_static("\"origin\""));
response.headers_mut().insert(
"vary",
HeaderValue::from_static("Accept-Encoding, X-Theme"),
);
let response = apply(response, "/data.json", &settings, false).await.unwrap();
response
.headers_mut()
.insert("vary", HeaderValue::from_static("Accept-Encoding, X-Theme"));
let response = apply(response, "/data.json", &settings, false)
.await
.unwrap();
assert_eq!(response.headers()["etag"], "\"origin\"");
assert_eq!(response.headers()["vary"], "X-Theme");
assert!(response.extensions().get::<CanonicalIdentity>().is_some());
@@ -252,13 +253,18 @@ mod tests {
.headers_mut()
.insert("etag", HeaderValue::from_static("\"origin\""));
let response = apply(response, "/data.json", &settings, false).await.unwrap();
let response = apply(response, "/data.json", &settings, false)
.await
.unwrap();
assert!(!response.headers().contains_key("content-encoding"));
assert_eq!(
response.headers()["content-length"].to_str().unwrap(),
identity.len().to_string()
);
assert!(response.headers()["etag"].to_str().unwrap().starts_with("\"rustproxy-"));
assert!(response.headers()["etag"]
.to_str()
.unwrap()
.starts_with("\"rustproxy-"));
let validators = response.extensions().get::<OriginValidators>().unwrap();
assert_eq!(validators.etag.as_ref().unwrap(), "\"origin\"");
let applied = response.extensions().get::<Applied>().unwrap();
@@ -287,7 +293,9 @@ mod tests {
.headers_mut()
.insert("etag", HeaderValue::from_static("\"origin\""));
let response = apply(response, "/data.json", &settings, true).await.unwrap();
let response = apply(response, "/data.json", &settings, true)
.await
.unwrap();
assert_eq!(response.headers()["etag"], "\"origin\"");
assert!(response.extensions().get::<Applied>().is_none());
let received = response.into_body().collect().await.unwrap().to_bytes();
@@ -315,7 +323,9 @@ mod tests {
HeaderValue::from_static("public, max-age=60, no-transform"),
);
let response = apply(response, "/data.json", &settings, false).await.unwrap();
let response = apply(response, "/data.json", &settings, false)
.await
.unwrap();
assert!(response.extensions().get::<Applied>().is_none());
let received = response.into_body().collect().await.unwrap().to_bytes();
assert_eq!(&received[..], body);
@@ -339,7 +349,9 @@ mod tests {
.headers_mut()
.insert("etag", HeaderValue::from_static("\"origin-head\""));
let response = apply(response, "/data.json", &settings, false).await.unwrap();
let response = apply(response, "/data.json", &settings, false)
.await
.unwrap();
assert_eq!(response.headers()["content-length"], "76");
assert_eq!(response.headers()["etag"], "\"origin-head\"");
assert!(response.extensions().get::<Applied>().is_none());
@@ -383,7 +395,9 @@ mod tests {
.headers_mut()
.insert("etag", HeaderValue::from_static("\"origin\""));
let response = apply(response, "/data.json", &settings, false).await.unwrap();
let response = apply(response, "/data.json", &settings, false)
.await
.unwrap();
assert_eq!(response.headers()["content-encoding"], "gzip");
assert_eq!(response.headers()["etag"], "\"origin\"");
assert_eq!(response.headers()["vary"], "Accept-Encoding");
@@ -391,5 +405,4 @@ mod tests {
let received = response.into_body().collect().await.unwrap().to_bytes();
assert_eq!(&received[..], encoded.as_slice());
}
}