import pytest

from tests._fake_llm import FakeLLMProvider

CTX = {
    "projects": [
        {"id": "life-trip", "name": "여행 — 한국", "folder_id": "life"},
        {"id": "life-fam", "name": "가족", "folder_id": "life"},
        {"id": "onb", "name": "온보딩 리디자인", "folder_id": "work"},
    ]
}


@pytest.mark.parametrize(
    "raw,ctype",
    [
        ("수요일 11시 자전거 수리 맡기기", "event"),  # 시각 명시
        ("다음 주에 비행기 티켓 사기", "task"),  # 행동+기한
        ("온보딩에 애니메이션 넣으면 어떨까", "idea"),  # 막연
    ],
)
def test_heuristic_type(raw, ctype):
    c = FakeLLMProvider().classify_capture(raw, CTX)
    assert c.type == ctype
