Support enums in the ObjectGeneration stuff

1 files changed, 11 insertions, 5 deletions

diff --git a/src/test/java/au/id/zancanaro/javacheck/object/SubObject.java b/src/test/java/au/id/zancanaro/javacheck/object/SubObject.java
index bdf2ce8..1e49cfe 100644
--- a/src/test/java/au/id/zancanaro/javacheck/object/SubObject.java
+++ b/src/test/java/au/id/zancanaro/javacheck/object/SubObject.java
@@ -4,9 +4,11 @@ import java.util.*;
 
 public class SubObject<T> {
     public final Map<String, List<T>> obj;
+    public final EnumValues value;
 
-    public SubObject(Map<String, List<T>> obj) {
+    public SubObject(Map<String, List<T>> obj, EnumValues value) {
         this.obj = obj;
+        this.value = value;
     }
 
     public SubObject<T> add(SubObject<T> other) {
@@ -21,12 +23,12 @@ public class SubObject<T> {
                 values.put(key, entry.getValue());
             }
         }
-        return new SubObject<>(Collections.unmodifiableMap(values));
+        return new SubObject<>(Collections.unmodifiableMap(values), this.value);
     }
 
     @Override
     public String toString() {
-        return obj.toString();
+        return "{" + obj.toString() + ", " + this.value + "}";
     }
 
     @Override
@@ -34,11 +36,15 @@ public class SubObject<T> {
         if (this == o) return true;
         if (o == null || getClass() != o.getClass()) return false;
         SubObject subObject = (SubObject) o;
-        return obj.equals(subObject.obj);
+        return obj.equals(subObject.obj) && value.equals(subObject.value);
     }
 
     @Override
     public int hashCode() {
-        return obj.hashCode();
+        return obj.hashCode() + 37 * value.hashCode();
+    }
+
+    public static enum EnumValues {
+        A, B, C
     }
 }