tests/test_app.py

@@ -0,0 +1,89 @@
+from __future__ import annotations
+from typing import Any
+
+from fastapi.testclient import TestClient
+
+from gibby.app import create_app
+from gibby.manager import AccountManager, NoUsableAccountError
+
+
+class StubManager(AccountManager):
+    def __init__(
+        self,
+        response: dict[str, Any] | None = None,
+        usage_response: dict[str, Any] | None = None,
+        error: Exception | None = None,
+    ):
+        self.response = response
+        self.usage_response = usage_response
+        self.error = error
+
+    async def issue_token_response(self):
+        if self.error is not None:
+            raise self.error
+        if self.response is not None:
+            return self.response
+        return {}
+
+    async def get_usage_report(self):
+        if self.usage_response is not None:
+            return self.usage_response
+        return {"accounts": [], "active_account_id": None, "count": 0}
+
+
+def test_health_ok() -> None:
+    client = TestClient(create_app(StubManager(response={})))
+    response = client.get("/health")
+    assert response.status_code == 200
+    assert response.text == "ok"
+
+
+def test_token_success_shape() -> None:
+    payload = {
+        "token": "abc",
+        "limit": {
+            "used_percent": 10,
+            "remaining_percent": 90,
+            "exhausted": False,
+            "needs_prepare": False,
+        },
+        "usage": {"primary_window": None, "secondary_window": None},
+    }
+    client = TestClient(create_app(StubManager(response=payload)))
+    response = client.get("/token")
+    assert response.status_code == 200
+    assert response.json() == payload
+
+
+def test_token_error_shape() -> None:
+    client = TestClient(
+        create_app(
+            StubManager(error=NoUsableAccountError("No usable account available"))
+        )
+    )
+    response = client.get("/token")
+    assert response.status_code == 503
+    assert response.json() == {"error": "No usable account available"}
+
+
+def test_usage_success_shape() -> None:
+    payload = {
+        "accounts": [
+            {
+                "id": "a1",
+                "email": "a1@example.com",
+                "status": "ok",
+                "used_percent": 12,
+                "remaining_percent": 88,
+                "cooldown_until": None,
+                "primary_window": {"used_percent": 12},
+                "secondary_window": {"used_percent": 1},
+            }
+        ],
+        "active_account_id": "a1",
+        "count": 1,
+    }
+    client = TestClient(create_app(StubManager(usage_response=payload)))
+    response = client.get("/usage")
+    assert response.status_code == 200
+    assert response.json() == payload