feat(api): add health probes, OpenAPI doc, and router

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
2026-06-02 00:58:29 +02:00
parent 8da3eefdce
commit b9acc03761
4 changed files with 180 additions and 0 deletions
@@ -0,0 +1,70 @@
+use api::{AppState, build_app};
+use axum::body::Body;
+use axum::http::{Request, StatusCode};
+use http_body_util::BodyExt;
+use sqlx::PgPool;
+use tower::ServiceExt; // for `oneshot`
+
+fn state(pool: PgPool, app_name: &str) -> AppState {
+    AppState {
+        db: db::Db::from_pool(pool),
+        app_name: app_name.to_string(),
+    }
+}
+
+#[sqlx::test]
+async fn live_returns_ok(pool: PgPool) {
+    let app = build_app(state(pool, "Test"));
+    let resp = app
+        .oneshot(
+            Request::builder()
+                .uri("/health/live")
+                .body(Body::empty())
+                .unwrap(),
+        )
+        .await
+        .unwrap();
+
+    assert_eq!(resp.status(), StatusCode::OK);
+    let bytes = resp.into_body().collect().await.unwrap().to_bytes();
+    let json: serde_json::Value = serde_json::from_slice(&bytes).unwrap();
+    assert_eq!(json["status"], "ok");
+}
+
+#[sqlx::test]
+async fn ready_reports_database_true(pool: PgPool) {
+    let app = build_app(state(pool, "Test"));
+    let resp = app
+        .oneshot(
+            Request::builder()
+                .uri("/health/ready")
+                .body(Body::empty())
+                .unwrap(),
+        )
+        .await
+        .unwrap();
+
+    assert_eq!(resp.status(), StatusCode::OK);
+    let bytes = resp.into_body().collect().await.unwrap().to_bytes();
+    let json: serde_json::Value = serde_json::from_slice(&bytes).unwrap();
+    assert_eq!(json["database"], true);
+}
+
+#[sqlx::test]
+async fn openapi_doc_uses_configured_title(pool: PgPool) {
+    let app = build_app(state(pool, "My Museum CMS"));
+    let resp = app
+        .oneshot(
+            Request::builder()
+                .uri("/api-docs/openapi.json")
+                .body(Body::empty())
+                .unwrap(),
+        )
+        .await
+        .unwrap();
+
+    assert_eq!(resp.status(), StatusCode::OK);
+    let bytes = resp.into_body().collect().await.unwrap().to_bytes();
+    let json: serde_json::Value = serde_json::from_slice(&bytes).unwrap();
+    assert_eq!(json["info"]["title"], "My Museum CMS");
+}