diff --git a/tests/test_streaming_agent_cache_lifecycle.py b/tests/test_streaming_agent_cache_lifecycle.py index 815b9fcba..1cf513be4 100644 --- a/tests/test_streaming_agent_cache_lifecycle.py +++ b/tests/test_streaming_agent_cache_lifecycle.py @@ -97,6 +97,7 @@ def test_identity_mismatch_cache_evictions_close_entries_outside_cache_lock(): "_stale_runtime_entry = SESSION_AGENT_CACHE.pop(session_id, None)", "_skipped_agent_migration_entry = _cached_entry", "evicted_cached_entry = _cfg.SESSION_AGENT_CACHE.pop(sid, None)", + "_evicted_entry = SESSION_AGENT_CACHE.pop(session_id, None)", ] for marker in expected_markers: assert marker in src @@ -106,6 +107,7 @@ def test_identity_mismatch_cache_evictions_close_entries_outside_cache_lock(): "_close_cached_agent_entry_at_session_boundary(session_id, _stale_runtime_entry)", "_close_cached_agent_entry_at_session_boundary(old_sid, _skipped_agent_migration_entry)", "_close_cached_agent_entry_at_session_boundary(sid, evicted_cached_entry)", + "_close_cached_agent_entry_at_session_boundary(session_id, _evicted_entry)", ] lines = src.splitlines() for marker in close_markers: