cfc/cfc-langgraph/tests/test_health_plan.py

from app.models.health_plan import PlanTask, HealthPlanSection, HealthPlanResponse


def test_plan_task_valid():
    t = PlanTask(action_type="buy", title="购买维生素D3", dimension="wealth", frequency="once")
    assert t.frequency == "once"
    assert t.notes is None


def test_plan_task_default_frequency_daily():
    t = PlanTask(action_type="exercise", title="每周跑步3次", dimension="body")
    assert t.frequency == "daily"


def test_plan_task_invalid_action_type():
    import pytest
    from pydantic import ValidationError
    with pytest.raises(ValidationError):
        PlanTask(action_type="cook", title="做饭", dimension="body")


def test_plan_task_invalid_dimension():
    import pytest
    from pydantic import ValidationError
    with pytest.raises(ValidationError):
        PlanTask(action_type="diet", title="少油少盐", dimension="earth")


def test_plan_response_with_tasks():
    resp = HealthPlanResponse(
        overview="总览",
        sections=[HealthPlanSection(
            key="nutrition", title="营养补充", content="## 建议",
            tasks=[PlanTask(action_type="buy", title="购买鱼油", dimension="wealth", frequency="once")]
        )]
    )
    assert resp.sections[0].tasks[0].action_type == "buy"
    # 序列化后 tasks 字段存在
    data = resp.model_dump()
    assert "tasks" in data["sections"][0]