Allows subclasses to be passed into withPrefabValuesForField()

jqno · Dec 4, 2024 · cb0f83e · cb0f83e
1 parent 829767c
commit cb0f83e
Show file tree

Hide file tree

Showing 2 changed files with 39 additions and 5 deletions.
diff --git a/equalsverifier-core/src/main/java/nl/jqno/equalsverifier/internal/util/Validations.java b/equalsverifier-core/src/main/java/nl/jqno/equalsverifier/internal/util/Validations.java
@@ -96,12 +96,12 @@ public static <T> void validateFieldTypeMatches(
     ) {
         try {
             Field f = container.getDeclaredField(fieldName);
-            boolean sameFields = f.getType().equals(fieldType);
-            boolean compatibleFields = fieldType.equals(
+            boolean typeCompatible = f.getType().isAssignableFrom(fieldType);
+            boolean wrappingCompatible = fieldType.equals(
                 PrimitiveMappers.PRIMITIVE_OBJECT_MAPPER.get(f.getType())
             );
             validate(
-                !sameFields && !compatibleFields,
+                !typeCompatible && !wrappingCompatible,
                 "Prefab values for field " +
                 fieldName +
                 " should be of type " +

diff --git a/...est/java/nl/jqno/equalsverifier/integration/operational/WithPrefabValuesForFieldTest.java b/...est/java/nl/jqno/equalsverifier/integration/operational/WithPrefabValuesForFieldTest.java
@@ -2,8 +2,7 @@
 
 import java.time.LocalDate;
 import java.time.Month;
-import java.util.Arrays;
-import java.util.Objects;
+import java.util.*;
 import nl.jqno.equalsverifier.EqualsVerifier;
 import nl.jqno.equalsverifier.Warning;
 import nl.jqno.equalsverifier.internal.testhelpers.ExpectedException;
@@ -166,6 +165,18 @@ public void succeed_whenPrefabForArrayIsOverridden() {
             .verify();
     }
 
+    @Test
+    public void succeed_whenClassContainsSomethingThatAllowsSubclassesAndASubclassIsGiven() {
+        EqualsVerifier
+            .forClass(ListContainer.class)
+            .withPrefabValuesForField(
+                "list",
+                Collections.singletonList("x"),
+                Collections.singletonList("y")
+            )
+            .verify();
+    }
+
     static final class SinglePrecondition {
 
         private final FinalPoint point;
@@ -274,4 +285,27 @@ public int hashCode() {
             return Arrays.hashCode(field);
         }
     }
+
+    static final class ListContainer {
+
+        private final List<String> list;
+
+        public ListContainer(List<String> list) {
+            this.list = list;
+        }
+
+        @Override
+        public boolean equals(Object obj) {
+            if (!(obj instanceof ListContainer)) {
+                return false;
+            }
+            ListContainer other = (ListContainer) obj;
+            return Objects.equals(list, other.list);
+        }
+
+        @Override
+        public int hashCode() {
+            return Objects.hash(list);
+        }
+    }
 }