Rocket/lib/tests/local-request-content-type-...

94 lines
2.7 KiB
Rust
Raw Normal View History

2017-12-18 14:26:55 +00:00
#![feature(plugin, decl_macro)]
#![plugin(rocket_codegen)]
extern crate rocket;
use rocket::Outcome::*;
use rocket::{Request, Data};
use rocket::request::{self, FromRequest};
struct HasContentType;
impl<'a, 'r> FromRequest<'a, 'r> for HasContentType {
type Error = ();
fn from_request(request: &'a Request<'r>) -> request::Outcome<Self, ()> {
if request.content_type().is_some() {
Success(HasContentType)
} else {
Forward(())
}
}
}
use rocket::data::{self, FromData};
impl FromData for HasContentType {
type Error = ();
fn from_data(request: &Request, data: Data) -> data::Outcome<Self, ()> {
if request.content_type().is_some() {
Success(HasContentType)
} else {
Forward(data)
}
}
}
#[post("/")]
fn rg_ct(ct: Option<HasContentType>) -> &'static str {
ct.map_or("Absent", |_| "Present")
}
#[post("/data", data = "<_ct>", rank = 1)]
fn data_has_ct(_ct: HasContentType) -> &'static str {
"Data Present"
}
#[post("/data", rank = 2)]
fn data_no_ct() -> &'static str {
"Data Absent"
}
mod local_request_content_type_tests {
use super::*;
use rocket::Rocket;
use rocket::local::Client;
use rocket::http::ContentType;
fn rocket() -> Rocket {
rocket::ignite().mount("/", routes![rg_ct, data_has_ct, data_no_ct])
}
#[test]
fn has_no_ct() {
let client = Client::new(rocket()).unwrap();
let mut req = client.post("/");
assert_eq!(req.cloned_dispatch().body_string(), Some("Absent".to_string()));
assert_eq!(req.mut_dispatch().body_string(), Some("Absent".to_string()));
assert_eq!(req.dispatch().body_string(), Some("Absent".to_string()));
let mut req = client.post("/data");
assert_eq!(req.cloned_dispatch().body_string(), Some("Data Absent".to_string()));
assert_eq!(req.mut_dispatch().body_string(), Some("Data Absent".to_string()));
assert_eq!(req.dispatch().body_string(), Some("Data Absent".to_string()));
}
#[test]
fn has_ct() {
let client = Client::new(rocket()).unwrap();
let mut req = client.post("/").header(ContentType::JSON);
assert_eq!(req.cloned_dispatch().body_string(), Some("Present".to_string()));
assert_eq!(req.mut_dispatch().body_string(), Some("Present".to_string()));
assert_eq!(req.dispatch().body_string(), Some("Present".to_string()));
let mut req = client.post("/data").header(ContentType::JSON);
assert_eq!(req.cloned_dispatch().body_string(), Some("Data Present".to_string()));
assert_eq!(req.mut_dispatch().body_string(), Some("Data Present".to_string()));
assert_eq!(req.dispatch().body_string(), Some("Data Present".to_string()));
}
}