import pytest

from app.memory.context_builder import ContextBuilder
from app.memory.summarizer import Summarizer


@pytest.mark.asyncio
async def test_summarization_trigger_uses_token_limit() -> None:
    class Memory:
        async def context_messages(self, *args, **kwargs):
            return [type("Message", (), {"text": "many tokens " * 20})()]

    summarizer = Summarizer(Memory(), ContextBuilder(1000), None, trigger_tokens=5)
    assert await summarizer.needs_summary(1, None)