Including Set as possible GraphQLType (#57)

Author: lassetyr

src/main/java/graphql/annotations/DefaultTypeFunction.java

@@ -139,6 +139,29 @@ public GraphQLType apply(Class<?> aClass, AnnotatedType annotatedType) {
         }
     }
 
+    private class SetFunction implements TypeFunction {
+
+        @Override
+        public GraphQLType apply(Class<?> aClass, AnnotatedType annotatedType) {
+            if (!(annotatedType instanceof AnnotatedParameterizedType)) {
+                throw new IllegalArgumentException("List type parameter should be specified");
+            }
+            AnnotatedParameterizedType parameterizedType = (AnnotatedParameterizedType) annotatedType;
+            AnnotatedType arg = parameterizedType.getAnnotatedActualTypeArguments()[0];
+            Class<?> klass;
+            if (arg.getType() instanceof ParameterizedType) {
+                klass = (Class<?>)((ParameterizedType)(arg.getType())).getRawType();
+            } else {
+                klass = (Class<?>) arg.getType();
+            }
+            return new GraphQLList(DefaultTypeFunction.this.apply(klass, arg));
+        }
+
+        @Override public Collection<Class<?>> getAcceptedTypes() {
+            return Arrays.asList(Set.class);
+        }
+    }
+
     private class StreamFunction implements TypeFunction {
 
         @Override
@@ -266,6 +289,7 @@ public DefaultTypeFunction() {
 
         register(new ListFunction());
         register(new StreamFunction());
+        register(new SetFunction());
 
         register(new EnumFunction());
 

src/test/java/graphql/annotations/DefaultTypeFunctionTest.java

@@ -18,10 +18,7 @@
 import graphql.schema.GraphQLType;
 import org.testng.annotations.Test;
 
-import java.util.Arrays;
-import java.util.LinkedList;
-import java.util.List;
-import java.util.Optional;
+import java.util.*;
 import java.util.stream.Collectors;
 import java.util.stream.Stream;
 
@@ -92,6 +89,9 @@ public void long_() {
     @SuppressWarnings("unused")
     public Stream<List<@GraphQLNonNull String>> streamMethod() { return null;}
 
+    @SuppressWarnings("unused")
+    public Set<Set<@GraphQLNonNull String>> setMethod() { return null;}
+
     @Test
     public void list() throws NoSuchMethodException {
         DefaultTypeFunction instance = new DefaultTypeFunction();
@@ -103,6 +103,17 @@ public void list() throws NoSuchMethodException {
         assertEquals(wrappedType.getWrappedType(), GraphQLString);
     }
 
+    @Test
+    public void set() throws NoSuchMethodException {
+        DefaultTypeFunction instance = new DefaultTypeFunction();
+        graphql.schema.GraphQLType type = instance.apply(getClass().getMethod("setMethod").getReturnType(), getClass().getMethod("setMethod").getAnnotatedReturnType());
+        assertTrue(type instanceof GraphQLList);
+        GraphQLList subtype = (GraphQLList) ((GraphQLList) type).getWrappedType();
+        assertTrue(subtype.getWrappedType() instanceof graphql.schema.GraphQLNonNull);
+        graphql.schema.GraphQLNonNull wrappedType = (graphql.schema.GraphQLNonNull) subtype.getWrappedType();
+        assertEquals(wrappedType.getWrappedType(), GraphQLString);
+    }
+
     @Test(expectedExceptions = IllegalArgumentException.class)
     public void unparametrizedList() {
         DefaultTypeFunction instance = new DefaultTypeFunction();