Merge pull request #16 from Bibimbap-Team/15-change-project-structure-for-easier-testing

Change project structure for easier testing

Author: utilForever

Cargo.toml

@@ -3,5 +3,12 @@ name = "bibimbap-backend"
 version = "0.1.0"
 edition = "2021"
 
+[lib]
+path = "src/lib.rs"
+
+[[bin]]
+path = "src/main.rs"
+name = "bibimbap-backend"
+
 [dependencies]
 rocket = "0.5.1"

src/lib.rs

@@ -0,0 +1,13 @@
+#[macro_use]
+extern crate rocket;
+
+use rocket::{Build, Rocket};
+
+#[get("/health_check")]
+fn health_check() {
+    ()
+}
+
+pub fn run() -> Rocket<Build> {
+    rocket::build().mount("/", routes![health_check])
+}

src/main.rs

@@ -1,29 +1,9 @@
-#[macro_use]
-extern crate rocket;
+use bibimbap_backend::run;
 
-#[get("/health_check")]
-fn health_check() {
-    ()
-}
-
-#[launch]
-fn rocket() -> _ {
-    rocket::build().mount("/", routes![health_check])
-}
-
-#[cfg(test)]
-mod test {
-    use super::rocket;
-    use rocket::http::Status;
-    use rocket::local::blocking::Client;
-    use std::io::Read;
-
-    #[test]
-    fn health_check_works() {
-        let client = Client::tracked(rocket()).expect("Failed to create client");
-        let response = client.get(uri!(super::health_check)).dispatch();
-
-        assert_eq!(response.status(), Status::Ok);
-        assert_eq!(response.bytes().count(), 0);
+#[rocket::main]
+async fn main() {
+    if let Err(e) = run().launch().await {
+        println!("Failed to launch Rocket: {e}");
+        drop(e);
     }
 }

tests/health_check.rs

@@ -0,0 +1,20 @@
+#[macro_use]
+extern crate rocket;
+
+use rocket::http::Status;
+use rocket::local::blocking::Client;
+use rocket::{Build, Rocket};
+use std::io::Read;
+
+fn spawn_app() -> Rocket<Build> {
+    bibimbap_backend::run()
+}
+
+#[test]
+fn health_check_works() {
+    let client = Client::tracked(spawn_app()).expect("Failed to create client");
+    let response = client.get(uri!("/health_check")).dispatch();
+
+    assert_eq!(response.status(), Status::Ok);
+    assert_eq!(response.bytes().count(), 0);
+}