port #409 to the OAS 3.0 module

Author: mattpolzin

Sources/OpenAPIKit30/Schema Object/JSONSchema.swift

@@ -1762,6 +1762,9 @@ extension JSONSchema: Decodable {
             }
 
             self = schema
+
+            // Ad-hoc vendor extension support since JSONSchema does coding keys differently.
+            self.vendorExtensions = try Self.decodeVenderExtensions(from: decoder)
             return
         }
 
@@ -1775,6 +1778,9 @@ extension JSONSchema: Decodable {
             }
 
             self = schema
+
+            // Ad-hoc vendor extension support since JSONSchema does coding keys differently.
+            self.vendorExtensions = try Self.decodeVenderExtensions(from: decoder)
             return
         }
 
@@ -1788,6 +1794,9 @@ extension JSONSchema: Decodable {
             }
 
             self = schema
+
+            // Ad-hoc vendor extension support since JSONSchema does coding keys differently.
+            self.vendorExtensions = try Self.decodeVenderExtensions(from: decoder)
             return
         }
 
@@ -1798,6 +1807,9 @@ extension JSONSchema: Decodable {
             )
 
             self = schema
+
+            // Ad-hoc vendor extension support since JSONSchema does coding keys differently.
+            self.vendorExtensions = try Self.decodeVenderExtensions(from: decoder)
             return
         }
 
@@ -1888,25 +1900,27 @@ extension JSONSchema: Decodable {
 
         self.warnings = _warnings
 
-        // Ad-hoc vendor extension support since JSONSchema does coding keys differently. 
+        self.vendorExtensions = try Self.decodeVenderExtensions(from: decoder)
+    }
+
+    private static func decodeVenderExtensions(from decoder: Decoder) throws -> [String: AnyCodable] {
         guard VendorExtensionsConfiguration.isEnabled else {
-            self.vendorExtensions = [:]
-            return
+            return [:]
         }
-
+        
         let decoded = try AnyCodable(from: decoder).value
-
+        
         guard (decoded as? [Any]) == nil else {
             throw VendorExtensionDecodingError.selfIsArrayNotDict
         }
-
+        
         guard let decodedAny = decoded as? [String: Any] else {
             throw VendorExtensionDecodingError.foundNonStringKeys
         }
-
-        let extensions = decodedAny.filter { $0.key.lowercased().starts(with: "x-") }
-
-        self.vendorExtensions = extensions.mapValues(AnyCodable.init)
+        
+        return decodedAny
+            .filter { $0.key.lowercased().starts(with: "x-") }
+            .mapValues(AnyCodable.init)
     }
 }
 

Tests/OpenAPIKit30Tests/Schema Object/JSONSchemaTests.swift

@@ -1508,6 +1508,64 @@ extension SchemaObjectTests {
         )
     }
 
+    func test_decodeAnyOfWithVendorExtension() throws {
+        let extensionSchema = """
+        {
+            "anyOf" : [
+                { "type": "string" },
+                { "type": "number" }
+            ],
+            "x-hello" : "hello"
+        }
+        """.data(using: .utf8)!
+
+        var desiredSchema = JSONSchema.one(of: [JSONSchema.string, JSONSchema.number])
+        desiredSchema.vendorExtensions = ["x-hello": "hello"]
+        XCTAssertEqual(
+            try orderUnstableDecode(JSONSchema.self, from: extensionSchema),
+            desiredSchema
+        )
+    }
+    
+    func test_decodeAllOfWithVendorExtension() throws {
+        let extensionSchema = """
+        {
+            "allOf" : [
+                { "type": "string" },
+                { "type": "number" }
+            ],
+            "x-hello" : "hello"
+        }
+        """.data(using: .utf8)!
+
+        var desiredSchema = JSONSchema.one(of: [JSONSchema.string, JSONSchema.number])
+        desiredSchema.vendorExtensions = ["x-hello": "hello"]
+        XCTAssertEqual(
+            try orderUnstableDecode(JSONSchema.self, from: extensionSchema),
+            desiredSchema
+        )
+    }
+    
+    func test_decodeOneOfWithVendorExtension() throws {
+        let extensionSchema = """
+        {
+            "oneOf" : [
+                { "type": "string" },
+                { "type": "number" }
+            ],
+            "x-hello" : "hello"
+        }
+        """.data(using: .utf8)!
+
+        var desiredSchema = JSONSchema.one(of: [JSONSchema.string, JSONSchema.number])
+        desiredSchema.vendorExtensions = ["x-hello": "hello"]
+        XCTAssertEqual(
+            try orderUnstableDecode(JSONSchema.self, from: extensionSchema),
+            desiredSchema
+        )
+    }
+
+
     func test_encodeBoolean() {
         let requiredBoolean = JSONSchema.boolean(.init(format: .unspecified, required: true))
         let optionalBoolean = JSONSchema.boolean(.init(format: .unspecified, required: false))

Tests/OpenAPIKitTests/Schema Object/JSONSchemaTests.swift

@@ -1918,6 +1918,24 @@ extension SchemaObjectTests {
             JSONSchema.fragment(.init(examples: ["hello"])).with(vendorExtensions: ["x-hello": "hello"])
         )
     }
+
+    func test_encodeExamplesVendorExtension() throws {
+        let fragment = JSONSchema.fragment(.init(examples: ["hello"])).with(vendorExtensions: ["x-hello": "hello"])
+
+        let encoded = try orderUnstableTestStringFromEncoding(of: fragment)
+
+        assertJSONEquivalent(
+            encoded,
+            """
+            {
+              "examples" : [
+                "hello"
+              ],
+              "x-hello" : "hello"
+            }
+            """
+        )
+    }
 
     func test_decodeAnyOfWithVendorExtension() throws {
         let extensionSchema = """
@@ -1970,24 +1988,6 @@ extension SchemaObjectTests {
         )
     }
 
-    func test_encodeExamplesVendorExtension() throws {
-        let fragment = JSONSchema.fragment(.init(examples: ["hello"])).with(vendorExtensions: ["x-hello": "hello"])
-
-        let encoded = try orderUnstableTestStringFromEncoding(of: fragment)
-
-        assertJSONEquivalent(
-            encoded,
-            """
-            {
-              "examples" : [
-                "hello"
-              ],
-              "x-hello" : "hello"
-            }
-            """
-        )
-    }
-
     func test_decodeNullType() throws {
         let nullTypeData = """
         {