Improved mutation examples

Author: syrusakbary

examples/complex_example.py

@@ -5,6 +5,10 @@ class GeoInput(graphene.InputObjectType):
     lat = graphene.Float(required=True)
     lng = graphene.Float(required=True)
 
+    @property
+    def latlng(self):
+        return "({},{})".format(self.lat, self.lng)
+
 
 class Address(graphene.ObjectType):
     latlng = graphene.String()
@@ -14,17 +18,39 @@ class Query(graphene.ObjectType):
     address = graphene.Field(Address, geo=GeoInput(required=True))
 
     def resolve_address(self, info, geo):
-        return Address(latlng="({},{})".format(geo.get('lat'), geo.get('lng')))
+        return Address(latlng=geo.latlng)
+
+
+class CreateAddress(graphene.Mutation):
+    
+    class Arguments:
+        geo = GeoInput(required=True)
+
+    Output = Address
+
+    def mutate(self, info, geo):
+        return Address(latlng=geo.latlng)
 
 
-schema = graphene.Schema(query=Query)
+class Mutation(graphene.ObjectType):
+    create_address = CreateAddress.Field()
+
+
+schema = graphene.Schema(query=Query, mutation=Mutation)
 query = '''
     query something{
       address(geo: {lat:32.2, lng:12}) {
         latlng
       }
     }
 '''
+mutation = '''
+    mutation addAddress{
+      createAddress(geo: {lat:32.2, lng:12}) {
+        latlng
+      }
+    }
+'''
 
 
 def test_query():
@@ -37,6 +63,16 @@ def test_query():
     }
 
 
+def test_mutation():
+    result = schema.execute(mutation)
+    assert not result.errors
+    assert result.data == {
+        'createAddress': {
+            'latlng': "(32.2,12.0)",
+        }
+    }
+
+
 if __name__ == '__main__':
     result = schema.execute(query)
     print(result.data['address']['latlng'])

graphene/types/tests/test_mutation.py

@@ -54,7 +54,7 @@ class User(ObjectType):
 
     class CreateUser(Mutation):
 
-        class Input:
+        class Arguments:
             name = String()
 
         Output = User
@@ -73,7 +73,7 @@ def mutate(self, info, name):
 def test_mutation_execution():
     class CreateUser(Mutation):
 
-        class Input:
+        class Arguments:
             name = String()
             dynamic = Dynamic(lambda: String())
             dynamic_none = Dynamic(lambda: None)