added set generators

Author: oubaydos

src/main/java/io/javarig/TypeEnum.java

@@ -4,6 +4,9 @@
 import io.javarig.generator.collection.list.ArrayListGenerator;
 import io.javarig.generator.collection.map.HashMapGenerator;
 import io.javarig.generator.collection.map.TreeMapGenerator;
+import io.javarig.generator.collection.set.HashSetGenerator;
+import io.javarig.generator.collection.set.LinkedHashSetGenerator;
+import io.javarig.generator.collection.set.TreeSetGenerator;
 import io.javarig.generator.primitive.*;
 
 import java.lang.reflect.Type;
@@ -31,6 +34,10 @@ public enum TypeEnum {
     TREE_MAP(TreeMap.class, TreeMapGenerator.class),
     LIST(List.class, ArrayListGenerator.class),
     ARRAY_LIST(ArrayList.class, ArrayListGenerator.class),
+    SET(Set.class, HashSetGenerator.class),
+    HASH_SET(HashSet.class, HashSetGenerator.class),
+    LINKED_HASH_SET(LinkedHashSet.class, LinkedHashSetGenerator.class),
+    TREE_SET(TreeSet.class, TreeSetGenerator.class),
     ENUM(null, EnumGenerator.class),
     ARRAY(null, ArrayGenerator.class),
     OBJECT(null, ObjectGenerator.class);

src/main/java/io/javarig/generator/collection/set/HashSetGenerator.java

@@ -0,0 +1,19 @@
+package io.javarig.generator.collection.set;
+
+import io.javarig.RandomInstanceGenerator;
+
+import java.lang.reflect.Type;
+import java.util.HashSet;
+import java.util.Set;
+
+public class HashSetGenerator extends SetGenerator{
+    public HashSetGenerator(Type type, RandomInstanceGenerator randomInstanceGenerator) {
+        super(type, randomInstanceGenerator);
+    }
+
+    @SuppressWarnings({"rawtypes"})
+    @Override
+    public Class<? extends Set> getImplementationType() {
+        return HashSet.class;
+    }
+}

src/main/java/io/javarig/generator/collection/set/LinkedHashSetGenerator.java

@@ -0,0 +1,19 @@
+package io.javarig.generator.collection.set;
+
+import io.javarig.RandomInstanceGenerator;
+
+import java.lang.reflect.Type;
+import java.util.LinkedHashSet;
+import java.util.Set;
+
+public class LinkedHashSetGenerator extends SetGenerator{
+    public LinkedHashSetGenerator(Type type, RandomInstanceGenerator randomInstanceGenerator) {
+        super(type, randomInstanceGenerator);
+    }
+
+    @SuppressWarnings({"rawtypes"})
+    @Override
+    public Class<? extends Set> getImplementationType() {
+        return LinkedHashSet.class;
+    }
+}

src/main/java/io/javarig/generator/collection/set/TreeSetGenerator.java

@@ -0,0 +1,19 @@
+package io.javarig.generator.collection.set;
+
+import io.javarig.RandomInstanceGenerator;
+
+import java.lang.reflect.Type;
+import java.util.Set;
+import java.util.TreeSet;
+
+public class TreeSetGenerator extends SetGenerator{
+    public TreeSetGenerator(Type type, RandomInstanceGenerator randomInstanceGenerator) {
+        super(type, randomInstanceGenerator);
+    }
+
+    @SuppressWarnings({"rawtypes"})
+    @Override
+    public Class<? extends Set> getImplementationType() {
+        return TreeSet.class;
+    }
+}