Add tests for cookies example.

This commit is contained in:
Seth Lopez 2016-12-29 15:06:31 -05:00 committed by Sergio Benitez
parent 83e33cf0be
commit da7026c781
3 changed files with 63 additions and 2 deletions

View File

@ -13,3 +13,6 @@ lazy_static = "*"
path = "../../contrib" path = "../../contrib"
default-features = false default-features = false
features = ["handlebars_templates"] features = ["handlebars_templates"]
[dev-dependencies]
rocket = { path = "../../lib", features = ["testing"] }

View File

@ -1,10 +1,14 @@
#![feature(plugin, custom_derive, custom_attribute)] #![feature(plugin, custom_derive, custom_attribute)]
#![plugin(rocket_codegen)] #![plugin(rocket_codegen)]
#[macro_use] extern crate lazy_static; #[macro_use]
extern crate lazy_static;
extern crate rocket_contrib; extern crate rocket_contrib;
extern crate rocket; extern crate rocket;
#[cfg(test)]
mod tests;
use std::collections::HashMap; use std::collections::HashMap;
use rocket::request::Form; use rocket::request::Form;
@ -14,7 +18,7 @@ use rocket_contrib::Template;
#[derive(FromForm)] #[derive(FromForm)]
struct Message { struct Message {
message: String message: String,
} }
#[post("/submit", data = "<message>")] #[post("/submit", data = "<message>")]

View File

@ -0,0 +1,54 @@
use std::collections::HashMap;
use super::rocket;
use rocket::testing::MockRequest;
use rocket::http::*;
use rocket_contrib::Template;
#[test]
fn test_submit() {
let rocket = rocket::ignite().mount("/", routes![super::submit]);
let mut request = MockRequest::new(Method::Post, "/submit")
.header(ContentType::Form)
.body("message=Hello from Rocket!");
let response = request.dispatch_with(&rocket);
let cookie_headers: Vec<_> = response.header_values("Set-Cookie").collect();
let location_headers: Vec<_> = response.header_values("Location").collect();
assert_eq!(response.status(), Status::SeeOther);
assert_eq!(cookie_headers, vec!["message=Hello%20from%20Rocket!".to_string()]);
assert_eq!(location_headers, vec!["/".to_string()]);
}
fn test_body(optional_cookie: Option<Cookie>, expected_body: String) {
let rocket = rocket::ignite().mount("/", routes![super::index]);
let mut request = MockRequest::new(Method::Get, "/");
// Attach a cookie if one is given.
if let Some(cookie) = optional_cookie {
request = request.cookie(cookie);
}
let mut response = request.dispatch_with(&rocket);
assert_eq!(response.status(), Status::Ok);
assert_eq!(response.body().and_then(|b| b.into_string()), Some(expected_body));
}
#[test]
fn test_index() {
// Render the template with an empty context to test against.
let mut context: HashMap<&str, &str> = HashMap::new();
let template = Template::render("index", &context);
// Test the route without sending the "message" cookie.
test_body(None, template.to_string());
// Render the template with a context that contains the message.
context.insert("message", "Hello from Rocket!");
let template = Template::render("index", &context);
// Test the route with the "message" cookie.
test_body(Some(Cookie::new("message".to_string(),
"Hello from Rocket!".to_string())),
template.to_string());
}