update/add tests for new HostGuard implementation

This commit is contained in:
Anton Lazarev 2019-07-18 09:37:32 -07:00
parent c995fbc05d
commit b2098922bd
No known key found for this signature in database
GPG Key ID: FBD20243E0CD9104
1 changed files with 56 additions and 13 deletions

View File

@ -341,21 +341,64 @@ mod tests {
assert!(!pred.check(req.head())); assert!(!pred.check(req.head()));
} }
// #[test] #[test]
// fn test_host() { fn test_host() {
// let req = TestServiceRequest::default() let req = TestRequest::default()
// .header( .header(
// header::HOST, header::HOST,
// header::HeaderValue::from_static("www.rust-lang.org"), header::HeaderValue::from_static("www.rust-lang.org"),
// ) )
// .request(); .to_http_request();
// let pred = Host("www.rust-lang.org"); let pred = Host("www.rust-lang.org");
// assert!(pred.check(&req)); assert!(pred.check(req.head()));
// let pred = Host("localhost"); let pred = Host("www.rust-lang.org").scheme("https");
// assert!(!pred.check(&req)); assert!(pred.check(req.head()));
// }
let pred = Host("blog.rust-lang.org");
assert!(!pred.check(req.head()));
let pred = Host("blog.rust-lang.org").scheme("https");
assert!(!pred.check(req.head()));
let pred = Host("crates.io");
assert!(!pred.check(req.head()));
let pred = Host("localhost");
assert!(!pred.check(req.head()));
}
#[test]
fn test_host_scheme() {
let req = TestRequest::default()
.header(
header::HOST,
header::HeaderValue::from_static("https://www.rust-lang.org"),
)
.to_http_request();
let pred = Host("www.rust-lang.org").scheme("https");
assert!(pred.check(req.head()));
let pred = Host("www.rust-lang.org");
assert!(pred.check(req.head()));
let pred = Host("www.rust-lang.org").scheme("http");
assert!(!pred.check(req.head()));
let pred = Host("blog.rust-lang.org");
assert!(!pred.check(req.head()));
let pred = Host("blog.rust-lang.org").scheme("https");
assert!(!pred.check(req.head()));
let pred = Host("crates.io").scheme("https");
assert!(!pred.check(req.head()));
let pred = Host("localhost");
assert!(!pred.check(req.head()));
}
#[test] #[test]
fn test_methods() { fn test_methods() {