mirror of
https://github.com/rwf2/Rocket.git
synced 2025-01-06 01:32:36 +00:00
9b955747e4
This commit includes two major changes to core:
1. Configuration state is no longer global. The `config::active()`
function has been removed. The active configuration can be
retrieved via the `config` method on a `Rocket` instance.
2. The `Responder` trait has changed. `Responder::respond(self)` has
been removed in favor of `Responder::respond_to(self, &Request)`.
This allows responders to dynamically adjust their response based
on the incoming request.
Additionally, it includes the following changes to core and codegen:
* The `Request::guard` method was added to allow for simple
retrivial of request guards.
* The `Request::limits` method was added to retrieve configured
limits.
* The `File` `Responder` implementation now uses a fixed size body
instead of a chunked body.
* The `Outcome::of<R: Responder>(R)` method was removed while
`Outcome::from<R: Responder(&Request, R)` was added.
* The unmounted and unmanaged limits are more cautious: they will only
emit warnings when the `Rocket` receiver is known.
This commit includes one major change to contrib:
1. To use contrib's templating, the fairing returned by
`Template::fairing()` must be attached to the running Rocket
instance.
Additionally, the `Display` implementation of `Template` was removed. To
directly render a template to a `String`, the new `Template::show`
method can be used.
72 lines
2.6 KiB
Rust
72 lines
2.6 KiB
Rust
use rocket::{self, State};
|
|
use rocket::fairing::AdHoc;
|
|
use rocket::config::{self, Config, Environment};
|
|
use rocket::http::{Method, Status};
|
|
use rocket::LoggingLevel;
|
|
use rocket::testing::MockRequest;
|
|
|
|
struct LocalConfig(Config);
|
|
|
|
#[get("/check_config")]
|
|
fn check_config(config: State<LocalConfig>) -> Option<()> {
|
|
let environment = match ::std::env::var("ROCKET_ENV") {
|
|
Ok(name) => name,
|
|
Err(_) => return None
|
|
};
|
|
|
|
let config = &config.0;
|
|
match &*environment {
|
|
"development" => {
|
|
assert_eq!(config.address, "localhost".to_string());
|
|
assert_eq!(config.port, 8000);
|
|
assert_eq!(config.workers, 1);
|
|
assert_eq!(config.log_level, LoggingLevel::Normal);
|
|
assert_eq!(config.environment, config::Environment::Development);
|
|
assert_eq!(config.extras().count(), 2);
|
|
assert_eq!(config.get_str("hi"), Ok("Hello!"));
|
|
assert_eq!(config.get_bool("is_extra"), Ok(true));
|
|
}
|
|
"staging" => {
|
|
assert_eq!(config.address, "0.0.0.0".to_string());
|
|
assert_eq!(config.port, 80);
|
|
assert_eq!(config.workers, 8);
|
|
assert_eq!(config.log_level, LoggingLevel::Normal);
|
|
assert_eq!(config.environment, config::Environment::Staging);
|
|
assert_eq!(config.extras().count(), 0);
|
|
}
|
|
"production" => {
|
|
assert_eq!(config.address, "0.0.0.0".to_string());
|
|
assert_eq!(config.port, 80);
|
|
assert_eq!(config.workers, 12);
|
|
assert_eq!(config.log_level, LoggingLevel::Critical);
|
|
assert_eq!(config.environment, config::Environment::Production);
|
|
assert_eq!(config.extras().count(), 0);
|
|
}
|
|
_ => {
|
|
panic!("Unknown environment in envvar: {}", environment);
|
|
}
|
|
}
|
|
|
|
Some(())
|
|
}
|
|
|
|
pub fn test_config(environment: Environment) {
|
|
// Manually set the config environment variable. Rocket will initialize the
|
|
// environment in `ignite()`. We'll read this back in the handler to config.
|
|
::std::env::set_var("ROCKET_ENV", environment.to_string());
|
|
|
|
// FIXME: launch fairings aren't run during tests since...the Rocket isn't
|
|
// being launch
|
|
let rocket = rocket::ignite()
|
|
.attach(AdHoc::on_attach(|rocket| {
|
|
println!("Attaching local config.");
|
|
let config = rocket.config().clone();
|
|
Ok(rocket.manage(LocalConfig(config)))
|
|
}))
|
|
.mount("/", routes![check_config]);
|
|
|
|
let mut request = MockRequest::new(Method::Get, "/check_config");
|
|
let response = request.dispatch_with(&rocket);
|
|
assert_eq!(response.status(), Status::Ok);
|
|
}
|