From da7026c78140db823c39d4b1391cb3df5eeb9c0f Mon Sep 17 00:00:00 2001
From: Seth Lopez <sethlopezme@users.noreply.github.com>
Date: Thu, 29 Dec 2016 15:06:31 -0500
Subject: [PATCH] Add tests for cookies example.

---
 examples/cookies/Cargo.toml   |  3 ++
 examples/cookies/src/main.rs  |  8 ++++--
 examples/cookies/src/tests.rs | 54 +++++++++++++++++++++++++++++++++++
 3 files changed, 63 insertions(+), 2 deletions(-)
 create mode 100644 examples/cookies/src/tests.rs

diff --git a/examples/cookies/Cargo.toml b/examples/cookies/Cargo.toml
index ecfac4b4..0f84f4d9 100644
--- a/examples/cookies/Cargo.toml
+++ b/examples/cookies/Cargo.toml
@@ -13,3 +13,6 @@ lazy_static = "*"
 path = "../../contrib"
 default-features = false
 features = ["handlebars_templates"]
+
+[dev-dependencies]
+rocket = { path = "../../lib", features = ["testing"] }
diff --git a/examples/cookies/src/main.rs b/examples/cookies/src/main.rs
index 7a10613c..d11a5af4 100644
--- a/examples/cookies/src/main.rs
+++ b/examples/cookies/src/main.rs
@@ -1,10 +1,14 @@
 #![feature(plugin, custom_derive, custom_attribute)]
 #![plugin(rocket_codegen)]
 
-#[macro_use] extern crate lazy_static;
+#[macro_use]
+extern crate lazy_static;
 extern crate rocket_contrib;
 extern crate rocket;
 
+#[cfg(test)]
+mod tests;
+
 use std::collections::HashMap;
 
 use rocket::request::Form;
@@ -14,7 +18,7 @@ use rocket_contrib::Template;
 
 #[derive(FromForm)]
 struct Message {
-    message: String
+    message: String,
 }
 
 #[post("/submit", data = "<message>")]
diff --git a/examples/cookies/src/tests.rs b/examples/cookies/src/tests.rs
new file mode 100644
index 00000000..12106a8b
--- /dev/null
+++ b/examples/cookies/src/tests.rs
@@ -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());
+}