Allow mutations to be required. Improved testing. Fixed #694

Author: syrusakbary

graphene/types/mutation.py

@@ -73,15 +73,17 @@ def __init_subclass_with_meta__(cls, resolver=None, output=None, arguments=None,
         _meta.resolver = resolver
         _meta.arguments = arguments
 
-        super(Mutation, cls).__init_subclass_with_meta__(_meta=_meta, **options)
+        super(Mutation, cls).__init_subclass_with_meta__(
+            _meta=_meta, **options)
 
     @classmethod
-    def Field(cls, name=None, description=None, deprecation_reason=None):
+    def Field(cls, name=None, description=None, deprecation_reason=None, required=False):
         return Field(
             cls._meta.output,
             args=cls._meta.arguments,
             resolver=cls._meta.resolver,
             name=name,
             description=description,
             deprecation_reason=deprecation_reason,
+            required=required,
         )

graphene/types/tests/test_mutation.py

@@ -6,6 +6,7 @@
 from ..objecttype import ObjectType
 from ..scalars import String
 from ..schema import Schema
+from ..structures import NonNull
 
 
 def test_generate_mutation_no_args():
@@ -133,3 +134,27 @@ class MyMutation(ObjectType):
             'name': None,
         }
     }
+
+
+def test_mutation_allow_to_have_custom_args():
+    class CreateUser(Mutation):
+
+        class Arguments:
+            name = String()
+
+        name = String()
+
+        def mutate(self, info, name):
+            return CreateUser(name=name)
+
+    class MyMutation(ObjectType):
+        create_user = CreateUser.Field(
+            description='Create a user',
+            deprecation_reason='Is deprecated',
+            required=True
+        )
+
+    field = MyMutation._meta.fields['create_user']
+    assert field.description == 'Create a user'
+    assert field.deprecation_reason == 'Is deprecated'
+    assert field.type == NonNull(CreateUser)