56 lines
1.7 KiB
Python
56 lines
1.7 KiB
Python
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import asyncio
|
||
|
|
import json
|
||
|
|
import unittest
|
||
|
|
from unittest.mock import AsyncMock, patch
|
||
|
|
|
||
|
|
from fastapi.responses import JSONResponse
|
||
|
|
|
||
|
|
from app import create_app
|
||
|
|
from settings import get_settings
|
||
|
|
|
||
|
|
|
||
|
|
class AppTestCase(unittest.TestCase):
|
||
|
|
def setUp(self) -> None:
|
||
|
|
get_settings.cache_clear()
|
||
|
|
self.app = create_app()
|
||
|
|
|
||
|
|
def tearDown(self) -> None:
|
||
|
|
get_settings.cache_clear()
|
||
|
|
|
||
|
|
def _get_route_response(self, path: str) -> JSONResponse:
|
||
|
|
route = next(route for route in self.app.routes if getattr(route, "path", None) == path)
|
||
|
|
return asyncio.run(route.endpoint())
|
||
|
|
|
||
|
|
def test_health(self) -> None:
|
||
|
|
response = self._get_route_response("/health")
|
||
|
|
|
||
|
|
self.assertEqual(response.status_code, 200)
|
||
|
|
self.assertEqual(json.loads(response.body), {"status": "ok"})
|
||
|
|
|
||
|
|
def test_prototype_payload_shape(self) -> None:
|
||
|
|
payload = {
|
||
|
|
"hero": {"title": "Robot U"},
|
||
|
|
"featured_courses": [],
|
||
|
|
"recent_posts": [],
|
||
|
|
"recent_discussions": [],
|
||
|
|
"upcoming_events": [],
|
||
|
|
"source_of_truth": [],
|
||
|
|
}
|
||
|
|
with patch("app.build_live_prototype_payload", new=AsyncMock(return_value=payload)):
|
||
|
|
response = self._get_route_response("/api/prototype")
|
||
|
|
payload = json.loads(response.body)
|
||
|
|
|
||
|
|
self.assertEqual(response.status_code, 200)
|
||
|
|
self.assertIn("hero", payload)
|
||
|
|
self.assertIn("featured_courses", payload)
|
||
|
|
self.assertIn("recent_posts", payload)
|
||
|
|
self.assertIn("recent_discussions", payload)
|
||
|
|
self.assertIn("upcoming_events", payload)
|
||
|
|
self.assertIn("source_of_truth", payload)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
unittest.main()
|