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()