Add an ObjectGenerator<>, and related machinery (also a mapOf generator)

1 files changed, 44 insertions, 0 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
new file mode 100644
index 0000000..bdf2ce8
--- /dev/null
+++ b/src/test/java/au/id/zancanaro/javacheck/object/SubObject.java
@@ -0,0 +1,44 @@
+package au.id.zancanaro.javacheck.object;
+
+import java.util.*;
+
+public class SubObject<T> {
+    public final Map<String, List<T>> obj;
+
+    public SubObject(Map<String, List<T>> obj) {
+        this.obj = obj;
+    }
+
+    public SubObject<T> add(SubObject<T> other) {
+        Map<String, List<T>> values = new HashMap<>(obj);
+        for (Map.Entry<String, List<T>> entry : other.obj.entrySet()) {
+            String key = entry.getKey();
+            if (values.containsKey(key)) {
+                List<T> result = new ArrayList<>(values.get(key));
+                result.addAll(entry.getValue());
+                values.put(key, Collections.unmodifiableList(result));
+            } else {
+                values.put(key, entry.getValue());
+            }
+        }
+        return new SubObject<>(Collections.unmodifiableMap(values));
+    }
+
+    @Override
+    public String toString() {
+        return obj.toString();
+    }
+
+    @Override
+    public boolean equals(Object o) {
+        if (this == o) return true;
+        if (o == null || getClass() != o.getClass()) return false;
+        SubObject subObject = (SubObject) o;
+        return obj.equals(subObject.obj);
+    }
+
+    @Override
+    public int hashCode() {
+        return obj.hashCode();
+    }
+}