mirror of
https://github.com/rwf2/Rocket.git
synced 2024-12-30 22:32:38 +00:00
03127f4dae
This commit adds the 'local::blocking' module and moves the existing asynchronous testing to 'local::asynchronous'. It also includes several changes to improve the local API, bringing it to parity (and beyond) with master. These changes are: * 'LocalRequest' implements 'Clone'. * 'LocalResponse' doesn't implement 'DerefMut<Target=Response>'. Instead, direct methods on the type, such as 'into_string()', can be used to read the 'Response'. * 'Response::body()' returns an '&ResponseBody' as opposed to '&mut ResponseBody', which is returned by a new 'Response::body_mut()'. * '&ResponseBody' implements 'known_size()` to retrieve a body's size, if it is known. Co-authored-by: Jeb Rosen <jeb@jebrosen.com>
56 lines
1.7 KiB
Rust
56 lines
1.7 KiB
Rust
#![feature(proc_macro_hygiene)]
|
|
|
|
#[macro_use] extern crate rocket;
|
|
|
|
use rocket::local::asynchronous::Client;
|
|
|
|
// Test that manual/auto ranking works as expected.
|
|
|
|
#[get("/<_number>")]
|
|
fn get0(_number: u8) -> &'static str { "0" }
|
|
|
|
#[get("/<_number>", rank = 1)]
|
|
fn get1(_number: u16) -> &'static str { "1" }
|
|
|
|
#[get("/<_number>", rank = 2)]
|
|
fn get2(_number: u32) -> &'static str { "2" }
|
|
|
|
#[get("/<_number>", rank = 3)]
|
|
fn get3(_number: u64) -> &'static str { "3" }
|
|
|
|
#[rocket::async_test]
|
|
async fn test_ranking() {
|
|
let rocket = rocket::ignite().mount("/", routes![get0, get1, get2, get3]);
|
|
let client = Client::new(rocket).await.unwrap();
|
|
|
|
let response = client.get("/0").dispatch().await;
|
|
assert_eq!(response.into_string().await.unwrap(), "0");
|
|
|
|
let response = client.get(format!("/{}", 1 << 8)).dispatch().await;
|
|
assert_eq!(response.into_string().await.unwrap(), "1");
|
|
|
|
let response = client.get(format!("/{}", 1 << 16)).dispatch().await;
|
|
assert_eq!(response.into_string().await.unwrap(), "2");
|
|
|
|
let response = client.get(format!("/{}", 1u64 << 32)).dispatch().await;
|
|
assert_eq!(response.into_string().await.unwrap(), "3");
|
|
}
|
|
|
|
// Test a collision due to same auto rank.
|
|
|
|
#[get("/<_n>")]
|
|
fn get0b(_n: u8) { }
|
|
|
|
#[rocket::async_test]
|
|
async fn test_rank_collision() {
|
|
use rocket::error::LaunchErrorKind;
|
|
|
|
let rocket = rocket::ignite().mount("/", routes![get0, get0b]);
|
|
let client_result = Client::new(rocket).await;
|
|
match client_result.as_ref().map_err(|e| e.kind()) {
|
|
Err(LaunchErrorKind::Collision(..)) => { /* o.k. */ },
|
|
Ok(_) => panic!("client succeeded unexpectedly"),
|
|
Err(e) => panic!("expected collision, got {}", e)
|
|
}
|
|
}
|