"""Prompt-cache ordering tests: stable instructions first, time in user messages."""

from pathlib import Path
import sys
import unittest
from types import SimpleNamespace
from unittest.mock import AsyncMock, patch

BASE = Path(__file__).resolve().parents[1]
sys.path[:0] = [str(BASE / "bot-utility"), str(BASE)]

from agent import ToolContext
from chat import engine


class PromptCacheOrderTests(unittest.IsolatedAsyncioTestCase):
    def build_private(self):
        channel = SimpleNamespace(name="tg", is_admin=lambda uid: True)
        ctx = ToolContext(channel=channel, msg_type="private", user_id=42, target_id=10001, is_admin=True)
        with patch.object(engine, "user_persona_prompt", return_value="PERSONA"), \
             patch.object(engine, "load_user_status", return_value={"pause": False, "sleep": False}), \
             patch.object(engine.memory_store, "get_relation", return_value=None), \
             patch.object(engine.memory_store, "list_entities", return_value=[]), \
             patch.object(engine.memory_store, "recall_facts", return_value=[(1, "长期事实", 5)]), \
             patch.object(engine.memory_store, "migrate_legacy"), \
             patch.object(engine, "get_history_within_budget", return_value=["最近历史"]), \
             patch.object(engine.notebook.notebook, "context_for_chat", return_value="笔记本背景"), \
             patch.object(engine, "_sticker_prompt_summary", return_value="贴纸场景"):
            return engine.build_system_prompt(42, query="当前问题", tool_context=ctx)

    def test_prompt_budget_is_128k(self):
        self.assertEqual(engine.PROMPT_TOKEN_BUDGET, 131072)

    def test_private_stable_instructions_precede_dynamic_sections(self):
        prompt = self.build_private()
        self.assertNotIn("现在是 ", prompt)
        agent = prompt.index("当你回复时")
        memory = prompt.index("【长期记忆】")
        status = prompt.index("当前用户状态")
        history = prompt.index("以下是最近你和用户的聊天记录")
        self.assertLess(agent, memory)
        self.assertLess(memory, status)
        self.assertLess(status, history)
        self.assertIn("你还可以调用以下 utility 工具", prompt[:history])

    def test_sticker_prompt_prefers_meaning_over_description(self):
        expected = "理解其情绪、含义和使用场景即可，不要主动描述画面内容"
        self.assertIn(expected, engine.load_instruction("agent_instructions.txt"))
        self.assertIn(expected, engine.load_instruction("group_instructions.txt"))

    def test_time_is_prepended_to_user_message(self):
        message = engine._message_with_time("用户输入")
        self.assertTrue(message.startswith("现在是 "))
        self.assertTrue(message.endswith("\n用户输入"))

    async def test_group_prompt_has_stable_prefix_and_time_in_user_message(self):
        response = '{"message":[{"content":"ok"}],"tools":[],"need_reasoning":false}'
        model = AsyncMock(return_value=response)
        with patch.object(engine, "get_recent_group_history", return_value=["群历史"]), \
             patch.object(engine, "call_deepseek", new=model), \
             patch.object(engine, "send_group_message", new=AsyncMock(return_value=True)), \
             patch.object(engine, "append_group_history"):
            await engine._call_group_model_and_reply_impl(None, 1, "群消息", 0)
        system, user = model.await_args.args[:2]
        self.assertNotIn("现在是 ", system)
        self.assertTrue(user.startswith("现在是 "))
        self.assertLess(system.index("当你回复时"), system.index("以下是本群的最近聊天记录"))


if __name__ == "__main__":
    unittest.main()