diff --git a/src/__tests__/health-ready.test.ts b/src/__tests__/health-ready.test.ts new file mode 100644 index 0000000..6a8ea88 --- /dev/null +++ b/src/__tests__/health-ready.test.ts @@ -0,0 +1,102 @@ +import { describe, it, expect, vi, beforeEach } from "vitest"; +import { Hono } from "hono"; + +// ─── Mock db module ─────────────────────────────────────────────────────────── + +let selectImpl: () => Promise; + +vi.mock("@groombook/db", () => { + const staff = new Proxy( + { _name: "staff" }, + { + get(_target, prop) { + if (prop === "_name") return "staff"; + return { table: "staff", column: prop }; + }, + } + ); + + return { + getDb: () => ({ + select: (_fields: unknown) => ({ + from: (_table: unknown) => ({ + limit: (_n: number) => selectImpl(), + }), + }), + }), + staff, + }; +}); + +// ─── Build test app ─────────────────────────────────────────────────────────── + +async function makeApp() { + const { getDb, staff } = await import("@groombook/db"); + + const app = new Hono(); + app.get("/health/ready", async (c) => { + try { + await getDb().select({ id: staff.id }).from(staff).limit(1); + return c.json({ status: "ready" }, 200); + } catch (err) { + const pgCode = (err as Record).code ?? "unknown"; + console.error("[health/ready] DB check failed:", pgCode); + return c.json({ status: "degraded" }, 503); + } + }); + return app; +} + +// ─── Tests ──────────────────────────────────────────────────────────────────── + +describe("GET /health/ready", () => { + beforeEach(() => { + vi.restoreAllMocks(); + }); + + it("returns 200 {status:'ready'} when DB query succeeds", async () => { + selectImpl = () => Promise.resolve([{ id: "staff-1" }]); + + const app = await makeApp(); + const res = await app.request("/health/ready", { method: "GET" }); + const body = (await res.json()) as Record; + + expect(res.status).toBe(200); + expect(body.status).toBe("ready"); + }); + + it("returns 503 {status:'degraded'} when DB query throws (schema dropped)", async () => { + const schemaErr = Object.assign(new Error("relation \"staff\" does not exist"), { code: "42P01" }); + selectImpl = () => Promise.reject(schemaErr); + + const consoleSpy = vi.spyOn(console, "error").mockImplementation(() => {}); + + const app = await makeApp(); + const res = await app.request("/health/ready", { method: "GET" }); + const body = (await res.json()) as Record; + + expect(res.status).toBe(503); + expect(body.status).toBe("degraded"); + + // Must not leak SQL error details in the response body + expect(JSON.stringify(body)).not.toContain("42P01"); + expect(JSON.stringify(body)).not.toContain("relation"); + + consoleSpy.mockRestore(); + }); + + it("returns 503 {status:'degraded'} on any DB connection error", async () => { + selectImpl = () => Promise.reject(new Error("ECONNREFUSED")); + + const consoleSpy = vi.spyOn(console, "error").mockImplementation(() => {}); + + const app = await makeApp(); + const res = await app.request("/health/ready", { method: "GET" }); + const body = (await res.json()) as Record; + + expect(res.status).toBe(503); + expect(body.status).toBe("degraded"); + + consoleSpy.mockRestore(); + }); +}); diff --git a/src/index.ts b/src/index.ts index 6da003b..99a66fa 100644 --- a/src/index.ts +++ b/src/index.ts @@ -65,6 +65,17 @@ app.use( app.get("/health", (c) => c.json({ status: "ok" })); // /api/health: used by Gateway HTTPRoute (/api/* → API pod) app.get("/api/health", (c) => c.json({ status: "ok" })); +// /health/ready: DB-touching readiness probe — K8s removes pod from endpoints when schema is dropped (GRO-2689) +app.get("/health/ready", async (c) => { + try { + await getDb().select({ id: staff.id }).from(staff).limit(1); + return c.json({ status: "ready" }, 200); + } catch (err) { + const pgCode = (err as Record).code ?? "unknown"; + console.error("[health/ready] DB check failed:", pgCode); + return c.json({ status: "degraded" }, 503); + } +}); // /api/readyz: DB-touching deep health check consumed by monitoring (not K8s probes) // Distinct from /health so a dropped schema triggers an alert without cycling pods (GRO-2678) app.get("/api/readyz", async (c) => {