Ed_
e747a783a5
This change adds a label to the Provider panel to show the count and total size of active Gemini caches when the Gemini provider is selected. This information is hidden for other providers.
112 lines
4.1 KiB
Python
112 lines
4.1 KiB
Python
import pytest
|
|
from unittest.mock import patch, MagicMock
|
|
import importlib.util
|
|
import sys
|
|
import dearpygui.dearpygui as dpg
|
|
|
|
# Load gui.py as a module for testing
|
|
spec = importlib.util.spec_from_file_location("gui", "gui.py")
|
|
gui = importlib.util.module_from_spec(spec)
|
|
sys.modules["gui"] = gui
|
|
spec.loader.exec_module(gui)
|
|
from gui import App
|
|
|
|
@pytest.fixture
|
|
def app_instance():
|
|
"""
|
|
Fixture to create an instance of the App class for testing.
|
|
It creates a real DPG context but mocks functions that would
|
|
render a window or block execution.
|
|
"""
|
|
dpg.create_context()
|
|
|
|
# Patch only the functions that would show a window or block,
|
|
# and the App methods that rebuild UI on init.
|
|
with patch('dearpygui.dearpygui.create_viewport'), \
|
|
patch('dearpygui.dearpygui.setup_dearpygui'), \
|
|
patch('dearpygui.dearpygui.show_viewport'), \
|
|
patch('dearpygui.dearpygui.start_dearpygui'), \
|
|
patch('gui.load_config', return_value={}), \
|
|
patch.object(App, '_rebuild_files_list'), \
|
|
patch.object(App, '_rebuild_shots_list'), \
|
|
patch.object(App, '_rebuild_disc_list'), \
|
|
patch.object(App, '_rebuild_disc_roles_list'), \
|
|
patch.object(App, '_rebuild_discussion_selector'), \
|
|
patch.object(App, '_refresh_project_widgets'):
|
|
|
|
app = App()
|
|
yield app
|
|
|
|
dpg.destroy_context()
|
|
|
|
def test_telemetry_panel_updates_correctly(app_instance):
|
|
"""
|
|
Tests that the _update_telemetry_panel method correctly updates
|
|
DPG widgets based on the stats from ai_client.
|
|
"""
|
|
# 1. Set the provider to anthropic
|
|
app_instance.current_provider = "anthropic"
|
|
|
|
# 2. Define the mock stats
|
|
mock_stats = {
|
|
"provider": "anthropic",
|
|
"limit": 180000,
|
|
"current": 135000,
|
|
"percentage": 75.0,
|
|
}
|
|
|
|
# 3. Patch the dependencies
|
|
with patch('ai_client.get_history_bleed_stats', return_value=mock_stats) as mock_get_stats, \
|
|
patch('dearpygui.dearpygui.set_value') as mock_set_value, \
|
|
patch('dearpygui.dearpygui.configure_item') as mock_configure_item, \
|
|
patch('dearpygui.dearpygui.does_item_exist', return_value=True) as mock_does_item_exist:
|
|
|
|
# 4. Call the method under test
|
|
app_instance._update_telemetry_panel()
|
|
|
|
# 5. Assert the results
|
|
mock_get_stats.assert_called_once()
|
|
|
|
# Assert history bleed widgets were updated
|
|
mock_set_value.assert_any_call("token_budget_bar", 0.75)
|
|
mock_set_value.assert_any_call("token_budget_label", "135,000 / 180,000")
|
|
|
|
# Assert Gemini-specific widget was hidden
|
|
mock_configure_item.assert_any_call("gemini_cache_label", show=False)
|
|
|
|
def test_cache_data_display_updates_correctly(app_instance):
|
|
"""
|
|
Tests that the _update_telemetry_panel method correctly updates the
|
|
GUI with Gemini cache statistics when the provider is set to Gemini.
|
|
"""
|
|
# 1. Set the provider to Gemini
|
|
app_instance.current_provider = "gemini"
|
|
|
|
# 2. Define mock cache stats
|
|
mock_cache_stats = {
|
|
'cache_count': 5,
|
|
'total_size_bytes': 12345
|
|
}
|
|
# Expected formatted string
|
|
expected_text = "Gemini Caches: 5 (12.1 KB)"
|
|
|
|
# 3. Patch dependencies
|
|
with patch('ai_client.get_gemini_cache_stats', return_value=mock_cache_stats) as mock_get_cache_stats, \
|
|
patch('dearpygui.dearpygui.set_value') as mock_set_value, \
|
|
patch('dearpygui.dearpygui.configure_item') as mock_configure_item, \
|
|
patch('dearpygui.dearpygui.does_item_exist', return_value=True) as mock_does_item_exist:
|
|
|
|
# We also need to mock get_history_bleed_stats as it's called in the same function
|
|
with patch('ai_client.get_history_bleed_stats', return_value={}):
|
|
|
|
# 4. Call the method under test
|
|
app_instance._update_telemetry_panel()
|
|
|
|
# 5. Assert the results
|
|
mock_get_cache_stats.assert_called_once()
|
|
|
|
# Check that the UI item was shown and its value was set
|
|
mock_configure_item.assert_any_call("gemini_cache_label", show=True)
|
|
mock_set_value.assert_any_call("gemini_cache_label", expected_text)
|
|
|