Minor rework of test_field2property_nested_spec_metadatas

marshmallow-code · Dec 2, 2024 · a37c3e5 · a37c3e5
1 parent a52e7bc
commit a37c3e5
Showing 1 changed file with 3 additions and 13 deletions.
diff --git a/tests/test_ext_marshmallow_field.py b/tests/test_ext_marshmallow_field.py
@@ -438,25 +438,15 @@ class Child(Schema):
     )
     result = spec_fixture.openapi.field2property(category)
     version = spec_fixture.openapi.openapi_version
-    if version.major < 3:
+    if version.major < 3 or version.minor < 1:
         assert result == {
-            "allOf": [
-                {"$ref": "#/definitions/Child"},
-            ],
-            "description": "A category",
-            "x-extension": "A great extension",
-        }
-    elif version.minor < 1:
-        assert result == {
-            "allOf": [
-                {"$ref": "#/components/schemas/Child"},
-            ],
+            "allOf": [build_ref(spec_fixture.spec, "schema", "Child")],
             "description": "A category",
             "x-extension": "A great extension",
         }
     else:
         assert result == {
-            "$ref": "#/components/schemas/Child",
+            **build_ref(spec_fixture.spec, "schema", "Child"),
             "description": "A category",
             "x-extension": "A great extension",
         }