mirror of
https://github.com/rwf2/Rocket.git
synced 2024-12-31 23:02:37 +00:00
2893ce754d
Catchers can now be scoped to paths, with preference given to the longest-prefix, then the status code. This a breaking change for all applications that register catchers: * `Rocket::register()` takes a base path to scope catchers under. - The previous behavior is recovered with `::register("/", ...)`. * Catchers now fallibly, instead of silently, collide. * `ErrorKind::Collision` is now `ErrorKind::Collisions`. Related changes: * `Origin` implements `TryFrom<String>`, `TryFrom<&str>`. * All URI variants implement `TryFrom<Uri>`. * Added `Segments::prefix_of()`. * `Rocket::mount()` takes a `TryInto<Origin<'_>>` instead of `&str` for the base mount point. * Extended `errors` example with scoped catchers. * Added scoped sections to catchers guide. Internal changes: * Moved router code to `router/router.rs`.
91 lines
2.7 KiB
Rust
91 lines
2.7 KiB
Rust
#[macro_use] extern crate rocket;
|
|
|
|
use rocket::{Rocket, Route, Request};
|
|
use rocket::data::Data;
|
|
use rocket::http::{Method, Status};
|
|
use rocket::local::blocking::Client;
|
|
use rocket::catcher::{Catcher, ErrorHandlerFuture};
|
|
use rocket::handler::HandlerFuture;
|
|
|
|
#[get("/panic")]
|
|
fn panic_route() -> &'static str {
|
|
panic!("Panic in route")
|
|
}
|
|
|
|
#[catch(404)]
|
|
fn panic_catcher() -> &'static str {
|
|
panic!("Panic in catcher")
|
|
}
|
|
|
|
#[catch(500)]
|
|
fn ise() -> &'static str {
|
|
"Hey, sorry! :("
|
|
}
|
|
|
|
fn pre_future_route<'r>(_: &'r Request<'_>, _: Data) -> HandlerFuture<'r> {
|
|
panic!("hey now...");
|
|
}
|
|
|
|
fn rocket() -> Rocket {
|
|
rocket::ignite()
|
|
.mount("/", routes![panic_route])
|
|
.mount("/", vec![Route::new(Method::Get, "/pre", pre_future_route)])
|
|
}
|
|
|
|
#[test]
|
|
fn catches_route_panic() {
|
|
let rocket = rocket().register("/", catchers![panic_catcher, ise]);
|
|
let client = Client::debug(rocket).unwrap();
|
|
let response = client.get("/panic").dispatch();
|
|
assert_eq!(response.status(), Status::InternalServerError);
|
|
assert_eq!(response.into_string().unwrap(), "Hey, sorry! :(");
|
|
}
|
|
|
|
#[test]
|
|
fn catches_catcher_panic() {
|
|
let rocket = rocket().register("/", catchers![panic_catcher, ise]);
|
|
let client = Client::debug(rocket).unwrap();
|
|
let response = client.get("/noroute").dispatch();
|
|
assert_eq!(response.status(), Status::InternalServerError);
|
|
assert_eq!(response.into_string().unwrap(), "Hey, sorry! :(");
|
|
}
|
|
|
|
#[test]
|
|
fn catches_double_panic() {
|
|
#[catch(500)]
|
|
fn double_panic() {
|
|
panic!("so, so sorry...")
|
|
}
|
|
|
|
let rocket = rocket().register("/", catchers![panic_catcher, double_panic]);
|
|
let client = Client::debug(rocket).unwrap();
|
|
let response = client.get("/noroute").dispatch();
|
|
assert_eq!(response.status(), Status::InternalServerError);
|
|
assert!(response.into_string().unwrap().contains("Rocket"));
|
|
}
|
|
|
|
#[test]
|
|
fn catches_early_route_panic() {
|
|
let rocket = rocket().register("/", catchers![panic_catcher, ise]);
|
|
let client = Client::debug(rocket).unwrap();
|
|
let response = client.get("/pre").dispatch();
|
|
assert_eq!(response.status(), Status::InternalServerError);
|
|
assert_eq!(response.into_string().unwrap(), "Hey, sorry! :(");
|
|
}
|
|
|
|
#[test]
|
|
fn catches_early_catcher_panic() {
|
|
fn pre_future_catcher<'r>(_: Status, _: &'r Request) -> ErrorHandlerFuture<'r> {
|
|
panic!("a panicking pre-future catcher")
|
|
}
|
|
|
|
let rocket = rocket()
|
|
.register("/", vec![Catcher::new(404, pre_future_catcher)])
|
|
.register("/", catchers![ise]);
|
|
|
|
let client = Client::debug(rocket).unwrap();
|
|
let response = client.get("/idontexist").dispatch();
|
|
assert_eq!(response.status(), Status::InternalServerError);
|
|
assert_eq!(response.into_string().unwrap(), "Hey, sorry! :(");
|
|
}
|