From 29d7d899dc6bf94c9dbd002d7780d195581f0726 Mon Sep 17 00:00:00 2001
From: Mustapha Zorgati <15628173+mustaphazorgati@users.noreply.github.com>
Date: Fri, 20 Nov 2020 22:37:54 +0100
Subject: [PATCH] #423: fixed Enum Constraint message for array types

---
 .../scalable/restdocs/constraints/ConstraintReaderImpl.java   | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/spring-auto-restdocs-core/src/main/java/capital/scalable/restdocs/constraints/ConstraintReaderImpl.java b/spring-auto-restdocs-core/src/main/java/capital/scalable/restdocs/constraints/ConstraintReaderImpl.java
index f2113996..40e1b849 100644
--- a/spring-auto-restdocs-core/src/main/java/capital/scalable/restdocs/constraints/ConstraintReaderImpl.java
+++ b/spring-auto-restdocs-core/src/main/java/capital/scalable/restdocs/constraints/ConstraintReaderImpl.java
@@ -134,6 +134,8 @@ private List<String> getEnumConstraintMessage(Class<?> javaBaseClass, String jav
 
         if (field.getType().isEnum()) {
             return getEnumConstraintMessage(field.getType());
+        } else if (field.getType().isArray()) {
+            return getEnumConstraintMessage(field.getType().getComponentType());
         } else {
             return getEnumConstraintMessage(firstGenericType(field.getGenericType(), javaBaseClass));
         }
@@ -142,6 +144,8 @@ private List<String> getEnumConstraintMessage(Class<?> javaBaseClass, String jav
     private List<String> getEnumConstraintMessage(MethodParameter param) {
         if (param.getParameterType().isEnum()) {
             return getEnumConstraintMessage(param.getParameterType());
+        } else if (param.getParameterType().isArray()) {
+            return getEnumConstraintMessage(param.getParameterType().getComponentType());
         } else {
             return getEnumConstraintMessage(firstGenericType(param));
         }