Add tests for extended_validation example.

examples/extended_validation/Cargo.toml

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

examples/extended_validation/src/main.rs

@@ -4,6 +4,8 @@
 extern crate rocket;
 
 mod files;
+#[cfg(test)]
+mod tests;
 
 use rocket::response::Redirect;
 use rocket::request::{Form, FromFormValue};
@@ -77,8 +79,11 @@ fn user_page(username: &str) -> String {
     format!("This is {}'s page.", username)
 }
 
-fn main() {
+fn rocket() -> rocket::Rocket {
     rocket::ignite()
         .mount("/", routes![files::index, files::files, user_page, login])
-        .launch();
+}
+
+fn main() {
+    rocket().launch()
 }

examples/extended_validation/src/tests.rs

@@ -0,0 +1,86 @@
+use rocket::testing::MockRequest;
+use rocket::http::Method::*;
+use rocket::http::{ContentType, Status};
+
+use super::rocket;
+
+fn test_login<T>(user: &str, pass: &str, age: &str, status: Status, body: T)
+    where T: Into<Option<&'static str>>
+{
+    let rocket = rocket();
+    let query = format!("username={}&password={}&age={}", user, pass, age);
+
+    let mut req = MockRequest::new(Post, "/login")
+        .header(ContentType::Form)
+        .body(&query);
+
+    let mut response = req.dispatch_with(&rocket);
+    assert_eq!(response.status(), status);
+
+    let body_str = response.body().and_then(|body| body.into_string());
+    if let Some(expected_str) = body.into() {
+        assert!(body_str.map_or(false, |s| s.contains(expected_str)));
+    }
+}
+
+#[test]
+fn test_good_login() {
+    test_login("Sergio", "password", "30", Status::SeeOther, None);
+}
+
+#[test]
+fn test_invalid_user() {
+    test_login("-1", "password", "30", Status::Ok, "Unrecognized user");
+    test_login("Mike", "password", "30", Status::Ok, "Unrecognized user");
+}
+
+#[test]
+fn test_invalid_password() {
+    test_login("Sergio", "password1", "30", Status::Ok, "Wrong password!");
+    test_login("Sergio", "ok", "30", Status::Ok, "Password is invalid: Too short!");
+}
+
+#[test]
+fn test_invalid_age() {
+    test_login("Sergio", "password", "20", Status::Ok, "Must be at least 21.");
+    test_login("Sergio", "password", "-100", Status::Ok, "Must be at least 21.");
+    test_login("Sergio", "password", "hi", Status::Ok, "Age value is not a number");
+}
+
+fn check_bad_form(form_str: &str, status: Status) {
+    let rocket = rocket();
+    let mut req = MockRequest::new(Post, "/login")
+        .header(ContentType::Form)
+        .body(form_str);
+
+    let response = req.dispatch_with(&rocket);
+    assert_eq!(response.status(), status);
+}
+
+#[test]
+fn test_bad_form_abnromal_inputs() {
+    check_bad_form("&", Status::BadRequest);
+    check_bad_form("=", Status::BadRequest);
+    check_bad_form("&&&===&", Status::BadRequest);
+}
+
+#[test]
+fn test_bad_form_missing_fields() {
+    let bad_inputs: [&str; 6] = [
+        "username=Sergio",
+        "password=pass",
+        "age=30",
+        "username=Sergio&password=pass",
+        "username=Sergio&age=30",
+        "password=pass&age=30"
+    ];
+
+    for bad_input in bad_inputs.into_iter() {
+        check_bad_form(bad_input, Status::UnprocessableEntity);
+    }
+}
+
+#[test]
+fn test_bad_form_additional_fields() {
+    check_bad_form("username=Sergio&password=pass&age=30&addition=1", Status::UnprocessableEntity);
+}