"""일정 응답 해석 — 모델이 준 JSON 에서 화면에 설 수 있는 코스만 남긴다. DB 도 네트워크도 쓰지 않는 순수 단위 테스트다. ★ 스키마 전체를 검증하지 않는다. 항목 모양의 단일 출처는 `shared/lib/section-data.ts` 이고, 그 모양을 파이썬에 한 벌 더 적으면 프론트가 필드를 늘린 날 서버가 조용히 떨어뜨린다 (`grounding/story.py` 머리주석과 같은 판단). 여기서는 "화면에 설 수 있는가" 만 본다. """ import json from services.grounding import itinerary as grounding def _payload(items: list, *, fence: bool = False, search: list | None = None) -> dict: envelope = {"kind": "itinerary", "version": 1, "title": "추천 일정", "items": items} text = json.dumps(envelope, ensure_ascii=False) if fence: text = f"```json\n{text}\n```" return { "choices": [{"message": {"content": text}}], "search_results": search if search is not None else [ {"title": "군산문화관광", "url": "https://www.gunsan.go.kr/tour/"} ], } def _course(name: str, stops: list[str], *, days: int = 2) -> dict: per_day = max(1, len(stops) // days) chunks = [stops[i:i + per_day] for i in range(0, len(stops), per_day)][:days] return { "name": name, "duration": "1박 2일", "audience": "누구에게나", "why": "이유 한 문장.", "days": [ { "label": f"{i + 1}일차", "startTime": "14:00", "stops": [ {"name": s, "minutes": 60, "moveMinutes": 10, "searchQuery": s, "latitude": 35.99, "longitude": 126.71} for s in chunk ], } for i, chunk in enumerate(chunks) ], } def test_accepts_five_distinct_courses(): """검증: 정상 응답 5개. 기대결과: 전부 채택되고 버린 것이 없다.""" items = [_course(f"코스{i}", [f"장소{i}-{j}" for j in range(4)]) for i in range(5)] courses, dropped = grounding.parse_courses(_payload(items), "1박 2일") assert len(courses) == 5 assert dropped == [] assert [c["name"] for c in courses] == [f"코스{i}" for i in range(5)] def test_strips_code_fence(): """검증: 코드펜스를 두르고 온 응답. 기대결과: 규칙 1 로 금지했지만 모델은 종종 어긴다 — 벗겨서 읽는다.""" items = [_course("코스", ["가", "나"])] courses, dropped = grounding.parse_courses(_payload(items, fence=True), "1박 2일") assert len(courses) == 1 def test_forces_duration_even_if_model_wrote_something_else(): """검증: 모델이 duration 을 다르게 적은 코스. 기대결과: 우리가 요청한 기간으로 덮어쓴다 — 이 값이 화면 탭을 가르므로 틀리면 탭이 갈린다.""" item = _course("코스", ["가", "나"]) item["duration"] = "반나절" courses, _ = grounding.parse_courses(_payload([item]), "1박 2일") assert courses[0]["duration"] == "1박 2일" def test_drops_course_without_name(): """검증: 이름 없는 코스. 기대결과: 버린다 — 제목 없는 카드가 된다.""" item = _course("", ["가", "나"]) courses, dropped = grounding.parse_courses(_payload([item]), "1박 2일") assert courses == [] assert dropped and "name" in dropped[0] def test_drops_course_without_stops(): """검증: days 가 있어도 정거장이 하나도 없는 코스. 기대결과: 버린다 — 발행본 ItinerarySection 이 어차피 걸러 빈 탭을 만든다.""" item = {"name": "빈 코스", "duration": "1박 2일", "days": [{"label": "1일차", "stops": []}]} courses, dropped = grounding.parse_courses(_payload([item]), "1박 2일") assert courses == [] assert dropped def test_drops_a_course_whose_stop_set_duplicates_an_earlier_one(): """검증: 앞 코스와 정거장 집합이 완전히 같은 코스(순서만 다름). 기대결과: 뒤에 온 것을 버린다 — 손님 눈에는 같은 코스 둘이다. ★ 정거장 '겹침' 자체는 허용이다(2026-09-11 결정). 집합이 **같을 때**만 버린다.""" first = _course("원도심 코스", ["가", "나", "다", "라"]) same = _course("미식 코스", ["라", "다", "나", "가"]) # 순서만 다르다 overlap = _course("자연 코스", ["가", "나", "마", "바"]) # 절반 겹치지만 다른 코스다 courses, dropped = grounding.parse_courses(_payload([first, same, overlap]), "1박 2일") assert [c["name"] for c in courses] == ["원도심 코스", "자연 코스"] assert dropped and "미식 코스" in dropped[0] def test_attaches_search_result_as_source_when_model_gave_none(): """검증: 코스에 source 가 없을 때. 기대결과: Perplexity 가 실제로 읽은 첫 출처를 붙인다 — 모델 답변보다 이게 실재한다. ★ source 가 아예 없으면 발행본 카드 하단에 빈 점선 띠가 남는다(ItinerarySection 껍데기).""" courses, _ = grounding.parse_courses(_payload([_course("코스", ["가", "나"])]), "1박 2일") assert courses[0]["source"]["url"] == "https://www.gunsan.go.kr/tour/" def test_keeps_model_source_when_it_is_a_real_url(): """검증: 모델이 준 source 가 http 로 시작할 때. 기대결과: 그걸 쓴다.""" item = _course("코스", ["가", "나"]) item["source"] = {"name": "군산시", "url": "https://www.gunsan.go.kr/index.gunsan"} courses, _ = grounding.parse_courses(_payload([item]), "1박 2일") assert courses[0]["source"]["url"] == "https://www.gunsan.go.kr/index.gunsan" def test_course_survives_without_any_source(): """검증: 모델 source 도 없고 search_results 도 빈 응답. 기대결과: 코스는 살린다 — 출처가 없다고 일정을 버리면 화면이 통째로 빈다. (지역 이야기와 다른 판단이다. 저쪽은 '사실' 이고 이쪽은 '제안' 이다.)""" items = [_course("코스", ["가", "나"])] courses, _ = grounding.parse_courses(_payload(items, search=[]), "1박 2일") assert len(courses) == 1 assert "source" not in courses[0] def test_returns_empty_on_non_json(): """검증: JSON 이 아닌 응답. 기대결과: 빈 목록 + 이유. 예외를 던지지 않는다 — 잡이 죽으면 다른 기간도 못 받는다.""" payload = {"choices": [{"message": {"content": "죄송합니다, 일정을 만들 수 없습니다."}}]} courses, dropped = grounding.parse_courses(payload, "1박 2일") assert courses == [] assert dropped def test_returns_empty_when_items_is_not_a_list(): """검증: items 가 배열이 아닌 응답.""" payload = {"choices": [{"message": {"content": json.dumps({"items": "없음"})}}]} courses, dropped = grounding.parse_courses(payload, "1박 2일") assert courses == [] assert dropped