from typing_extensions import Annotated

import pytest
from fastapi import Body, FastAPI
from fastapi.exceptions import ResponseValidationError
from fastapi.testclient import TestClient
from pydantic import v1

from tests.utils import needs_pydanticv2


class Item(v1.BaseModel):
    name: str
    description: str | None = None
    price: float
    tax: float | None = None
    tags: list = []


class Model(v1.BaseModel):
    name: str


app = FastAPI()


@app.post("/request_body")
async def request_body(body: Annotated[Item, Body()]):
    return body


@app.get("/response_model", response_model=Model)
async def response_model():
    return Model(name="valid_model")


@app.get("/response_model__invalid", response_model=Model)
async def response_model__invalid():
    return 1


@app.get("/response_model_list", response_model=list[Model])
async def response_model_list():
    return [Model(name="valid_model")]


@app.get("/response_model_list__invalid", response_model=list[Model])
async def response_model_list__invalid():
    return [1]


client = TestClient(app)


@needs_pydanticv2
class TestResponseModel:
    def test_simple__valid(self):
        response = client.get("/response_model")
        assert response.status_code == 200
        assert response.json() == {"name": "valid_model"}

    def test_simple__invalid(self):
        with pytest.raises(ResponseValidationError):
            client.get("/response_model__invalid")

    def test_list__valid(self):
        response = client.get("/response_model_list")
        assert response.status_code == 200
        assert response.json() == [{"name": "valid_model"}]

    def test_list__invalid(self):
        with pytest.raises(ResponseValidationError):
            client.get("/response_model_list__invalid")


@needs_pydanticv2
class TestRequestBody:
    def test_model__valid(self):
        response = client.post("/request_body", json={"name": "myname", "price": 1.0})
        assert response.status_code == 200, response.text

    def test_model__invalid(self):
        response = client.post("/request_body", json={"name": "myname"})
        assert response.status_code == 422, response.text