diff --git a/tests/test_token_viz.py b/tests/test_token_viz.py index 1f0af0b2..8f9fa8f2 100644 --- a/tests/test_token_viz.py +++ b/tests/test_token_viz.py @@ -85,6 +85,10 @@ def test_gemini_cache_fields_accessible() -> None: assert hasattr(ai_client, "_GEMINI_CACHE_TTL") def test_anthropic_history_lock_accessible() -> None: - """_anthropic_history_lock must be accessible for cache hint rendering.""" - assert hasattr(ai_client, "_anthropic_history_lock") - assert hasattr(ai_client, "_anthropic_history") \ No newline at end of file + """provider_state.get_history('anthropic').lock must be accessible for cache hint rendering.""" + from src import provider_state + hist = provider_state.get_history("anthropic") + assert hasattr(hist, "lock") + assert hasattr(hist, "messages") + assert not hasattr(ai_client, "_anthropic_history_lock") + assert not hasattr(ai_client, "_anthropic_history") \ No newline at end of file