Fix issue when additionalProperties had a chain of additionalProperti…

…es with object schema
Endava · Dec 6, 2020 · 5e01745 · 5e01745
1 parent 0c062bf
commit 5e01745
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/main/java/com/endava/cats/generator/simple/PayloadGenerator.java b/src/main/java/com/endava/cats/generator/simple/PayloadGenerator.java
@@ -180,6 +180,10 @@ private Object getExampleFromAdditionalPropertiesSchema(String propertyName, Str
         if (property.getName() != null) {
             mp.put(property.getName(),
                     resolvePropertyToExample(propertyName, mediaType, (Schema) property.getAdditionalProperties()));
+        } else if (((Schema) property.getAdditionalProperties()).get$ref() != null) {
+            Schema innerSchema = (Schema) property.getAdditionalProperties();
+            Schema addPropSchema = this.schemaMap.get(innerSchema.get$ref().substring(innerSchema.get$ref().lastIndexOf('/') + 1));
+            mp.put("key", resolvePropertyToExample(propertyName, mediaType, addPropSchema));
         } else {
             mp.put("key",
                     resolvePropertyToExample(propertyName, mediaType, (Schema) property.getAdditionalProperties()));