mirror of
https://github.com/rwf2/Rocket.git
synced 2025-01-05 17:22:36 +00:00
4e06ee64aa
Prior to this commit, it was not possible to test Rocket crates in production mode without setting a global secret key or bypassing secret key checking - the testing script did the latter. The consequence is that it became impossible to test secret key related failures because the tests passed regardless. This commit undoes this. As a consequence, all tests are now aware of the difference between debug and release configurations, the latter of which validates 'secret_key' by default. New 'Client::debug()' and 'Client::debug_with()' simplify creating an instance of 'Client' with configuration in debug mode to avoid undesired test failures. The summary of changes in this commit are: * Config 'secret_key' success and failure are now tested. * 'secret_key' validation was moved to pre-launch from 'Config:from()'. * 'Config::from()' only extracts the config. * Added 'Config::try_from()' for non-panicking extraction. * 'Config' now knows the profile it was extracted from. * The 'Config' provider sets a profile of 'Config.profile'. * 'Rocket', 'Client', 'Fairings', implement 'Debug'. * 'fairing::Info' implements 'Copy', 'Clone'. * 'Fairings' keeps track of, logs attach fairings. * 'Rocket::reconfigure()' was added to allow modifying a config. Internally, the testing script was refactored to properly test the codebase with the new changes. In particular, it no longer sets a rustc 'cfg' to avoid secret-key checking. Resolves #1543. Fixes #1564.
123 lines
2.9 KiB
Rust
123 lines
2.9 KiB
Rust
#[macro_use] extern crate rocket;
|
|
#[macro_use] extern crate bencher;
|
|
|
|
#[get("/")]
|
|
fn hello_world() -> &'static str { "Hello, world!" }
|
|
|
|
#[put("/")]
|
|
fn put_index() -> &'static str { "index" }
|
|
|
|
#[post("/")]
|
|
fn post_index() -> &'static str { "index" }
|
|
|
|
#[get("/a")]
|
|
fn index_a() -> &'static str { "index" }
|
|
|
|
#[get("/b")]
|
|
fn index_b() -> &'static str { "index" }
|
|
|
|
#[get("/c")]
|
|
fn index_c() -> &'static str { "index" }
|
|
|
|
#[get("/<_a>")]
|
|
fn index_dyn_a(_a: &str) -> &'static str { "index" }
|
|
|
|
fn hello_world_rocket() -> rocket::Rocket {
|
|
let config = rocket::Config {
|
|
log_level: rocket::config::LogLevel::Off,
|
|
..rocket::Config::debug_default()
|
|
};
|
|
|
|
rocket::custom(config).mount("/", routes![hello_world])
|
|
}
|
|
|
|
fn rocket() -> rocket::Rocket {
|
|
hello_world_rocket()
|
|
.mount("/", routes![
|
|
put_index, post_index, index_a, index_b, index_c, index_dyn_a
|
|
])
|
|
}
|
|
|
|
use bencher::Bencher;
|
|
use rocket::local::blocking::Client;
|
|
|
|
fn bench_hello_world(b: &mut Bencher) {
|
|
let client = Client::tracked(hello_world_rocket()).unwrap();
|
|
|
|
b.iter(|| {
|
|
client.get("/").dispatch();
|
|
});
|
|
}
|
|
|
|
fn bench_single_get_index(b: &mut Bencher) {
|
|
let client = Client::tracked(rocket()).unwrap();
|
|
|
|
b.iter(|| {
|
|
client.get("/").dispatch();
|
|
});
|
|
}
|
|
|
|
fn bench_get_put_post_index(b: &mut Bencher) {
|
|
let client = Client::tracked(rocket()).unwrap();
|
|
|
|
// Hold all of the requests we're going to make during the benchmark.
|
|
let mut requests = vec![];
|
|
requests.push(client.get("/"));
|
|
requests.push(client.put("/"));
|
|
requests.push(client.post("/"));
|
|
|
|
b.iter(|| {
|
|
for request in &requests {
|
|
request.clone().dispatch();
|
|
}
|
|
});
|
|
}
|
|
|
|
fn bench_dynamic(b: &mut Bencher) {
|
|
let client = Client::tracked(rocket()).unwrap();
|
|
|
|
// Hold all of the requests we're going to make during the benchmark.
|
|
let mut requests = vec![];
|
|
requests.push(client.get("/abc"));
|
|
requests.push(client.get("/abcdefg"));
|
|
requests.push(client.get("/123"));
|
|
|
|
b.iter(|| {
|
|
for request in &requests {
|
|
request.clone().dispatch();
|
|
}
|
|
});
|
|
}
|
|
|
|
fn bench_simple_routing(b: &mut Bencher) {
|
|
let client = Client::tracked(rocket()).unwrap();
|
|
|
|
// Hold all of the requests we're going to make during the benchmark.
|
|
let mut requests = vec![];
|
|
for route in client.rocket().routes() {
|
|
let request = client.req(route.method, route.uri.path().as_str());
|
|
requests.push(request);
|
|
}
|
|
|
|
// A few more for the dynamic route.
|
|
requests.push(client.get("/abc"));
|
|
requests.push(client.get("/abcdefg"));
|
|
requests.push(client.get("/123"));
|
|
|
|
b.iter(|| {
|
|
for request in &requests {
|
|
request.clone().dispatch();
|
|
}
|
|
});
|
|
}
|
|
|
|
benchmark_main!(benches);
|
|
benchmark_group! {
|
|
benches,
|
|
bench_hello_world,
|
|
bench_single_get_index,
|
|
bench_get_put_post_index,
|
|
bench_dynamic,
|
|
bench_simple_routing,
|
|
}
|