"""Tests Phase C2: Varianten-Metadaten in Planungs-Treffern."""
from planning_exercise_suggest import _enrich_planning_hits_with_variant_meta


class _Cur:
    def __init__(self, rows):
        self._rows = rows
        self.last_query = ''

    def execute(self, query, params=None):
        self.last_query = query
        self._params = params

    def fetchall(self):
        q = self.last_query
        if 'FROM exercise_variants WHERE id IN' in q:
            return [
                {'id': 7, 'variant_name': 'Leicht'},
            ]
        if 'FROM exercise_variants' in q and 'exercise_id IN' in q:
            return [
                {
                    'exercise_id': 10,
                    'id': 7,
                    'variant_name': 'Leicht',
                    'sequence_order': 1,
                },
                {
                    'exercise_id': 10,
                    'id': 8,
                    'variant_name': 'Schwer',
                    'sequence_order': 2,
                },
            ]
        return []


def test_enrich_planning_hits_with_variant_meta():
    hits = [{'id': 10, 'title': 'Test', 'suggested_variant_id': 7}]
    out = _enrich_planning_hits_with_variant_meta(_Cur([]), hits)
    assert out[0]['suggested_variant_name'] == 'Leicht'
    assert len(out[0]['variants']) == 2
    assert out[0]['variants'][0]['id'] == 7