"""Tests for unified placeholder fallback slash rules."""

from __future__ import annotations

from apps.documents.services.generation.prompts import PromptSpec
from apps.documents.services.placeholders.fallback import (
    PLACEHOLDER_FALLBACK_VALUE,
    build_docx_render_context,
    resolve_render_variable,
)
from apps.litigation_ai.services.placeholder_render_service import PlaceholderRenderService


class _DocStub:
    def get_undeclared_template_variables(self, context: dict[str, object] | None = None) -> set[str]:
        if context and "known" in context:
            return {"missing_key"}
        return set()


def test_build_docx_render_context_fills_undeclared_with_slash() -> None:
    context = build_docx_render_context(doc=_DocStub(), context={"known": "ok"})

    assert context["known"] == "ok"
    assert context["missing_key"] == PLACEHOLDER_FALLBACK_VALUE


def test_resolve_render_variable_returns_slash_for_none_and_missing() -> None:
    hit, value = resolve_render_variable({"a": None}, "a")
    assert hit is False
    assert value == PLACEHOLDER_FALLBACK_VALUE

    hit_missing, value_missing = resolve_render_variable({}, "missing")
    assert hit_missing is False
    assert value_missing == PLACEHOLDER_FALLBACK_VALUE


def test_placeholder_render_service_renders_slash_for_missing_and_none() -> None:
    rendered, stats = PlaceholderRenderService().render(
        "姓名:{name};地址:{address}",
        {"name": None},
        syntax="single",
        keep_unmatched=True,
    )

    assert rendered == f"姓名:{PLACEHOLDER_FALLBACK_VALUE};地址:{PLACEHOLDER_FALLBACK_VALUE}"
    assert stats.placeholders_found == ["name", "address"]
    assert stats.placeholders_hit == []
    assert stats.placeholders_missed == ["name", "address"]


def test_prompt_spec_render_user_message_uses_slash_for_none_and_missing() -> None:
    prompt = PromptSpec(system_prompt="s", user_template="A:{a};B:{b}", format_instructions="")

    rendered = prompt.render_user_message({"a": None})

    assert rendered == f"A:{PLACEHOLDER_FALLBACK_VALUE};B:{PLACEHOLDER_FALLBACK_VALUE}"