from pathlib import Path

from gfa_core.mas.agents.deepseek_summary import (
    _validate_report,
    collect_figure_inventory,
    select_primary_scope,
)


def test_select_primary_scope_prefers_target_plus_reference() -> None:
    summaries = [
        {"stage": "identify", "scope_id": "YABBY.target-BNA.target_only"},
        {"stage": "phylogeny", "scope_id": "YABBY.target-BNA.ref-At.target_plus_reference"},
        {"stage": "motif", "scope_id": "YABBY.target-BNA.ref-At.target_plus_reference"},
    ]

    assert select_primary_scope(summaries) == "YABBY.target-BNA.ref-At.target_plus_reference"


def test_collect_figure_inventory_deduplicates_identical_aliases(tmp_path: Path) -> None:
    final_dir = tmp_path / "final"
    final_dir.mkdir()
    (final_dir / "A.tfbs_motif_source_summary.png").write_bytes(b"same-image")
    (final_dir / "A.tfbs.motif_source_summary.png").write_bytes(b"same-image")
    (final_dir / "A.pep_properties.png").write_bytes(b"different-image")

    figures = collect_figure_inventory(final_dir, [], scope_id="scope")

    assert len(figures) == 2
    assert {item["stage"] for item in figures} == {"tfbs", "pep"}


def test_report_validation_requires_every_figure_marker() -> None:
    figures = [
        {"marker": "[[FIG:a.png]]"},
        {"marker": "[[FIG:b.png]]"},
    ]
    report = (
        "# 报告\n\n任务 job_1\n\n"
        + "\n\n".join(f"## 章节{i}\n内容" for i in range(1, 10))
        + "\n\n[[FIG:a.png]]\n"
        + ("通俗解读。" * 800)
    )

    validation = _validate_report(report, job_id="job_1", figures=figures)

    assert validation["passed"] is False
    assert validation["missing_figure_markers"] == ["[[FIG:b.png]]"]