From cdd8e0a2714b8469534d05496aad4e1290ac4293 Mon Sep 17 00:00:00 2001 From: teknium1 <127238744+teknium1@users.noreply.github.com> Date: Mon, 29 Jun 2026 04:07:21 -0700 Subject: [PATCH] test(gateway): exercise last_prompt_tokens in reset-activity tests The reset-had-activity tests set total_tokens (dead state) to simulate activity; production records activity via last_prompt_tokens. Update the fixtures to match the field the fix and runtime actually use. --- tests/gateway/test_session_reset_notify.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/tests/gateway/test_session_reset_notify.py b/tests/gateway/test_session_reset_notify.py index c73ed640ccd4e..75f7d6ab38659 100644 --- a/tests/gateway/test_session_reset_notify.py +++ b/tests/gateway/test_session_reset_notify.py @@ -153,8 +153,9 @@ class TestSessionEntryReason: source = _make_source() entry1 = store.get_or_create_session(source) - # Simulate some conversation happened - entry1.total_tokens = 5000 + # Simulate some conversation happened (last_prompt_tokens is the field + # written on every turn; total_tokens is never persisted). + entry1.last_prompt_tokens = 5000 entry1.updated_at = datetime.now() - timedelta(minutes=5) store._save() @@ -245,7 +246,7 @@ class TestSessionEntryAutoResetRoundtrip: source = _make_source() entry = store.get_or_create_session(source) - entry.total_tokens = 1000 + entry.last_prompt_tokens = 1000 entry.updated_at = datetime.now() - timedelta(minutes=5) store._save()