haneulai
f2bb131884
Some checks are pending
CI / Crawler (Python ${{ matrix.python-version }}) (3.10) (push) Waiting to run
CI / Crawler (Python ${{ matrix.python-version }}) (3.11) (push) Waiting to run
CI / API (Python 3.11) (push) Waiting to run
CI / Database migration (push) Waiting to run
CI / App web build (Node 20) (push) Waiting to run
108 lines
3.4 KiB
Python
108 lines
3.4 KiB
Python
import unittest
|
|
|
|
from fastapi.testclient import TestClient
|
|
|
|
from app.core.database import get_db
|
|
from app.main import app
|
|
from tests.test_plants import FakeDb
|
|
|
|
|
|
class FailingDb:
|
|
def table(self, _table_name):
|
|
raise RuntimeError("database unavailable")
|
|
|
|
|
|
class ApiContractTest(unittest.TestCase):
|
|
def setUp(self):
|
|
self.db = FakeDb()
|
|
app.dependency_overrides[get_db] = lambda: self.db
|
|
self.client = TestClient(app, raise_server_exceptions=False)
|
|
|
|
def tearDown(self):
|
|
app.dependency_overrides.clear()
|
|
|
|
def test_health_checks_real_database_query(self):
|
|
response = self.client.get("/health")
|
|
|
|
self.assertEqual(200, response.status_code)
|
|
self.assertTrue(response.json()["supabase_connected"])
|
|
self.assertEqual(1, len(self.db.queries))
|
|
self.assertEqual("plants", self.db.queries[0].table_name)
|
|
|
|
def test_health_returns_503_when_database_query_fails(self):
|
|
app.dependency_overrides[get_db] = lambda: FailingDb()
|
|
|
|
response = self.client.get("/health")
|
|
|
|
self.assertEqual(503, response.status_code)
|
|
self.assertEqual(
|
|
"데이터베이스 연결 상태를 확인할 수 없습니다.",
|
|
response.json()["detail"],
|
|
)
|
|
|
|
def test_string_plant_detail_path_returns_200(self):
|
|
response = self.client.get("/plants/1/nrems-03")
|
|
|
|
self.assertEqual(200, response.status_code)
|
|
self.assertEqual("nrems-03", response.json()["data"]["plant"]["id"])
|
|
|
|
def test_invalid_comparison_date_returns_400(self):
|
|
response = self.client.get(
|
|
"/plants/stats/comparison?period=day&date=2026-8-07"
|
|
)
|
|
|
|
self.assertEqual(400, response.status_code)
|
|
|
|
def test_invalid_comparison_company_returns_422(self):
|
|
response = self.client.get(
|
|
"/plants/stats/comparison?period=day&company_id=0"
|
|
)
|
|
|
|
self.assertEqual(422, response.status_code)
|
|
|
|
def test_invalid_month_returns_422(self):
|
|
response = self.client.get(
|
|
"/plants/nrems-03/stats?period=day&year=2026&month=13"
|
|
)
|
|
|
|
self.assertEqual(422, response.status_code)
|
|
|
|
def test_unknown_plant_stats_returns_404(self):
|
|
app.dependency_overrides[get_db] = lambda: FakeDb(plant_exists=False)
|
|
|
|
response = self.client.get(
|
|
"/plants/missing/stats?period=day&year=2026&month=8"
|
|
)
|
|
|
|
self.assertEqual(404, response.status_code)
|
|
|
|
def test_openapi_declares_string_plant_id_and_response_models(self):
|
|
schema = self.client.get("/openapi.json").json()
|
|
detail_operation = schema["paths"]["/plants/{company_id}/{plant_id}"]["get"]
|
|
plant_id_param = next(
|
|
item for item in detail_operation["parameters"]
|
|
if item["name"] == "plant_id"
|
|
)
|
|
|
|
self.assertEqual("string", plant_id_param["schema"]["type"])
|
|
self.assertIn("200", detail_operation["responses"])
|
|
self.assertIn(
|
|
"PlantDetailResponse",
|
|
str(detail_operation["responses"]["200"]),
|
|
)
|
|
self.assertIn(
|
|
"DetailedHealthResponse",
|
|
str(schema["paths"]["/health"]["get"]["responses"]["200"]),
|
|
)
|
|
self.assertIn(
|
|
"UploadResponse",
|
|
str(
|
|
schema["paths"]["/plants/{plant_id}/upload"]
|
|
["post"]["responses"]["200"]
|
|
),
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|