feat: Better structure, add tests

backend/tests/test_main.py

@@ -0,0 +1,124 @@
+from __future__ import annotations
+
+from fastapi.testclient import TestClient
+from main import create_app
+from models import AgentResponse, LevelInfo
+
+
+def write_config(path) -> None:
+    path.write_text(
+        """
+[llm]
+model = "test-model"
+api_key = "test-key"
+
+[server]
+host = "127.0.0.1"
+port = 8000
+reload = false
+
+[game]
+hard_mode_rotation_interval = 5
+blocked_response_text = "blocked"
+password_words = [
+  "w01", "w02", "w03", "w04", "w05",
+  "w06", "w07", "w08", "w09", "w10",
+  "w11", "w12", "w13", "w14", "w15",
+  "w16", "w17", "w18", "w19", "w20",
+]
+""".strip(),
+        encoding="utf-8",
+    )
+
+
+def test_create_app_exposes_level_and_query_routes(monkeypatch, tmp_path) -> None:
+    config_path = tmp_path / "config.toml"
+    write_config(config_path)
+
+    class FakeService:
+        def __init__(self, settings) -> None:
+            self.settings = settings
+
+        def list_levels(self) -> list[LevelInfo]:
+            return [
+                LevelInfo(
+                    id=1,
+                    title="Level 1",
+                    description="Simple agent",
+                )
+            ]
+
+        def run_level(
+            self, level_id, session_id, user_text, hard_mode
+        ) -> AgentResponse:
+            return AgentResponse(
+                session_id=session_id,
+                response_text=f"echo:{user_text}:{hard_mode}",
+                success=False,
+                session_rotated=False,
+                level_id=level_id,
+            )
+
+    monkeypatch.setattr("main.AgentService", FakeService)
+    app = create_app(config_path)
+    client = TestClient(app)
+
+    levels_response = client.get("/api/v1/levels")
+    query_response = client.post(
+        "/api/v1/levels/query/1/",
+        json={
+            "session_id": "00000000-0000-0000-0000-000000000111",
+            "text": "hello",
+            "hard_mode": True,
+        },
+    )
+
+    assert levels_response.status_code == 200
+    assert levels_response.json() == [
+        {
+            "id": 1,
+            "title": "Level 1",
+            "description": "Simple agent",
+        }
+    ]
+    assert query_response.status_code == 200
+    assert query_response.json() == {
+        "session_id": "00000000-0000-0000-0000-000000000111",
+        "response_text": "echo:hello:True",
+        "success": False,
+        "session_rotated": False,
+        "level_id": 1,
+    }
+
+
+def test_query_route_returns_404_for_unknown_level(monkeypatch, tmp_path) -> None:
+    config_path = tmp_path / "config.toml"
+    write_config(config_path)
+
+    class FakeService:
+        def __init__(self, settings) -> None:
+            self.settings = settings
+
+        def list_levels(self) -> list[LevelInfo]:
+            return []
+
+        def run_level(
+            self, level_id, session_id, user_text, hard_mode
+        ) -> AgentResponse:
+            raise ValueError(f"Unknown level_id={level_id}")
+
+    monkeypatch.setattr("main.AgentService", FakeService)
+    app = create_app(config_path)
+    client = TestClient(app)
+
+    response = client.post(
+        "/api/v1/levels/query/99/",
+        json={
+            "session_id": "00000000-0000-0000-0000-000000000222",
+            "text": "hello",
+            "hard_mode": False,
+        },
+    )
+
+    assert response.status_code == 404
+    assert response.json() == {"detail": "Unknown level_id=99"}