rkt 0.6.0

Web framework with a focus on usability, security, extensibility, and speed. (Community Fork)
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

extern crate rkt;

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

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

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

fn rocket() -> Rocket<Build> {
    rkt::build()
        .mount("/", rkt::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(rkt::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");
}