from typing import Any
from unittest.mock import patch

import httpx
import pytest

from great_ai import RemoteCallError, call_remote_great_ai_async


def response_trace() -> Any:
    return dict(
        trace_id="remote",
        created="2026-01-01",
        original_execution_time_ms=0,
        models=[],
        tags=[],
        logged_values={},
        output=7,
    )


@pytest.mark.asyncio
async def test_remote_call_retries_response_read_failures() -> None:
    calls = 0

    def handler(request: httpx.Request) -> httpx.Response:
        nonlocal calls
        calls += 1
        if calls == 1:
            raise httpx.ReadError("response disconnected")
        return httpx.Response(200, json=response_trace())

    with patch(
        "great_ai.remote.call_remote_great_ai_async.httpx.AsyncHTTPTransport",
        return_value=httpx.MockTransport(handler),
    ):
        assert (
            await call_remote_great_ai_async("http://test", {}, retry_count=2)
        ).output == 7
    assert calls == 2


@pytest.mark.asyncio
async def test_remote_retry_limit_is_respected() -> None:
    calls = 0

    def handler(request: httpx.Request) -> httpx.Response:
        nonlocal calls
        calls += 1
        raise httpx.ReadError("response disconnected")

    with patch(
        "great_ai.remote.call_remote_great_ai_async.httpx.AsyncHTTPTransport",
        return_value=httpx.MockTransport(handler),
    ):
        with pytest.raises(RemoteCallError, match="response disconnected"):
            await call_remote_great_ai_async("http://test", {}, retry_count=1)
    assert calls == 2