from genesis.data_models import ChapterMarker, ParsedTemplate
from genesis.writer.models import ChapterSpec
from genesis.writer.template_mapper import map_template


def _parsed():
    sections = [
        ChapterMarker(type="heading", name="はじめに", level=1),
        ChapterMarker(type="placeholder", name="section:introduction", level=0),
        ChapterMarker(type="heading", name="DB 設計", level=1),
        ChapterMarker(type="placeholder", name="section:db_design", level=0),
        ChapterMarker(type="bookmark", name="bm1", level=0),
    ]
    return ParsedTemplate(file_name="t.docx", sections=sections, placeholders={"section:introduction": "x"}, styles={})


def test_map_template_ordered_and_assigns_ids():
    specs = map_template(_parsed())
    assert [s.title for s in specs] == ["はじめに", "DB 設計"]
    assert specs[0].chapter_id == "introduction"
    assert specs[0].section_placeholder == "section:introduction"
    assert specs[1].chapter_id == "db_design"
    assert specs[1].section_placeholder == "section:db_design"
    assert all(isinstance(s, ChapterSpec) for s in specs)


def test_map_template_falls_back_without_placeholder():
    sections = [ChapterMarker(type="heading", name="附録", level=1)]
    parsed = ParsedTemplate(file_name="t.docx", sections=sections, placeholders={}, styles={})
    specs = map_template(parsed)
    assert specs[0].chapter_id == "chapter_1"
    assert specs[0].section_placeholder is None