57 lines
2.0 KiB
Python
57 lines
2.0 KiB
Python
from src.vendor_state import get_vendor_state, 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
|