rocket 0.3.6

Web framework for nightly with a focus on ease-of-use, expressibility, and speed.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93

#![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()));
    }
}