import json
from pathlib import Path
from buddhagpt.corpus import segments_to_text, load_bilara_suttas

def test_segments_to_text_joins_in_key_order():
    segs = {"mn21:1.2": "second.", "mn21:1.1": "First."}
    assert segments_to_text(segs) == "First. second."

def test_segments_to_text_handles_range_keys():
    """Regression test: range keys like 'an1.142-149:1.0' must not crash.
    This would fail with the brief's original int/str-mixing sort key."""
    segs = {
        "an1.142-149:1.0": "Range intro.",
        "an1.142-149:1.1": "First sentence.",
        "an1.140:1.0": "Earlier part.",
    }
    result = segments_to_text(segs)
    # Should not raise TypeError; should contain all texts
    assert "Range intro." in result
    assert "First sentence." in result
    assert "Earlier part." in result

def test_load_bilara_suttas(tmp_path):
    """Original test: flat structure."""
    d = tmp_path / "translation/en/sujato/sutta/mn"
    d.mkdir(parents=True)
    (d / "mn21_translation-en-sujato.json").write_text(
        json.dumps({"mn21:0.1": "Middle Discourses 21", "mn21:1.1": "So I have heard."})
    )
    suttas = list(load_bilara_suttas(tmp_path))
    assert suttas[0]["uid"] == "mn21"
    assert "So I have heard." in suttas[0]["text"]
    assert suttas[0]["collection"] == "mn"

def test_load_bilara_suttas_nested_layout(tmp_path):
    """Regression test: nested structure (an/an1/, sn/sn35/, etc.)
    verifies collection is extracted from top-level nikaya, not immediate parent."""
    d = tmp_path / "translation/en/sujato/sutta/an/an1"
    d.mkdir(parents=True)
    (d / "an1.1-10_translation-en-sujato.json").write_text(
        json.dumps({"an1.1-10:0.1": "AN 1.1-10", "an1.1-10:1.1": "Content."})
    )
    suttas = list(load_bilara_suttas(tmp_path))
    assert suttas[0]["uid"] == "an1.1-10"
    assert suttas[0]["collection"] == "an", f"Expected 'an', got '{suttas[0]['collection']}'"
    assert "Content." in suttas[0]["text"]