mirror of https://github.com/rwf2/Rocket.git
35 lines
1.0 KiB
Rust
35 lines
1.0 KiB
Rust
#[cfg(all(feature = "diesel_sqlite_pool"))]
|
|
#[cfg(test)]
|
|
mod sqlite_shutdown_test {
|
|
use rocket::{async_test, Build, Rocket};
|
|
use rocket_sync_db_pools::database;
|
|
|
|
#[database("test")]
|
|
#[allow(dead_code)]
|
|
struct Pool(diesel::SqliteConnection);
|
|
|
|
async fn rocket() -> Rocket<Build> {
|
|
use rocket::figment::{util::map, Figment};
|
|
|
|
let options = map!["url" => ":memory:"];
|
|
let config = Figment::from(rocket::Config::debug_default())
|
|
.merge(("port", 0))
|
|
.merge(("databases", map!["test" => &options]));
|
|
|
|
rocket::custom(config).attach(Pool::fairing())
|
|
}
|
|
|
|
#[test]
|
|
fn test_shutdown() {
|
|
let _rocket = async_test(
|
|
async {
|
|
let rocket = rocket().await.ignite().await.expect("unable to ignite");
|
|
// request shutdown
|
|
rocket.shutdown().notify();
|
|
rocket.launch().await.expect("unable to launch")
|
|
}
|
|
);
|
|
// _rocket is dropped here after the runtime is dropped
|
|
}
|
|
}
|