test(api): add tests for ai suggestion endpoints and helpers

backend/tests/test_routines_helpers.py

@@ -0,0 +1,236 @@
+from datetime import date, timedelta
+import uuid
+
+import pytest
+from sqlmodel import Session
+
+from innercontext.api.routines import (
+    _contains_minoxidil_text,
+    _is_minoxidil_product,
+    _ev,
+    _build_skin_context,
+    _build_grooming_context,
+    _build_recent_history,
+    _build_products_context,
+    _build_objectives_context,
+    _build_day_context,
+)
+from innercontext.models import (
+    Product,
+    SkinConditionSnapshot,
+    GroomingSchedule,
+    Routine,
+    RoutineStep,
+    ProductInventory,
+)
+from innercontext.models.enums import PartOfDay, GroomingAction
+
+
+def test_contains_minoxidil_text():
+    assert _contains_minoxidil_text(None) is False
+    assert _contains_minoxidil_text("") is False
+    assert _contains_minoxidil_text("some random text") is False
+    assert _contains_minoxidil_text("contains MINOXIDIL here") is True
+    assert _contains_minoxidil_text("minoksydyl 5%") is True
+
+
+def test_is_minoxidil_product():
+    # Setup product
+    p = Product(id=uuid.uuid4(), name="Test", brand="Brand", is_medication=True)
+    assert _is_minoxidil_product(p) is False
+
+    p.name = "Minoxidil 5%"
+    assert _is_minoxidil_product(p) is True
+
+    p.name = "Test"
+    p.brand = "Brand with minoksydyl"
+    assert _is_minoxidil_product(p) is True
+
+    p.brand = "Brand"
+    p.line_name = "Minoxidil Line"
+    assert _is_minoxidil_product(p) is True
+
+    p.line_name = None
+    p.usage_notes = "Use minoxidil daily"
+    assert _is_minoxidil_product(p) is True
+
+    p.usage_notes = None
+    p.inci = ["water", "minoxidil"]
+    assert _is_minoxidil_product(p) is True
+
+    p.inci = None
+    p.actives = [{"name": "minoxidil", "strength": "5%"}]
+    assert _is_minoxidil_product(p) is True
+
+    # As Pydantic model representation isn't exactly a dict in db sometimes, we just test dict
+    p.actives = [{"name": "Retinol", "strength": "1%"}]
+    assert _is_minoxidil_product(p) is False
+
+
+def test_ev():
+    class DummyEnum:
+        value = "dummy"
+
+    assert _ev(None) == ""
+    assert _ev(DummyEnum()) == "dummy"
+    assert _ev("string") == "string"
+
+
+def test_build_skin_context(session: Session):
+    # Empty
+    assert _build_skin_context(session) == "SKIN CONDITION: no data\n"
+
+    # With data
+    snap = SkinConditionSnapshot(
+        id=uuid.uuid4(),
+        snapshot_date=date.today(),
+        overall_state="good",
+        hydration_level=4,
+        barrier_state="intact",
+        active_concerns=["acne", "dryness"],
+        priorities=["hydration"],
+        notes="Feeling good"
+    )
+    session.add(snap)
+    session.commit()
+
+    ctx = _build_skin_context(session)
+    assert "SKIN CONDITION (snapshot from" in ctx
+    assert "Overall state: good" in ctx
+    assert "Hydration: 4/5" in ctx
+    assert "Barrier: intact" in ctx
+    assert "Active concerns: acne, dryness" in ctx
+    assert "Priorities: hydration" in ctx
+    assert "Notes: Feeling good" in ctx
+
+
+def test_build_grooming_context(session: Session):
+    assert _build_grooming_context(session) == "GROOMING SCHEDULE: none\n"
+
+    sch = GroomingSchedule(
+        id=uuid.uuid4(),
+        day_of_week=0,
+        action="shaving_oneblade",
+        notes="Morning"
+    )
+    session.add(sch)
+    session.commit()
+
+    ctx = _build_grooming_context(session)
+    assert "GROOMING SCHEDULE:" in ctx
+    assert "poniedziałek: shaving_oneblade (Morning)" in ctx
+
+    # Test weekdays filter
+    ctx2 = _build_grooming_context(session, weekdays=[1]) # not monday
+    assert "(no entries for specified days)" in ctx2
+
+
+def test_build_recent_history(session: Session):
+    assert _build_recent_history(session) == "RECENT ROUTINES: none\n"
+
+    r = Routine(
+        id=uuid.uuid4(),
+        routine_date=date.today(),
+        part_of_day="am"
+    )
+    session.add(r)
+    p = Product(id=uuid.uuid4(), name="Cleanser", category="cleanser", brand="Test", recommended_time="both", leave_on=False, product_effect_profile={})
+    session.add(p)
+    session.commit()
+
+    s1 = RoutineStep(id=uuid.uuid4(), routine_id=r.id, order_index=1, product_id=p.id)
+    s2 = RoutineStep(id=uuid.uuid4(), routine_id=r.id, order_index=2, action_type="shaving_razor")
+    # Step with non-existent product
+    s3 = RoutineStep(id=uuid.uuid4(), routine_id=r.id, order_index=3, product_id=uuid.uuid4())
+
+    session.add_all([s1, s2, s3])
+    session.commit()
+
+    ctx = _build_recent_history(session)
+    assert "RECENT ROUTINES:" in ctx
+    assert "AM:" in ctx
+    assert "cleanser [" in ctx
+    assert "action: shaving_razor" in ctx
+    assert "unknown [" in ctx
+
+
+def test_build_products_context(session: Session):
+    p1 = Product(
+        id=uuid.uuid4(),
+        name="Regaine",
+        category="serum",
+        is_medication=True,
+        brand="J&J", recommended_time="both", leave_on=True, product_effect_profile={}
+    )
+    p2 = Product(
+        id=uuid.uuid4(),
+        name="Sunscreen",
+        category="spf", brand="Test", leave_on=True,
+        recommended_time="am",
+        pao_months=6,
+        product_effect_profile={"hydration_immediate": 2, "exfoliation_strength": 0},
+        incompatible_with=[{"target": "retinol", "scope": "same_routine"}],
+        context_rules={"safe_after_shaving": False},
+        min_interval_hours=12,
+        max_frequency_per_week=7
+    )
+    session.add_all([p1, p2])
+    session.commit()
+
+    # Inventory
+    inv1 = ProductInventory(
+        id=uuid.uuid4(),
+        product_id=p2.id,
+        is_opened=True,
+        opened_at=date.today() - timedelta(days=10),
+        expiry_date=date.today() + timedelta(days=365)
+    )
+    inv2 = ProductInventory(
+        id=uuid.uuid4(),
+        product_id=p2.id,
+        is_opened=False
+    )
+    session.add_all([inv1, inv2])
+    session.commit()
+
+    # Usage
+    r = Routine(id=uuid.uuid4(), routine_date=date.today(), part_of_day="am")
+    session.add(r)
+    session.commit()
+    s = RoutineStep(id=uuid.uuid4(), routine_id=r.id, order_index=1, product_id=p2.id)
+    session.add(s)
+    session.commit()
+
+    ctx = _build_products_context(session, time_filter="am", reference_date=date.today())
+    # p1 is medication but not minoxidil (wait, Regaine name doesn't contain minoxidil!) -> skipped
+    assert "Regaine" not in ctx
+    
+    # Let's fix p1 to be minoxidil
+    p1.name = "Regaine Minoxidil"
+    session.add(p1)
+    session.commit()
+    
+    ctx = _build_products_context(session, time_filter="am", reference_date=date.today())
+    assert "Regaine Minoxidil" in ctx
+    assert "Sunscreen" in ctx
+    assert "inventory_status={active:2,opened:1,sealed:1}" in ctx
+    assert "nearest_open_expiry=" in ctx
+    assert "nearest_open_pao_deadline=" in ctx
+    assert "pao_months=6" in ctx
+    assert "effects={'hydration_immediate': 2}" in ctx
+    assert "incompatible_with=['avoid retinol (same_routine)']" in ctx
+    assert "context_rules={'safe_after_shaving': False}" in ctx
+    assert "min_interval_hours=12" in ctx
+    assert "max_frequency_per_week=7" in ctx
+    assert "used_in_last_7_days=1" in ctx
+
+
+def test_build_objectives_context():
+    assert _build_objectives_context(False) == ""
+    assert "improve beard" in _build_objectives_context(True)
+
+
+def test_build_day_context():
+    assert _build_day_context(None) == ""
+    assert "Leaving home: yes" in _build_day_context(True)
+    assert "Leaving home: no" in _build_day_context(False)