mirror of
https://github.com/rwf2/Rocket.git
synced 2025-01-05 17:22:36 +00:00
560f0977d3
* body_string_wait and body_bytes_wait are removed; use `.await` instead
* `dispatch()` is now an async fn and must be .await-ed
* Add `#[rocket::async_test]` macro, similar in purpose to `tokio::test`
* Tests now use either `rocket::async_test(async { })` or
`#[rocket::async_test]` in order to `.await` the futures returned
from `dispatch()` and `body_{string,bytes}()`
* Update 'test.sh' to reflect the tests that should be passing.
Broken:
* Cloned dispatch and mut_dispatch() with a live previous response now both fail, due to a (partial) check for mutable aliasing in LocalRequest.
* Some tests are still failing and need example-specific changes.
75 lines
2.1 KiB
Rust
75 lines
2.1 KiB
Rust
#![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)).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)).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)).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)).unwrap();
|
|
let mut response = client.post("/")
|
|
.body("value=Hello+world")
|
|
.header(ContentType::Form)
|
|
.dispatch().await;
|
|
|
|
assert_eq!(response.body_string().await, Some("Hell".into()));
|
|
}
|
|
}
|