Add tests for redirect example.

This commit is contained in:
Cliff H 2016-12-29 11:41:23 -05:00 committed by Sergio Benitez
parent a6084ab3e2
commit 83e33cf0be
3 changed files with 44 additions and 0 deletions

View File

@ -7,3 +7,6 @@ workspace = "../../"
[dependencies] [dependencies]
rocket = { path = "../../lib" } rocket = { path = "../../lib" }
rocket_codegen = { path = "../../codegen" } rocket_codegen = { path = "../../codegen" }
[dev-dependencies]
rocket = { path = "../../lib", features = ["testing"] }

View File

@ -1,7 +1,10 @@
#![feature(plugin)] #![feature(plugin)]
#![plugin(rocket_codegen)] #![plugin(rocket_codegen)]
extern crate rocket; extern crate rocket;
#[cfg(test)] mod tests;
use rocket::response::Redirect; use rocket::response::Redirect;
#[get("/")] #[get("/")]

View File

@ -0,0 +1,38 @@
use super::rocket;
use rocket::testing::MockRequest;
use rocket::Response;
use rocket::http::Method::*;
use rocket::http::Status;
macro_rules! run_test {
($path:expr, $test_fn:expr) => ({
let rocket = rocket::ignite().mount("/", routes![super::root, super::login]);
let mut request = MockRequest::new(Get, format!($path));
$test_fn(request.dispatch_with(&rocket));
})
}
#[test]
fn test_root() {
run_test!("/", |mut response: Response| {
assert!(response.body().is_none());
assert_eq!(response.status(), Status::SeeOther);
for h in response.headers() {
match h.name.as_str() {
"Location" => assert_eq!(h.value, "/login"),
"Content-Length" => assert_eq!(h.value.parse::<i32>().unwrap(), 0),
_ => { /* let these through */ }
}
}
});
}
#[test]
fn test_login() {
run_test!("/login", |mut response: Response| {
let body_string = response.body().and_then(|body| body.into_string());
assert_eq!(body_string, Some("Hi! Please log in before continuing.".to_string()));
assert_eq!(response.status(), Status::Ok);
});
}