import json
from typing import Annotated, Any

from fastapi import FastAPI, Form, Query
from fastapi.testclient import TestClient
from pydantic import Json

app = FastAPI()


@app.post("/form-json-equals-syntax")
def form_json_equals_syntax(data: Json[dict[str, Any]] = Form()) -> dict[str, Any]:
    return data


@app.post("/form-json-annotated-syntax")
def form_json_annotated_syntax(
    data: Annotated[Json[dict[str, Any]], Form()],
) -> dict[str, Any]:
    return data


@app.post("/form-json-list-equals")
def form_json_list_equals(items: Json[list[str]] = Form()) -> list[str]:
    return items


@app.post("/form-json-list-annotated")
def form_json_list_annotated(
    items: Annotated[Json[list[str]], Form()],
) -> list[str]:
    return items


@app.post("/form-json-nested-equals")
def form_json_nested_equals(
    config: Json[dict[str, list[int]]] = Form(),
) -> dict[str, list[int]]:
    return config


@app.post("/form-json-nested-annotated")
def form_json_nested_annotated(
    config: Annotated[Json[dict[str, list[int]]], Form()],
) -> dict[str, list[int]]:
    return config


client = TestClient(app)


def test_form_json_equals_syntax():
    test_data = {"key": "value", "nested": {"inner": 123}}
    response = client.post(
        "/form-json-equals-syntax", data={"data": json.dumps(test_data)}
    )
    assert response.status_code == 200, response.text
    assert response.json() == test_data


def test_form_json_annotated_syntax():
    test_data = {"key": "value", "nested": {"inner": 123}}
    response = client.post(
        "/form-json-annotated-syntax", data={"data": json.dumps(test_data)}
    )
    assert response.status_code == 200, response.text
    assert response.json() == test_data


def test_form_json_list_equals():
    test_data = ["abc", "def", "ghi"]
    response = client.post(
        "/form-json-list-equals", data={"items": json.dumps(test_data)}
    )
    assert response.status_code == 200, response.text
    assert response.json() == test_data


def test_form_json_list_annotated():
    test_data = ["abc", "def", "ghi"]
    response = client.post(
        "/form-json-list-annotated", data={"items": json.dumps(test_data)}
    )
    assert response.status_code == 200, response.text
    assert response.json() == test_data


def test_form_json_nested_equals():
    test_data = {"groups": [1, 2, 3], "ids": [4, 5]}
    response = client.post(
        "/form-json-nested-equals", data={"config": json.dumps(test_data)}
    )
    assert response.status_code == 200, response.text
    assert response.json() == test_data


def test_form_json_nested_annotated():
    test_data = {"groups": [1, 2, 3], "ids": [4, 5]}
    response = client.post(
        "/form-json-nested-annotated", data={"config": json.dumps(test_data)}
    )
    assert response.status_code == 200, response.text
    assert response.json() == test_data


def test_form_json_invalid_json_equals():
    response = client.post(
        "/form-json-equals-syntax", data={"data": "not valid json{"}
    )
    assert response.status_code == 422


def test_form_json_invalid_json_annotated():
    response = client.post(
        "/form-json-annotated-syntax", data={"data": "not valid json{"}
    )
    assert response.status_code == 422


def test_form_json_empty_dict_equals():
    test_data = {}
    response = client.post(
        "/form-json-equals-syntax", data={"data": json.dumps(test_data)}
    )
    assert response.status_code == 200, response.text
    assert response.json() == test_data


def test_form_json_empty_list_equals():
    test_data = []
    response = client.post(
        "/form-json-list-equals", data={"items": json.dumps(test_data)}
    )
    assert response.status_code == 200, response.text
    assert response.json() == test_data