Merge pull request #5 from NorthskySocial/aitorres/healthcheck

Add healthcheck route

Author: TheRipperoni

Cargo.toml

@@ -19,4 +19,4 @@ serde_json = "1.0.91"
 reqwest = { version = "0.12.14", features = ["json"] }
 rpassword = "7.4.0"
 rust-argon2 = "3.0.0"
-tracing = "0.1.41"
+tracing = "0.1.41"

src/main.rs

@@ -14,6 +14,7 @@ use crate::routes::create_invite_codes::create_invite_codes_handler;
 use crate::routes::disable_invite_codes::disable_invite_codes_handler;
 use crate::routes::generate_otp::generate_otp_handler;
 use crate::routes::get_invite_codes::get_invite_codes_handler;
+use crate::routes::health::healthcheck_handler;
 use crate::routes::login::login_user;
 use crate::routes::validate_otp::validate_otp_handler;
 use crate::routes::verify_otp::verify_otp_handler;
@@ -121,6 +122,7 @@ async fn main() -> io::Result<()> {
             ))
             .app_data(Data::new(db_pool.clone()))
             .app_data(Data::new(config.clone()))
+            .service(healthcheck_handler)
             .service(login_user)
             .service(generate_otp_handler)
             .service(verify_otp_handler)

src/routes/health.rs

@@ -0,0 +1,22 @@
+use actix_web::{HttpResponse, Responder, get};
+
+#[get("/health")]
+pub async fn healthcheck_handler() -> impl Responder {
+    HttpResponse::Ok().message_body("ok")
+}
+
+#[cfg(test)]
+mod tests {
+    use actix_web::{App, body::MessageBody, test};
+
+    use super::*;
+
+    #[actix_web::test]
+    async fn test_healthcheck_get() {
+        let app = test::init_service(App::new().service(healthcheck_handler)).await;
+        let req = test::TestRequest::get().uri("/health").to_request();
+        let resp = test::call_service(&app, req).await;
+        assert!(resp.status().is_success());
+        assert!(resp.into_body().try_into_bytes().unwrap() == "ok");
+    }
+}

src/routes/mod.rs

@@ -8,6 +8,7 @@ pub mod create_invite_codes;
 pub mod disable_invite_codes;
 pub mod generate_otp;
 pub mod get_invite_codes;
+pub mod health;
 pub mod login;
 pub mod validate_otp;
 pub mod verify_otp;