from typing import Optional

from fastapi import FastAPI
from fastapi.params import Query
from fastapi.testclient import TestClient

app = FastAPI()


@app.get("/")
def get(
    x: Optional[float] = Query(default=0, allow_inf_nan=False),
    y: Optional[float] = Query(default=0, allow_inf_nan=True),
    z: Optional[float] = Query(default=0),
) -> str:  # type: ignore
    return "OK"


client = TestClient(app)


def test_allow_inf_nan_false():
    response = client.get("/?x=inf")
    assert response.status_code == 422, response.text

    response = client.get("/?x=-inf")
    assert response.status_code == 422, response.text

    response = client.get("/?x=nan")
    assert response.status_code == 422, response.text

    response = client.get("/?x=0")
    assert response.status_code == 200, response.text


def test_allow_inf_nan_true():
    response = client.get("/?y=inf")
    assert response.status_code == 200, response.text

    response = client.get("/?y=-inf")
    assert response.status_code == 200, response.text

    response = client.get("/?y=nan")
    assert response.status_code == 200, response.text

    response = client.get("/?y=0")
    assert response.status_code == 200, response.text


def test_allow_inf_nan_not_specified():
    response = client.get("/?z=inf")
    assert response.status_code == 200, response.text

    response = client.get("/?z=-inf")
    assert response.status_code == 200, response.text

    response = client.get("/?z=nan")
    assert response.status_code == 200, response.text

    response = client.get("/?z=0")
    assert response.status_code == 200, response.text