mirror of
https://github.com/rustfs/rustfs.git
synced 2026-08-20 03:22:18 +00:00
test(signer): cover header fallback helpers (#2711)
This commit is contained in:
@@ -72,3 +72,34 @@ pub(crate) fn error_chain_contains_signer_header_marker(err: &(dyn StdError + 's
|
|||||||
|
|
||||||
false
|
false
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
use super::{error_chain_contains_signer_header_marker, invalid_utf8_header_error, signer_error_to_io_error};
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn invalid_utf8_header_error_is_detected_through_error_chain() {
|
||||||
|
let err = invalid_utf8_header_error("failed to sign request", "x-amz-meta-invalid");
|
||||||
|
|
||||||
|
assert!(error_chain_contains_signer_header_marker(&err));
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn mapped_signer_header_error_is_detected_through_error_chain() {
|
||||||
|
let err = signer_error_to_io_error(
|
||||||
|
"failed to sign request",
|
||||||
|
rustfs_signer::SignV4Error::InvalidHeaderValue {
|
||||||
|
name: "x-amz-meta-invalid".to_string(),
|
||||||
|
},
|
||||||
|
);
|
||||||
|
|
||||||
|
assert!(error_chain_contains_signer_header_marker(&err));
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn generic_io_errors_do_not_match_signer_header_marker() {
|
||||||
|
let err = std::io::Error::other("unrelated failure");
|
||||||
|
|
||||||
|
assert!(!error_chain_contains_signer_header_marker(&err));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|||||||
@@ -60,3 +60,56 @@ where
|
|||||||
{
|
{
|
||||||
v.sort_by(|lhs, rhs| lhs.0.cmp(&rhs.0));
|
v.sort_by(|lhs, rhs| lhs.0.cmp(&rhs.0));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
use super::{HostAddrError, try_get_host_addr};
|
||||||
|
use http::HeaderValue;
|
||||||
|
use http::request;
|
||||||
|
use s3s::Body;
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn try_get_host_addr_prefers_explicit_host_header_when_it_differs_from_uri() {
|
||||||
|
let mut req = request::Request::builder()
|
||||||
|
.method(http::Method::GET)
|
||||||
|
.uri("https://bucket.example.com/object")
|
||||||
|
.body(Body::empty())
|
||||||
|
.expect("request should build");
|
||||||
|
req.headers_mut()
|
||||||
|
.insert("host", HeaderValue::from_static("proxy.internal:9443"));
|
||||||
|
|
||||||
|
let host = try_get_host_addr(&req).expect("host lookup should succeed");
|
||||||
|
|
||||||
|
assert_eq!(host, "proxy.internal:9443");
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn try_get_host_addr_rejects_non_utf8_host_header_value() {
|
||||||
|
let mut req = request::Request::builder()
|
||||||
|
.method(http::Method::GET)
|
||||||
|
.uri("https://bucket.example.com/object")
|
||||||
|
.body(Body::empty())
|
||||||
|
.expect("request should build");
|
||||||
|
req.headers_mut().insert(
|
||||||
|
"host",
|
||||||
|
HeaderValue::from_bytes(&[0xFF]).expect("invalid utf8 bytes should be accepted by HeaderValue"),
|
||||||
|
);
|
||||||
|
|
||||||
|
let err = try_get_host_addr(&req).expect_err("invalid host header should fail");
|
||||||
|
|
||||||
|
assert!(matches!(err, HostAddrError::InvalidHostHeader));
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn try_get_host_addr_rejects_relative_uri_without_host() {
|
||||||
|
let req = request::Request::builder()
|
||||||
|
.method(http::Method::GET)
|
||||||
|
.uri("/object")
|
||||||
|
.body(Body::empty())
|
||||||
|
.expect("request should build");
|
||||||
|
|
||||||
|
let err = try_get_host_addr(&req).expect_err("relative uri should fail");
|
||||||
|
|
||||||
|
assert!(matches!(err, HostAddrError::MissingUriHost));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user