Fix issue when a response was an array and response body was not matc…

…hing properly
Endava · Nov 9, 2020 · 1b3461a · 1b3461a
1 parent 5e8edf4
commit 1b3461a
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/main/java/com/endava/cats/model/factory/FuzzingDataFactory.java b/src/main/java/com/endava/cats/model/factory/FuzzingDataFactory.java
@@ -441,7 +441,12 @@ private String extractResponseSchemaRef(Operation operation, String responseCode
             return apiResponse.get$ref();
         }
         if (apiResponse.getContent() != null && apiResponse.getContent().get(MimeTypeUtils.APPLICATION_JSON_VALUE) != null) {
-            return apiResponse.getContent().get(MimeTypeUtils.APPLICATION_JSON_VALUE).getSchema().get$ref();
+            Schema respSchema = apiResponse.getContent().get(MimeTypeUtils.APPLICATION_JSON_VALUE).getSchema();
+            if (respSchema instanceof ArraySchema) {
+                return ((ArraySchema) respSchema).getItems().get$ref();
+            } else {
+                return respSchema.get$ref();
+            }
         }
         return null;
     }