"""Tests Progressionsgraph-Auflösung für Planungs-KI (Phase C1)."""
from planning_exercise_progression import (
    edge_matches_anchor_from,
    filter_outgoing_progression_edges,
    parse_successors_from_edges,
    rank_progression_graph_rows,
)


def test_edge_matches_anchor_from_generic_edge():
    assert edge_matches_anchor_from({"from_exercise_variant_id": None}, None)
    assert edge_matches_anchor_from({"from_exercise_variant_id": None}, 5)


def test_edge_matches_anchor_from_variant_specific():
    assert edge_matches_anchor_from({"from_exercise_variant_id": 3}, 3)
    assert not edge_matches_anchor_from({"from_exercise_variant_id": 3}, None)
    assert not edge_matches_anchor_from({"from_exercise_variant_id": 3}, 4)


def test_filter_outgoing_progression_edges():
    edges = [
        {"from_exercise_variant_id": None, "to_exercise_id": 10},
        {"from_exercise_variant_id": 2, "to_exercise_id": 11},
    ]
    filtered = filter_outgoing_progression_edges(edges, from_variant_id=2)
    assert len(filtered) == 2
    only_generic = filter_outgoing_progression_edges(edges, from_variant_id=None)
    assert len(only_generic) == 1
    assert only_generic[0]["to_exercise_id"] == 10


def test_parse_successors_from_edges():
    ids, notes, variants = parse_successors_from_edges(
        [
            {
                "to_exercise_id": 20,
                "to_exercise_variant_id": 7,
                "notes": " leicht ",
            },
            {"to_exercise_id": 21, "to_exercise_variant_id": None, "notes": ""},
        ]
    )
    assert ids == {20, 21}
    assert notes[20] == "leicht"
    assert variants[20] == 7
    assert variants[21] is None


def test_rank_progression_graph_rows_prefers_variant_match():
    rows = [
        {"id": 1, "variant_match_count": 0, "outgoing_count": 5},
        {"id": 2, "variant_match_count": 2, "outgoing_count": 1},
    ]
    best = rank_progression_graph_rows(rows)
    assert best["id"] == 2