fix reflection for inheritance

Author: SirCotare

src/main/java/it/aboutbits/springboot/toolbox/reflection/util/CustomTypeReflectionUtil.java

@@ -20,15 +20,25 @@ public static <T extends CustomType<?>> Constructor<T> getCustomTypeConstructor(
         }
     }
 
-    public static Class<?> getWrappedType(Class<? extends CustomType<?>> customType) {
-        var customTypeInterface = Arrays.stream(customType.getGenericInterfaces())
-                .filter(i ->
-                                i instanceof ParameterizedType
-                                        && CustomType.class.isAssignableFrom((Class<?>) ((ParameterizedType) i).getRawType())
-                ).findFirst()
-                .map(i -> (ParameterizedType) i)
-                .orElseThrow();
-
-        return (Class<?>) customTypeInterface.getActualTypeArguments()[0];
+    public static Class<?> getWrappedType(Class<? extends CustomType<?>> customType) throws NoSuchMethodException {
+        Class<?> currentClass = customType;
+        while (currentClass != null) {
+            // Check interfaces of the current class
+            var customTypeInterface = Arrays.stream(currentClass.getGenericInterfaces())
+                    .filter(i ->
+                                    i instanceof ParameterizedType
+                                            && CustomType.class.isAssignableFrom((Class<?>) ((ParameterizedType) i).getRawType())
+                    ).findFirst()
+                    .map(i -> (ParameterizedType) i);
+
+            if (customTypeInterface.isPresent()) {
+                return (Class<?>) customTypeInterface.get().getActualTypeArguments()[0];
+            }
+
+            // Move to the parent class
+            currentClass = currentClass.getSuperclass();
+        }
+
+        throw new NoSuchMethodException();
     }
 }

src/main/java/it/aboutbits/springboot/toolbox/swagger/type/CustomTypeModelConverter.java

@@ -9,6 +9,7 @@
 import it.aboutbits.springboot.toolbox.type.CustomType;
 import it.aboutbits.springboot.toolbox.type.ScaledBigDecimal;
 import it.aboutbits.springboot.toolbox.type.identity.EntityId;
+import lombok.SneakyThrows;
 
 import java.math.BigDecimal;
 import java.math.BigInteger;
@@ -17,6 +18,7 @@
 public class CustomTypeModelConverter implements ModelConverter {
 
     @Override
+    @SneakyThrows(NoSuchMethodException.class)
     public Schema<?> resolve(
             AnnotatedType annotatedType,
             ModelConverterContext context,

src/main/java/it/aboutbits/springboot/toolbox/swagger/type/CustomTypePropertyCustomizer.java

@@ -8,6 +8,7 @@
 import it.aboutbits.springboot.toolbox.type.CustomType;
 import it.aboutbits.springboot.toolbox.type.ScaledBigDecimal;
 import it.aboutbits.springboot.toolbox.type.identity.EntityId;
+import lombok.SneakyThrows;
 import lombok.extern.slf4j.Slf4j;
 import org.springdoc.core.customizers.PropertyCustomizer;
 
@@ -17,6 +18,7 @@
 @Slf4j
 public class CustomTypePropertyCustomizer implements PropertyCustomizer {
     @Override
+    @SneakyThrows(NoSuchMethodException.class)
     public Schema<?> customize(Schema property, AnnotatedType annotatedType) {
         var type = annotatedType.getType();
 

src/test/java/it/aboutbits/springboot/toolbox/reflection/util/CustomTypeReflectionUtilTest.java

@@ -44,6 +44,20 @@ void testGetCustomTypeConstructorWithMismatchingConstructor() {
         );
     }
 
+    @Test
+    void testGetCustomTypeConstructorWithInheritedCustomType() throws Exception {
+        // given
+        var customTypeClass = ChildClass.class;
+
+        // when
+        var constructor = CustomTypeReflectionUtil.getCustomTypeConstructor(customTypeClass);
+
+        // then
+        assertThat(constructor).isNotNull();
+        assertThat(constructor.getParameterCount()).isEqualTo(1);
+        assertThat(constructor.getParameterTypes()[0]).isEqualTo(Integer.class);
+    }
+
     @SuppressWarnings("checkstyle:RedundantModifier")
     @CustomTypeScanner.DisableCustomTypeConfiguration
     public static class ValidCustomType implements CustomType<String> {
@@ -88,4 +102,27 @@ public String value() {
             return value;
         }
     }
+
+    @SuppressWarnings("checkstyle:RedundantModifier")
+    @CustomTypeScanner.DisableCustomTypeConfiguration
+    public static class ParentClass implements CustomType<Integer> {
+        private final Integer value;
+
+        public ParentClass(Integer value) {
+            this.value = value;
+        }
+
+        @Override
+        public Integer value() {
+            return value;
+        }
+    }
+
+    @SuppressWarnings("checkstyle:RedundantModifier")
+    @CustomTypeScanner.DisableCustomTypeConfiguration
+    public static class ChildClass extends ParentClass {
+        public ChildClass(Integer value) {
+            super(value);
+        }
+    }
 }