from __future__ import annotations
from typing import Any

from fastapi.testclient import TestClient

from gibby.app import create_app
from gibby.manager import AccountManager, NoUsableAccountError


class StubManager(AccountManager):
    def __init__(
        self,
        response: dict[str, Any] | None = None,
        error: Exception | None = None,
    ):
        self.response = response
        self.error = error

    async def issue_token_response(self):
        if self.error is not None:
            raise self.error
        if self.response is not None:
            return self.response
        return {}

def test_health_ok() -> None:
    client = TestClient(create_app(StubManager(response={})))
    response = client.get("/health")
    assert response.status_code == 200
    assert response.text == "ok"


def test_token_success_shape() -> None:
    payload = {
        "token": "abc",
        "limit": {
            "used_percent": 10,
            "remaining_percent": 90,
            "exhausted": False,
            "needs_prepare": False,
        },
        "usage": {"primary_window": None, "secondary_window": None},
    }
    client = TestClient(create_app(StubManager(response=payload)))
    response = client.get("/token")
    assert response.status_code == 200
    assert response.json() == payload


def test_token_error_shape() -> None:
    client = TestClient(
        create_app(
            StubManager(error=NoUsableAccountError("No usable account available"))
        )
    )
    response = client.get("/token")
    assert response.status_code == 503
    assert response.json() == {"error": "No usable account available"}