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

PrettyWood · tiangolo · web-flow · commit b0cd4d7e7ebd · 2021-12-12T12:28:35.000+01:00
Co-authored-by: Sebastián Ramírez &lt;tiangolo@gmail.com&gt;
diff --git a/tests/test_tutorial/test_dataclasses/test_tutorial002.py b/tests/test_tutorial/test_dataclasses/test_tutorial002.py
@@ -1,3 +1,5 @@
+from copy import deepcopy
+
 from fastapi.testclient import TestClient
 
 from docs_src.dataclasses.tutorial002 import app
@@ -29,7 +31,7 @@
         "schemas": {
             "Item": {
                 "title": "Item",
-                "required": ["name", "price", "tags"],
+                "required": ["name", "price"],
                 "type": "object",
                 "properties": {
                     "name": {"title": "Name", "type": "string"},
@@ -51,7 +53,18 @@
 def test_openapi_schema():
     response = client.get("/openapi.json")
     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():
