rocket 0.5.1

Web framework with a focus on usability, security, extensibility, 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

use rocket::{Rocket, Build};
use rocket::{fairing::AdHoc, http::ContentType, local::blocking::Client};

#[rocket::post("/", data = "<_data>", format = "json")]
fn index(_data: rocket::Data<'_>) -> &'static str { "json" }

#[rocket::post("/", data = "<_data>", rank = 2)]
fn other_index(_data: rocket::Data<'_>) -> &'static str { "other" }

fn rocket() -> Rocket<Build> {
    rocket::build()
        .mount("/", rocket::routes![index, other_index])
        .attach(AdHoc::on_request("Change CT", |req, _| Box::pin(async move {
            let need_ct = req.content_type().is_none();
            if req.uri().path().starts_with("/add") {
                req.set_uri(rocket::uri!(index));
                if need_ct { req.add_header(ContentType::JSON); }
            } else if need_ct {
                req.replace_header(ContentType::JSON);
            }
        })))
}

#[test]
fn check_fairing_changes_content_type() {
    let client = Client::debug(rocket()).unwrap();
    let response = client.post("/").header(ContentType::PNG).dispatch();
    assert_eq!(response.into_string().unwrap(), "other");

    let response = client.post("/").dispatch();
    assert_eq!(response.into_string().unwrap(), "json");

    let response = client.post("/add").dispatch();
    assert_eq!(response.into_string().unwrap(), "json");

    let response = client.post("/add").header(ContentType::HTML).dispatch();
    assert_eq!(response.into_string().unwrap(), "other");
}