"""记忆上下文组装测试。"""

from datetime import UTC, datetime, timedelta

from app.models.memory import Memory
from app.modules.memory.context import (
    MemoryRetrievalResult,
    build_memory_context,
)


def _make_memory(
    *,
    importance: float = 0.5,
    hours_ago: float = 1.0,
    title: str = "测试记忆",
    content: str = "这是一条测试记忆内容。",
) -> Memory:
    return Memory(
        id=f"mem_{title}",
        character_id="char_1",
        type="observation",
        title=title,
        content=content,
        importance=importance,
        created_at=datetime.now(UTC) - timedelta(hours=hours_ago),
        source="test",
        confidence=1.0,
        sensitivity="low",
        evidence_ids_json="[]",
        status="active",
    )


def test_empty_memories_returns_none() -> None:
    result = MemoryRetrievalResult(memories=[])
    assert build_memory_context(result) is None


def test_single_memory_returns_text() -> None:
    mem = _make_memory()
    result = MemoryRetrievalResult(memories=[mem])
    text = build_memory_context(result)
    assert text is not None
    assert "相关记忆：" in text
    assert "测试记忆" in text


def test_high_importance_memory_comes_first() -> None:
    low = _make_memory(importance=0.3, title="不重要的")
    high = _make_memory(importance=0.9, title="重要的")
    result = MemoryRetrievalResult(memories=[low, high])
    text = build_memory_context(result)
    assert text is not None
    important_pos = text.index("重要的")
    trivial_pos = text.index("不重要的")
    assert important_pos < trivial_pos, "重要性高的应该排前面"


def test_memories_are_sorted_by_importance_descending() -> None:
    """传入任意顺序，输出始终重要性降序。"""
    a = _make_memory(importance=0.5, title="中")
    b = _make_memory(importance=0.1, title="低")
    c = _make_memory(importance=0.9, title="高")
    result = MemoryRetrievalResult(memories=[a, b, c])
    text = build_memory_context(result)
    assert text is not None
    positions = {title: text.index(title) for title in ["高", "中", "低"]}
    assert positions["高"] < positions["中"]
    assert positions["中"] < positions["低"]


def test_importance_label_appears() -> None:
    high = _make_memory(importance=0.9, title="重要记忆", content="重要内容")
    medium = _make_memory(importance=0.5, title="普通记忆", content="普通内容")
    low = _make_memory(importance=0.2, title="低优先级", content="低优先级内容")

    result = MemoryRetrievalResult(memories=[high, medium, low])
    text = build_memory_context(result)
    assert text is not None
    assert "★重要" in text
    assert "●一般" in text
    assert "○参考" in text


def test_time_labels() -> None:
    just_now = _make_memory(title="刚刚发生", hours_ago=0.1)
    hours_ago = _make_memory(title="几小时前", hours_ago=5)
    days_ago = _make_memory(title="几天前", hours_ago=72)
    result = MemoryRetrievalResult(memories=[just_now, hours_ago, days_ago])
    text = build_memory_context(result)
    assert text is not None
    assert "刚刚" in text
    assert "5小时前" in text or "几小时前" in text
    assert "3天前" in text


def test_truncation_respects_max_length() -> None:
    content = "A" * 500
    mem = _make_memory(content=content, title="一长内容")
    result = MemoryRetrievalResult(memories=[mem] * 20)
    text = build_memory_context(result, max_length=500)
    assert text is not None
    assert len(text) <= 550  # max_length + header + padding
    assert "已截断" in text


def test_large_content_not_exceeding_max_length() -> None:
    """大量记忆时，内容不应超过 max_length 太多。"""
    mems = [_make_memory(content="B" * 200, title=f"记忆{i}", importance=0.5) for i in range(50)]
    result = MemoryRetrievalResult(memories=mems)
    text = build_memory_context(result, max_length=1000)
    assert text is not None
    assert len(text) <= 1200  # 1000 + header + some padding