Rocket/core/lib/tests/limits.rs

#![feature(proc_macro_hygiene)]

#[macro_use] extern crate rocket;

use rocket::request::Form;

#[derive(FromForm)]
struct Simple {
    value: String
}

#[post("/", data = "<form>")]
fn index(form: Form<Simple>) -> String {
    form.into_inner().value
}

mod limits_tests {
    use rocket;
    use rocket::config::{Environment, Config, Limits};
    use rocket::local::Client;
    use rocket::http::{Status, ContentType};

    fn rocket_with_forms_limit(limit: u64) -> rocket::Rocket {
        let config = Config::build(Environment::Development)
            .limits(Limits::default().limit("forms", limit))
            .unwrap();

        rocket::custom(config).mount("/", routes![super::index])
    }

    #[rocket::async_test]
    async fn large_enough() {
        let client = Client::new(rocket_with_forms_limit(128)).await.unwrap();
        let mut response = client.post("/")
            .body("value=Hello+world")
            .header(ContentType::Form)
            .dispatch().await;

        assert_eq!(response.body_string().await, Some("Hello world".into()));
    }

    #[rocket::async_test]
    async fn just_large_enough() {
        let client = Client::new(rocket_with_forms_limit(17)).await.unwrap();
        let mut response = client.post("/")
            .body("value=Hello+world")
            .header(ContentType::Form)
            .dispatch().await;

        assert_eq!(response.body_string().await, Some("Hello world".into()));
    }

    #[rocket::async_test]
    async fn much_too_small() {
        let client = Client::new(rocket_with_forms_limit(4)).await.unwrap();
        let response = client.post("/")
            .body("value=Hello+world")
            .header(ContentType::Form)
            .dispatch().await;

        assert_eq!(response.status(), Status::UnprocessableEntity);
    }

    #[rocket::async_test]
    async fn contracted() {
        let client = Client::new(rocket_with_forms_limit(10)).await.unwrap();
        let mut response = client.post("/")
            .body("value=Hello+world")
            .header(ContentType::Form)
            .dispatch().await;

        assert_eq!(response.body_string().await, Some("Hell".into()));
    }
}