2017-02-17 01:22:23 +00:00
|
|
|
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);
|
|
|
|
|
2017-04-14 08:59:28 +00:00
|
|
|
let body_str = response.body_string();
|
2017-02-17 01:22:23 +00:00
|
|
|
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!");
|
2017-03-31 06:06:53 +00:00
|
|
|
test_login("Sergio", "ok", "30", Status::Ok, "Password is invalid: too short!");
|
2017-02-17 01:22:23 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_invalid_age() {
|
2017-03-31 06:06:53 +00:00
|
|
|
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, "value is not a number");
|
2017-02-17 01:22:23 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
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);
|
|
|
|
}
|