From f3cefccefc9efde0cfb4e9b17a66a1806a199be1 Mon Sep 17 00:00:00 2001 From: worldmozara Date: Fri, 24 Jul 2026 19:41:37 +0800 Subject: [PATCH] test/memory: cover session context usage fields --- tests/test_memory/test_store.py | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) diff --git a/tests/test_memory/test_store.py b/tests/test_memory/test_store.py index f984466..f0b4502 100644 --- a/tests/test_memory/test_store.py +++ b/tests/test_memory/test_store.py @@ -34,6 +34,27 @@ async def test_create_session_uses_default_user(store: MemoryStore) -> None: assert session.uid == user.uid +async def test_session_context_usage_roundtrip(store: MemoryStore) -> None: + session = await store.create_session(name="usage") + row = await store.update_session_context_usage( + session.sid, + last_prompt_tokens=120_000, + peak_prompt_tokens=150_000, + last_completion_tokens=400, + usage_source="api", + reminder_last_band=1, + ) + assert row.last_prompt_tokens == 120_000 + assert row.peak_prompt_tokens == 150_000 + assert row.last_completion_tokens == 400 + assert row.usage_source == "api" + assert row.reminder_last_band == 1 + loaded = await store.get_session(session.sid) + assert loaded is not None + assert loaded.peak_prompt_tokens == 150_000 + assert loaded.reminder_last_band == 1 + + async def test_append_and_list_messages(store: MemoryStore) -> None: session = await store.create_session() user_msg = UserChatMessage(content="hello")