ed
d9cd7c557b
Per spec FR2 + Phase 2.2 + architecture feedback (data != view):
- VendorMetric (data) -> src/ai_client.py (alongside VendorCapabilities; all vendor data)
- get_vendor_state -> renamed to _get_vendor_state_metrics in src/gui_2.py
(it's a view-helper that builds the metrics for render_vendor_state's table)
- render_vendor_state in gui_2.py now calls _get_vendor_state_metrics directly
Tests:
- tests/test_vendor_state.py: imports get_vendor_state from src.gui_2, VendorMetric from src.ai_client
58 lines
2.0 KiB
Python
58 lines
2.0 KiB
Python
from src.gui_2 import _get_vendor_state_metrics as get_vendor_state
|
|
from src.ai_client import VendorMetric
|
|
|
|
class _StubTT:
|
|
def __init__(self, used=0, limit=0, cache_hits=0, cache_misses=0):
|
|
self.used = used
|
|
self.limit = limit
|
|
self.cache_hits = cache_hits
|
|
self.cache_misses = cache_misses
|
|
|
|
class _StubController:
|
|
def __init__(self, token_tracker=None, last_error=None, vendor_quota=None):
|
|
self.token_tracker = token_tracker
|
|
self.last_error = last_error
|
|
self.vendor_quota = vendor_quota if vendor_quota is not None else {}
|
|
|
|
class _StubApp:
|
|
def __init__(self, controller):
|
|
self.current_provider = "Anthropic"
|
|
self.current_model = "claude-opus-4"
|
|
self.controller = controller
|
|
|
|
def test_get_vendor_state_returns_core_metrics():
|
|
ctrl = _StubController(token_tracker=_StubTT(used=78234, limit=200000, cache_hits=1200, cache_misses=80), vendor_quota={"remaining_pct": 87})
|
|
app = _StubApp(ctrl)
|
|
metrics = get_vendor_state(app)
|
|
keys = {m.key for m in metrics}
|
|
assert "provider_model" in keys
|
|
assert "context_window" in keys
|
|
assert "cache" in keys
|
|
assert "quota" in keys
|
|
assert "last_error" in keys
|
|
|
|
def test_missing_data_renders_em_dash_not_crash():
|
|
ctrl = _StubController(token_tracker=None, last_error=None, vendor_quota={})
|
|
app = _StubApp(ctrl)
|
|
metrics = get_vendor_state(app)
|
|
assert len(metrics) > 0
|
|
for m in metrics:
|
|
assert m.value is not None
|
|
assert m.value != ""
|
|
|
|
def test_context_window_state_warn_above_75_percent():
|
|
ctrl = _StubController(token_tracker=_StubTT(used=160000, limit=200000, cache_hits=0, cache_misses=0))
|
|
app = _StubApp(ctrl)
|
|
metrics = get_vendor_state(app)
|
|
cw = next(m for m in metrics if m.key == "context_window")
|
|
assert cw.state == "warn"
|
|
assert "80%" in cw.value
|
|
|
|
def test_last_error_state_error_when_present():
|
|
ctrl = _StubController(token_tracker=None, last_error={"class": "ProviderError", "message": "rate limit"})
|
|
app = _StubApp(ctrl)
|
|
metrics = get_vendor_state(app)
|
|
err = next(m for m in metrics if m.key == "last_error")
|
|
assert err.state == "error"
|
|
assert "ProviderError" in err.value
|