import asyncio
import time
from typing import Annotated

from pydantic import BaseModel

from fastapi import FastAPI, Depends, Query, Response
from httpx import AsyncClient, ASGITransport


async def test_no_dependency_override_performance(loop_count: int = 100_000):
    app = FastAPI()

    async def sub_sub_dependency(more_nested: Annotated[int, Query()]) -> int:
        return more_nested

    class SubDependency(BaseModel):
        value: Annotated[int, Query()]
        other: Annotated[int, Depends(sub_sub_dependency)]

    async def some_dependency(nested: Annotated[SubDependency, Depends()]) -> int:
        return nested.value + nested.other

    @app.get("/data")
    async def get(
        one: int = Depends(some_dependency),
    ):
        return Response(content=f'{{"value": {one}}}'.encode(), status_code=200)

    async with AsyncClient(
        transport=ASGITransport(app=app),
        base_url="http://localhost",
    ) as client:
        start_time = time.perf_counter()

        for _ in range(loop_count):
            response = await client.get("/data", params={"value": 1, "more_nested": 2})
            assert response.status_code == 200, response.text
            assert response.text == '{"value": 3}'

        duration = time.perf_counter() - start_time

    print(f"Performed GET /data requests without dependency override: {loop_count:,} times in {duration:.4f}s")
    print(f"Average per call: {duration / loop_count * 1e6:.2f} µs")


async def test_dependency_override_nonempty_performance(loop_count: int = 100_000):
    app = FastAPI()

    async def sub_sub_dependency(more_nested: Annotated[int, Query()]) -> int:
        return more_nested

    class SubDependency(BaseModel):
        value: Annotated[int, Query()]
        other: Annotated[int, Depends(sub_sub_dependency)]

    async def some_dependency(nested: Annotated[SubDependency, Depends()]) -> int:
        return nested.value + nested.other

    async def unused_dependency(nested: Annotated[SubDependency, Depends()]) -> int:
        return nested.value + nested.other + 1

    @app.get("/data")
    async def get(
        one: int = Depends(some_dependency),
    ):
        return Response(content=f'{{"value": {one}}}'.encode(), status_code=200)

    async def override_unused_dependency() -> int:
        return 22

    app.dependency_overrides[unused_dependency] = override_unused_dependency

    async with AsyncClient(
        transport=ASGITransport(app=app),
        base_url="http://localhost",
    ) as client:
        start_time = time.perf_counter()

        for _ in range(loop_count):
            response = await client.get("/data", params={"value": 1, "more_nested": 2})
            assert response.status_code == 200, response.text
            assert response.text == '{"value": 3}'

        duration = time.perf_counter() - start_time

    print(f"Performed GET /data requests with non-empty dependency overrides: {loop_count:,} times in {duration:.4f}s")
    print(f"Average per call: {duration / loop_count * 1e6:.2f} µs")


async def test_dependency_override_performance(loop_count: int = 100_000):
    app = FastAPI()

    async def sub_sub_dependency(more_nested: Annotated[int, Query()]) -> int:
        return more_nested

    class SubDependency(BaseModel):
        value: Annotated[int, Query()]
        other: Annotated[int, Depends(sub_sub_dependency)]

    async def some_dependency(nested: Annotated[SubDependency, Depends()]) -> int:
        return nested.value + nested.other

    @app.get("/data")
    async def get(
        one: int = Depends(some_dependency),
    ):
        return Response(content=f'{{"value": {one}}}'.encode(), status_code=200)

    async def sub_sub_dependency_override(more_nested: Annotated[int, Query()]) -> int:
        return 11

    app.dependency_overrides[sub_sub_dependency] = sub_sub_dependency_override

    async with AsyncClient(
        transport=ASGITransport(app=app),
        base_url="http://localhost",
    ) as client:
        start_time = time.perf_counter()

        for _ in range(loop_count):
            response = await client.get("/data", params={"value": 1, "more_nested": 2})
            assert response.status_code == 200, response.text
            assert response.text == '{"value": 12}', response.text

        duration = time.perf_counter() - start_time

    print(f"Performed GET /data requests with actual dependency override: {loop_count:,} times in {duration:.4f}s")
    print(f"Average per call: {duration / loop_count * 1e6:.2f} µs")

if __name__ == "__main__":
    asyncio.run(test_no_dependency_override_performance())
    print("==" * 20)
    asyncio.run(test_dependency_override_nonempty_performance())
    print("==" * 20)
    asyncio.run(test_dependency_override_performance())
    print("==" * 20)