a4ab7d4d27
Co-authored-by: malulian <malulian@users.noreply.github.com> Co-authored-by: ai-ag2026 <ai-ag2026@users.noreply.github.com>
50 lines
2.0 KiB
Python
50 lines
2.0 KiB
Python
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
|
|
|
|
REPO = Path(__file__).resolve().parents[1]
|
|
INDEX = (REPO / "static" / "index.html").read_text(encoding="utf-8")
|
|
UI_JS = (REPO / "static" / "ui.js").read_text(encoding="utf-8")
|
|
BOOT_JS = (REPO / "static" / "boot.js").read_text(encoding="utf-8")
|
|
CSS = (REPO / "static" / "style.css").read_text(encoding="utf-8")
|
|
|
|
|
|
def test_quota_indicator_is_near_model_picker_in_composer_chrome():
|
|
model_idx = INDEX.find('id="composerModelChip"')
|
|
quota_idx = INDEX.find('id="providerQuotaChip"')
|
|
|
|
assert model_idx != -1, "composer model chip must exist"
|
|
assert quota_idx != -1, "provider quota chip must exist"
|
|
assert model_idx < quota_idx < INDEX.find('id="composerReasoningWrap"'), (
|
|
"quota chip should sit next to the model picker, before reasoning/toolset chrome"
|
|
)
|
|
assert 'class="provider-quota-chip"' in INDEX
|
|
assert 'hidden' in INDEX[quota_idx - 200 : quota_idx + 400]
|
|
|
|
|
|
def test_quota_indicator_fetches_provider_quota_on_boot():
|
|
assert "function refreshProviderQuotaIndicator" in UI_JS
|
|
assert "api('/api/provider/quota')" in UI_JS
|
|
assert "refreshProviderQuotaIndicator" in BOOT_JS
|
|
|
|
|
|
def test_quota_indicator_hides_unsupported_or_failed_statuses():
|
|
render_idx = UI_JS.find("function renderProviderQuotaIndicator")
|
|
assert render_idx != -1, "renderProviderQuotaIndicator helper must exist"
|
|
render_block = UI_JS[render_idx : UI_JS.find("function ", render_idx + 1)]
|
|
|
|
assert "providerQuotaChip" in render_block
|
|
assert "chip.hidden=true" in render_block
|
|
assert "status.status!=='available'" in render_block
|
|
assert "!status.quota" in render_block
|
|
assert "unsupported" not in render_block.lower(), "ambient chip should disappear instead of showing noisy unsupported text"
|
|
|
|
|
|
def test_quota_indicator_formats_openrouter_and_account_limit_shapes():
|
|
assert "function _providerQuotaIndicatorText" in UI_JS
|
|
assert "limit_remaining" in UI_JS
|
|
assert "account_limits" in UI_JS
|
|
assert "remaining_percent" in UI_JS
|
|
assert "provider-quota-chip" in CSS
|