140 lines
5.9 KiB
Python
140 lines
5.9 KiB
Python
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
import importlib.util
|
|
import sys
|
|
import unittest
|
|
|
|
|
|
def load_module():
|
|
module_path = Path(__file__).resolve().parents[1] / "sub2api_quota_tui.py"
|
|
spec = importlib.util.spec_from_file_location("sub2api_quota_tui", module_path)
|
|
assert spec and spec.loader
|
|
module = importlib.util.module_from_spec(spec)
|
|
sys.modules[spec.name] = module
|
|
spec.loader.exec_module(module)
|
|
return module
|
|
|
|
|
|
class Sub2APIQuotaTUITests(unittest.TestCase):
|
|
def test_normalize_rows_sorts_by_usage_and_formats_windows(self) -> None:
|
|
mod = load_module()
|
|
payload = {
|
|
"generated_at": "2026-06-09T12:00:00+08:00",
|
|
"totals": {"total_accounts": 2, "usable_accounts": 2, "today_cost_usd": 0.75, "today_tokens": 3000, "today_requests": 9},
|
|
"accounts": [
|
|
{
|
|
"id": 1,
|
|
"name": "quota-low",
|
|
"routing_group": "slow",
|
|
"provider": "anthropic",
|
|
"kind": "quota_limited",
|
|
"status": "ok",
|
|
"account_type": "oauth",
|
|
"today_cost_usd": 0.25,
|
|
"today_tokens": 2000,
|
|
"today_requests": 4,
|
|
"windows": [
|
|
{"id": "five-hour", "used_percent": 80, "remaining_percent": 20, "reset": "2026-06-09T15:00:00+08:00"},
|
|
{"id": "weekly", "used_percent": 10, "remaining_percent": 90, "reset": "2026-06-10T15:00:00+08:00"},
|
|
],
|
|
},
|
|
{
|
|
"id": 2,
|
|
"name": "payg-high",
|
|
"routing_group": "fast",
|
|
"provider": "openai",
|
|
"kind": "pay_as_you_go",
|
|
"status": "ok",
|
|
"account_type": "apikey",
|
|
"daily_quota_limit": 100,
|
|
"daily_quota_used": 91.2,
|
|
"daily_quota_remaining": 8.8,
|
|
"daily_quota_used_percent": 91.2,
|
|
"today_cost_usd": 0.5,
|
|
"today_tokens": 1000,
|
|
"today_requests": 5,
|
|
"windows": [],
|
|
},
|
|
],
|
|
}
|
|
|
|
rows = mod.normalize_account_rows(payload)
|
|
|
|
self.assertEqual([row["name"] for row in rows], ["payg-high", "quota-low"])
|
|
self.assertEqual(rows[0]["routing_group"], "fast")
|
|
self.assertEqual(rows[0]["provider"], "openai")
|
|
self.assertEqual(rows[0]["daily_quota_cell"], "91.2/100")
|
|
self.assertEqual(rows[0]["kind_label"], "usage")
|
|
self.assertEqual(rows[1]["five_hour"], "80%/20%")
|
|
self.assertEqual(rows[1]["weekly"], "10%/90%")
|
|
self.assertIn("today $0.75", mod.summary_line(payload))
|
|
|
|
def test_filter_matches_kind_and_name(self) -> None:
|
|
mod = load_module()
|
|
payload = {
|
|
"accounts": [
|
|
{"id": 1, "name": "alpha", "routing_group": "slow", "provider": "anthropic", "kind": "quota_limited", "today_cost_usd": 0},
|
|
{"id": 2, "name": "beta", "routing_group": "sfast", "provider": "openai", "kind": "pay_as_you_go", "today_cost_usd": 0},
|
|
]
|
|
}
|
|
|
|
self.assertEqual([row["name"] for row in mod.normalize_account_rows(payload, "pay")], ["beta"])
|
|
self.assertEqual([row["name"] for row in mod.normalize_account_rows(payload, "alp")], ["alpha"])
|
|
self.assertEqual([row["name"] for row in mod.normalize_account_rows(payload, "sfast")], ["beta"])
|
|
self.assertEqual([row["name"] for row in mod.normalize_account_rows(payload, "anthropic")], ["alpha"])
|
|
|
|
def test_once_output_is_name_first_and_includes_daily_quota(self) -> None:
|
|
mod = load_module()
|
|
payload = {
|
|
"generated_at": "2026-06-09T12:00:00+08:00",
|
|
"totals": {"total_accounts": 1, "usable_accounts": 1, "today_cost_usd": 1, "today_tokens": 100, "today_requests": 2},
|
|
"accounts": [
|
|
{
|
|
"id": 2,
|
|
"name": "input 300",
|
|
"routing_group": "fast",
|
|
"provider": "openai",
|
|
"kind": "daily_limited",
|
|
"status": "ok",
|
|
"account_type": "apikey",
|
|
"daily_quota_limit": 300,
|
|
"daily_quota_used": 96.6,
|
|
"daily_quota_remaining": 203.4,
|
|
"daily_quota_used_percent": 32.2,
|
|
"daily_quota_reset_at": "2026-06-09T16:00:00Z",
|
|
"today_cost_usd": 1,
|
|
"today_tokens": 100,
|
|
"today_requests": 2,
|
|
"windows": [],
|
|
}
|
|
],
|
|
}
|
|
|
|
rows = mod.normalize_account_rows(payload)
|
|
|
|
self.assertEqual(rows[0]["kind_label"], "daily")
|
|
self.assertEqual(rows[0]["routing_group"], "fast")
|
|
self.assertEqual(rows[0]["provider"], "openai")
|
|
self.assertEqual(rows[0]["daily_quota_cell"], "96.6/300")
|
|
self.assertTrue(rows[0]["reset"].endswith("00:00") or rows[0]["reset"] != "-")
|
|
|
|
def test_routing_group_sort_shows_sfast_first(self) -> None:
|
|
mod = load_module()
|
|
payload = {
|
|
"accounts": [
|
|
{"id": 1, "name": "manual", "routing_group": "id", "kind": "pay_as_you_go", "today_cost_usd": 10},
|
|
{"id": 2, "name": "slow", "routing_group": "slow", "kind": "pay_as_you_go", "today_cost_usd": 1},
|
|
{"id": 3, "name": "fast", "routing_group": "fast", "kind": "pay_as_you_go", "today_cost_usd": 1},
|
|
{"id": 4, "name": "sfast", "routing_group": "sfast", "kind": "pay_as_you_go", "today_cost_usd": 1},
|
|
]
|
|
}
|
|
|
|
rows = mod.normalize_account_rows(payload)
|
|
|
|
self.assertEqual([row["name"] for row in rows], ["sfast", "fast", "slow", "manual"])
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|