mirror of https://github.com/rwf2/Rocket.git
80 lines
2.4 KiB
Rust
80 lines
2.4 KiB
Rust
use rocket::error::Error;
|
|
use rocket::fairing::AdHoc;
|
|
|
|
#[rocket::async_test]
|
|
async fn test_inspectable_launch_state() -> Result<(), Error> {
|
|
let rocket = rocket::ignite()
|
|
.attach(AdHoc::on_launch("Add State", |rocket| async {
|
|
Ok(rocket.manage("Hi!"))
|
|
}))
|
|
._ignite()
|
|
.await?;
|
|
|
|
let state = rocket.state::<&'static str>();
|
|
assert_eq!(state, Some(&"Hi!"));
|
|
Ok(())
|
|
}
|
|
|
|
#[rocket::async_test]
|
|
async fn test_inspectable_launch_state_in_liftoff() -> Result<(), Error> {
|
|
let rocket = rocket::ignite()
|
|
.attach(AdHoc::on_launch("Add State", |rocket| async {
|
|
Ok(rocket.manage("Hi!"))
|
|
}))
|
|
.attach(AdHoc::on_launch("Inspect State", |rocket| async {
|
|
let state = rocket.state::<&'static str>();
|
|
assert_eq!(state, Some(&"Hi!"));
|
|
Ok(rocket)
|
|
}))
|
|
.attach(AdHoc::on_liftoff("Inspect State", |rocket| Box::pin(async move {
|
|
let state = rocket.state::<&'static str>();
|
|
assert_eq!(state, Some(&"Hi!"));
|
|
})))
|
|
._ignite()
|
|
.await?;
|
|
|
|
let state = rocket.state::<&'static str>();
|
|
assert_eq!(state, Some(&"Hi!"));
|
|
Ok(())
|
|
}
|
|
|
|
#[rocket::async_test]
|
|
async fn test_launch_state_is_well_ordered() -> Result<(), Error> {
|
|
let rocket = rocket::ignite()
|
|
.attach(AdHoc::on_launch("Inspect State Pre", |rocket| async {
|
|
let state = rocket.state::<&'static str>();
|
|
assert_eq!(state, None);
|
|
Ok(rocket)
|
|
}))
|
|
.attach(AdHoc::on_launch("Add State", |rocket| async {
|
|
Ok(rocket.manage("Hi!"))
|
|
}))
|
|
.attach(AdHoc::on_launch("Inspect State", |rocket| async {
|
|
let state = rocket.state::<&'static str>();
|
|
assert_eq!(state, Some(&"Hi!"));
|
|
Ok(rocket)
|
|
}))
|
|
._ignite()
|
|
.await?;
|
|
|
|
let state = rocket.state::<&'static str>();
|
|
assert_eq!(state, Some(&"Hi!"));
|
|
Ok(())
|
|
}
|
|
|
|
#[should_panic]
|
|
#[rocket::async_test]
|
|
async fn negative_test_launch_state() {
|
|
let _ = rocket::ignite()
|
|
.attach(AdHoc::on_launch("Add State", |rocket| async {
|
|
Ok(rocket.manage("Hi!"))
|
|
}))
|
|
.attach(AdHoc::on_launch("Inspect State", |rocket| async {
|
|
let state = rocket.state::<&'static str>();
|
|
assert_ne!(state, Some(&"Hi!"));
|
|
Ok(rocket)
|
|
}))
|
|
._ignite()
|
|
.await;
|
|
}
|