Added NonNull and List to Types

Author: syrusakbary

graphene/core/types/base.py

@@ -82,6 +82,16 @@ def __init__(self, *args, **kwargs):
         self.args = args
         self.kwargs = kwargs
 
+    @property
+    def List(self):  # noqa
+        from .definitions import List
+        return List(self, *self.args, **self.kwargs)
+
+    @property
+    def NonNull(self):  # noqa
+        from .definitions import NonNull
+        return NonNull(self, *self.args, **self.kwargs)
+
 
 class ArgumentType(MirroredType):
 

graphene/core/types/objecttype.py

@@ -10,6 +10,7 @@
 from graphene.core.options import Options
 from graphene.core.types.argument import ArgumentsGroup
 from graphene.core.types.base import BaseType
+from graphene.core.types.definitions import List, NonNull
 from graphql.core.type import (GraphQLArgument, GraphQLInputObjectType,
                                GraphQLInterfaceType, GraphQLObjectType)
 
@@ -99,6 +100,9 @@ def __new__(cls, name, bases, attrs):
                 new_class._meta.interfaces.append(base)
             # new_class._meta.parents.extend(base._meta.parents)
 
+        setattr(new_class, 'NonNull', NonNull(new_class))
+        setattr(new_class, 'List', List(new_class))
+
         new_class._prepare()
         return new_class
 

graphene/core/types/tests/test_base.py

@@ -5,6 +5,7 @@
 from ..argument import Argument
 from ..base import MountedType, OrderedType
 from ..field import Field, InputField
+from ..definitions import List, NonNull
 
 
 def test_orderedtype_equal():
@@ -71,3 +72,23 @@ def test_type_as_argument():
     a = MountedType(description='A')
     argument = a.as_argument()
     assert isinstance(argument, Argument)
+
+
+def test_type_as_list():
+    m = MountedType(2, 3, my_c='A')
+    a = m.List
+
+    assert isinstance(a, List)
+    assert a.of_type == m
+    assert a.args == (2, 3)
+    assert a.kwargs == {'my_c': 'A'}
+
+
+def test_type_as_nonnull():
+    m = MountedType(2, 3, my_c='A')
+    a = m.NonNull
+
+    assert isinstance(a, NonNull)
+    assert a.of_type == m
+    assert a.args == (2, 3)
+    assert a.kwargs == {'my_c': 'A'}

graphene/core/types/tests/test_objecttype.py

@@ -112,3 +112,17 @@ def resolve_name(self, *args):
 
     field = schema.T(Droid._meta.fields_map['name'])
     assert resolver_has_tag(field.resolver, 'test')
+
+
+def test_type_has_nonnull():
+    class Droid(Character):
+        name = String()
+
+    assert Droid.NonNull.of_type == Droid
+
+
+def test_type_has_list():
+    class Droid(Character):
+        name = String()
+
+    assert Droid.List.of_type == Droid