happyz
/
fastapi
mirror of https://github.com/tiangolo/fastapi.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

🐛 Fix JSON Schema for dataclasses, supporting the fixes in Pydantic 1.9 (#4272) 

Co-authored-by: Sebastián Ramírez <tiangolo@gmail.com>
This commit is contained in:
Eric Jolibois 2021-12-12 12:28:35 +01:00 committed by GitHub
parent 0a87bc88b8
commit b0cd4d7e7e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 15 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
tests/test_tutorial/test_dataclasses/test_tutorial002.py
View File

@ -1,3 +1,5 @@
from copy import deepcopy
from fastapi.testclient import TestClient from fastapi.testclient import TestClient
from docs_src.dataclasses.tutorial002 import app from docs_src.dataclasses.tutorial002 import app
@ -29,7 +31,7 @@ openapi_schema = {
"schemas": { "schemas": {
"Item": { "Item": {
"title": "Item", "title": "Item",
"required": ["name", "price", "tags"], "required": ["name", "price"],
"type": "object", "type": "object",
"properties": { "properties": {
"name": {"title": "Name", "type": "string"}, "name": {"title": "Name", "type": "string"},
@ -51,7 +53,18 @@ openapi_schema = {
def test_openapi_schema(): def test_openapi_schema():
response = client.get("/openapi.json") response = client.get("/openapi.json")
assert response.status_code == 200 assert response.status_code == 200
assert response.json() == openapi_schema # TODO: remove this once Pydantic 1.9 is released
# Ref: https://github.com/samuelcolvin/pydantic/pull/2557
data = response.json()
alternative_data1 = deepcopy(data)
alternative_data2 = deepcopy(data)
alternative_data1["components"]["schemas"]["Item"]["required"] = ["name", "price"]
alternative_data2["components"]["schemas"]["Item"]["required"] = [
"name",
"price",
"tags",
]
assert alternative_data1 == openapi_schema or alternative_data2 == openapi_schema
def test_get_item(): def test_get_item():