lqs
d20566fe07
Some checks failed
Test / lint (push) Has been cancelled
Test / test (pydantic-v1, 3.10) (push) Has been cancelled
Test / test (pydantic-v1, 3.11) (push) Has been cancelled
Test / test (pydantic-v1, 3.12) (push) Has been cancelled
Test / test (pydantic-v1, 3.8) (push) Has been cancelled
Test / test (pydantic-v1, 3.9) (push) Has been cancelled
Test / test (pydantic-v2, 3.10) (push) Has been cancelled
Test / test (pydantic-v2, 3.11) (push) Has been cancelled
Test / test (pydantic-v2, 3.12) (push) Has been cancelled
Test / test (pydantic-v2, 3.8) (push) Has been cancelled
Test / test (pydantic-v2, 3.9) (push) Has been cancelled
Test / coverage-combine (push) Has been cancelled
Test / check (push) Has been cancelled
Issue Manager / issue-manager (push) Has been cancelled
Label Approved / label-approved (push) Has been cancelled
311 lines
12 KiB
Python
311 lines
12 KiB
Python
from typing import List, Tuple
|
|
|
|
from dirty_equals import IsDict
|
|
from fastapi import FastAPI, Form
|
|
from fastapi.testclient import TestClient
|
|
from pydantic import BaseModel
|
|
|
|
app = FastAPI()
|
|
|
|
|
|
class ItemGroup(BaseModel):
|
|
items: List[Tuple[str, str]]
|
|
|
|
|
|
class Coordinate(BaseModel):
|
|
x: float
|
|
y: float
|
|
|
|
|
|
@app.post("/model-with-tuple/")
|
|
def post_model_with_tuple(item_group: ItemGroup):
|
|
return item_group
|
|
|
|
|
|
@app.post("/tuple-of-models/")
|
|
def post_tuple_of_models(square: Tuple[Coordinate, Coordinate]):
|
|
return square
|
|
|
|
|
|
@app.post("/tuple-form/")
|
|
def hello(values: Tuple[int, int] = Form()):
|
|
return values
|
|
|
|
|
|
client = TestClient(app)
|
|
|
|
|
|
def test_model_with_tuple_valid():
|
|
data = {"items": [["foo", "bar"], ["baz", "whatelse"]]}
|
|
response = client.post("/model-with-tuple/", json=data)
|
|
assert response.status_code == 200, response.text
|
|
assert response.json() == data
|
|
|
|
|
|
def test_model_with_tuple_invalid():
|
|
data = {"items": [["foo", "bar"], ["baz", "whatelse", "too", "much"]]}
|
|
response = client.post("/model-with-tuple/", json=data)
|
|
assert response.status_code == 422, response.text
|
|
|
|
data = {"items": [["foo", "bar"], ["baz"]]}
|
|
response = client.post("/model-with-tuple/", json=data)
|
|
assert response.status_code == 422, response.text
|
|
|
|
|
|
def test_tuple_with_model_valid():
|
|
data = [{"x": 1, "y": 2}, {"x": 3, "y": 4}]
|
|
response = client.post("/tuple-of-models/", json=data)
|
|
assert response.status_code == 200, response.text
|
|
assert response.json() == data
|
|
|
|
|
|
def test_tuple_with_model_invalid():
|
|
data = [{"x": 1, "y": 2}, {"x": 3, "y": 4}, {"x": 5, "y": 6}]
|
|
response = client.post("/tuple-of-models/", json=data)
|
|
assert response.status_code == 422, response.text
|
|
|
|
data = [{"x": 1, "y": 2}]
|
|
response = client.post("/tuple-of-models/", json=data)
|
|
assert response.status_code == 422, response.text
|
|
|
|
|
|
def test_tuple_form_valid():
|
|
response = client.post("/tuple-form/", data={"values": ("1", "2")})
|
|
assert response.status_code == 200, response.text
|
|
assert response.json() == [1, 2]
|
|
|
|
|
|
def test_tuple_form_invalid():
|
|
response = client.post("/tuple-form/", data={"values": ("1", "2", "3")})
|
|
assert response.status_code == 422, response.text
|
|
|
|
response = client.post("/tuple-form/", data={"values": ("1")})
|
|
assert response.status_code == 422, response.text
|
|
|
|
|
|
def test_openapi_schema():
|
|
response = client.get("/openapi.json")
|
|
assert response.status_code == 200, response.text
|
|
assert response.json() == {
|
|
"openapi": "3.1.0",
|
|
"info": {"title": "FastAPI", "version": "0.1.0"},
|
|
"paths": {
|
|
"/model-with-tuple/": {
|
|
"post": {
|
|
"summary": "Post Model With Tuple",
|
|
"operationId": "post_model_with_tuple_model_with_tuple__post",
|
|
"requestBody": {
|
|
"content": {
|
|
"application/json": {
|
|
"schema": {"$ref": "#/components/schemas/ItemGroup"}
|
|
}
|
|
},
|
|
"required": True,
|
|
},
|
|
"responses": {
|
|
"200": {
|
|
"description": "Successful Response",
|
|
"content": {"application/json": {"schema": {}}},
|
|
},
|
|
"422": {
|
|
"description": "Validation Error",
|
|
"content": {
|
|
"application/json": {
|
|
"schema": {
|
|
"$ref": "#/components/schemas/HTTPValidationError"
|
|
}
|
|
}
|
|
},
|
|
},
|
|
},
|
|
}
|
|
},
|
|
"/tuple-of-models/": {
|
|
"post": {
|
|
"summary": "Post Tuple Of Models",
|
|
"operationId": "post_tuple_of_models_tuple_of_models__post",
|
|
"requestBody": {
|
|
"content": {
|
|
"application/json": {
|
|
"schema": IsDict(
|
|
{
|
|
"title": "Square",
|
|
"maxItems": 2,
|
|
"minItems": 2,
|
|
"type": "array",
|
|
"prefixItems": [
|
|
{"$ref": "#/components/schemas/Coordinate"},
|
|
{"$ref": "#/components/schemas/Coordinate"},
|
|
],
|
|
}
|
|
)
|
|
| IsDict(
|
|
# TODO: remove when deprecating Pydantic v1
|
|
{
|
|
"title": "Square",
|
|
"maxItems": 2,
|
|
"minItems": 2,
|
|
"type": "array",
|
|
"items": [
|
|
{"$ref": "#/components/schemas/Coordinate"},
|
|
{"$ref": "#/components/schemas/Coordinate"},
|
|
],
|
|
}
|
|
)
|
|
}
|
|
},
|
|
"required": True,
|
|
},
|
|
"responses": {
|
|
"200": {
|
|
"description": "Successful Response",
|
|
"content": {"application/json": {"schema": {}}},
|
|
},
|
|
"422": {
|
|
"description": "Validation Error",
|
|
"content": {
|
|
"application/json": {
|
|
"schema": {
|
|
"$ref": "#/components/schemas/HTTPValidationError"
|
|
}
|
|
}
|
|
},
|
|
},
|
|
},
|
|
}
|
|
},
|
|
"/tuple-form/": {
|
|
"post": {
|
|
"summary": "Hello",
|
|
"operationId": "hello_tuple_form__post",
|
|
"requestBody": {
|
|
"content": {
|
|
"application/x-www-form-urlencoded": {
|
|
"schema": {
|
|
"$ref": "#/components/schemas/Body_hello_tuple_form__post"
|
|
}
|
|
}
|
|
},
|
|
"required": True,
|
|
},
|
|
"responses": {
|
|
"200": {
|
|
"description": "Successful Response",
|
|
"content": {"application/json": {"schema": {}}},
|
|
},
|
|
"422": {
|
|
"description": "Validation Error",
|
|
"content": {
|
|
"application/json": {
|
|
"schema": {
|
|
"$ref": "#/components/schemas/HTTPValidationError"
|
|
}
|
|
}
|
|
},
|
|
},
|
|
},
|
|
}
|
|
},
|
|
},
|
|
"components": {
|
|
"schemas": {
|
|
"Body_hello_tuple_form__post": {
|
|
"title": "Body_hello_tuple_form__post",
|
|
"required": ["values"],
|
|
"type": "object",
|
|
"properties": {
|
|
"values": IsDict(
|
|
{
|
|
"title": "Values",
|
|
"maxItems": 2,
|
|
"minItems": 2,
|
|
"type": "array",
|
|
"prefixItems": [
|
|
{"type": "integer"},
|
|
{"type": "integer"},
|
|
],
|
|
}
|
|
)
|
|
| IsDict(
|
|
# TODO: remove when deprecating Pydantic v1
|
|
{
|
|
"title": "Values",
|
|
"maxItems": 2,
|
|
"minItems": 2,
|
|
"type": "array",
|
|
"items": [{"type": "integer"}, {"type": "integer"}],
|
|
}
|
|
)
|
|
},
|
|
},
|
|
"Coordinate": {
|
|
"title": "Coordinate",
|
|
"required": ["x", "y"],
|
|
"type": "object",
|
|
"properties": {
|
|
"x": {"title": "X", "type": "number"},
|
|
"y": {"title": "Y", "type": "number"},
|
|
},
|
|
},
|
|
"HTTPValidationError": {
|
|
"title": "HTTPValidationError",
|
|
"type": "object",
|
|
"properties": {
|
|
"detail": {
|
|
"title": "Detail",
|
|
"type": "array",
|
|
"items": {"$ref": "#/components/schemas/ValidationError"},
|
|
}
|
|
},
|
|
},
|
|
"ItemGroup": {
|
|
"title": "ItemGroup",
|
|
"required": ["items"],
|
|
"type": "object",
|
|
"properties": {
|
|
"items": {
|
|
"title": "Items",
|
|
"type": "array",
|
|
"items": IsDict(
|
|
{
|
|
"maxItems": 2,
|
|
"minItems": 2,
|
|
"type": "array",
|
|
"prefixItems": [
|
|
{"type": "string"},
|
|
{"type": "string"},
|
|
],
|
|
}
|
|
)
|
|
| IsDict(
|
|
# TODO: remove when deprecating Pydantic v1
|
|
{
|
|
"maxItems": 2,
|
|
"minItems": 2,
|
|
"type": "array",
|
|
"items": [{"type": "string"}, {"type": "string"}],
|
|
}
|
|
),
|
|
}
|
|
},
|
|
},
|
|
"ValidationError": {
|
|
"title": "ValidationError",
|
|
"required": ["loc", "msg", "type"],
|
|
"type": "object",
|
|
"properties": {
|
|
"loc": {
|
|
"title": "Location",
|
|
"type": "array",
|
|
"items": {
|
|
"anyOf": [{"type": "string"}, {"type": "integer"}]
|
|
},
|
|
},
|
|
"msg": {"title": "Message", "type": "string"},
|
|
"type": {"title": "Error Type", "type": "string"},
|
|
},
|
|
},
|
|
}
|
|
},
|
|
}
|