from typing import Annotated

from fastapi import FastAPI, Query
from fastapi.testclient import TestClient
from pydantic import BaseModel, ConfigDict, Field

app = FastAPI()


class Raw(BaseModel):
    by_name: str = Field(default=..., alias="byName")
    model_config = ConfigDict(extra="forbid")


class Populate(BaseModel):
    by_name: str = Field(default=..., alias="byName")
    model_config = ConfigDict(populate_by_name=True, extra="forbid")


class NoAlias(BaseModel):
    by_name: str
    model_config = ConfigDict(extra="forbid")


@app.get("/raw")
def raw(e: Annotated[Raw, Query()]):
    return e


@app.get("/populate")
def populate(c: Annotated[Populate, Query()]):
    return c


@app.get("/no-alias")
def no_alias(n: Annotated[NoAlias, Query()]):
    return n


client = TestClient(app)


def test_raw():
    """
    Only the alias name should work
    """

    response = client.get("/raw", params={"by_name": "expect 422"})
    assert response.status_code == 422

    response = client.get("/raw", params={"byName": "ok"})
    assert response.status_code == 200

    parameters = app.openapi()["paths"]["/raw"]["get"]["parameters"]
    assert len(parameters) == 1
    assert parameters[0]["name"] == "byName"


def test_populate_by_name():
    """
    Both the alias name and the original name should work, but only alias is in OpenAPI schema
    """

    response = client.get("/populate", params={"byName": "ok"})
    assert response.status_code == 200
    assert response.json() == {"byName": "ok"}

    response = client.get("/populate", params={"by_name": "ok"})
    assert response.status_code == 200
    assert response.json() == {"byName": "ok"}

    parameters = app.openapi()["paths"]["/populate"]["get"]["parameters"]
    assert len(parameters) == 1
    assert parameters[0]["name"] == "byName"


def test_no_alias():
    """
    Only the original name should work
    """

    response = client.get("/no-alias", params={"byName": "expect 422"})
    assert response.status_code == 422

    response = client.get("/no-alias", params={"by_name": "ok"})
    assert response.status_code == 200

    parameters = app.openapi()["paths"]["/no-alias"]["get"]["parameters"]
    assert len(parameters) == 1
    assert parameters[0]["name"] == "by_name"