mirror of https://github.com/rwf2/Rocket.git
94 lines
2.7 KiB
Rust
94 lines
2.7 KiB
Rust
#![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()));
|
|
}
|
|
}
|