Added crud mutations and unittesting

Author: eduarde

examples/django_mongoengine/bike/fixtures.py

@@ -24,6 +24,17 @@ def fixture_bike_data():
     )
     bike_two.save()
 
+    bike_three = Bike(
+        id="507f1f77bcf86cd799439011",
+        name='Moterra Neo',
+        brand='Cannondale',
+        year='2019',
+        size=["M", "L", "XL"],
+        wheel_size=29,
+        type='EBike'
+    )
+    bike_three.save()
+
 
 def fixture_shop_data():
     Shop.drop_collection()

examples/django_mongoengine/bike/models.py

@@ -1,16 +1,21 @@
 from mongoengine import Document
-from mongoengine.fields import FloatField, StringField, ListField, URLField
+from mongoengine.fields import (
+    FloatField, StringField,
+    ListField, URLField, ObjectIdField
+)
 
 
 class Shop(Document):
     meta = {'collection': 'shop'}
+    ID = ObjectIdField()
     name = StringField()
     address = StringField()
     website = URLField()
 
 
 class Bike(Document):
     meta = {'collection': 'bike'}
+    ID = ObjectIdField()
     name = StringField()
     brand = StringField()
     year = StringField()

examples/django_mongoengine/bike/mutations.py

@@ -0,0 +1,80 @@
+import graphene
+from django.core.exceptions import ObjectDoesNotExist
+from .models import Bike
+from .types import BikeType
+
+
+class BikeInput(graphene.InputObjectType):
+    id = graphene.ID()
+    name = graphene.String()
+    brand = graphene.String()
+    year = graphene.String()
+    size = graphene.List(graphene.String)
+    wheel_size = graphene.Float()
+    type = graphene.String()
+
+
+class CreateBikeMutation(graphene.Mutation):
+    bike = graphene.Field(BikeType)
+
+    class Arguments:
+        bike_data = BikeInput(required=True)
+
+    def mutate(self, info, bike_data=None):
+        bike = Bike(
+            name=bike_data.name,
+            brand=bike_data.brand,
+            year=bike_data.year,
+            size=bike_data.size,
+            wheel_size=bike_data.wheel_size,
+            type=bike_data.type
+        )
+        bike.save()
+
+        return CreateBikeMutation(bike=bike)
+
+
+class UpdateBikeMutation(graphene.Mutation):
+    bike = graphene.Field(BikeType)
+
+    class Arguments:
+        bike_data = BikeInput(required=True)
+
+    @staticmethod
+    def get_object(id):
+        return Bike.objects.get(pk=id)
+
+    def mutate(self, info, bike_data=None):
+        bike = UpdateBikeMutation.get_object(bike_data.id)
+        if bike_data.name:
+            bike.name = bike_data.name
+        if bike_data.brand:
+            bike.brand = bike_data.brand
+        if bike_data.year:
+            bike.year = bike_data.year
+        if bike_data.size:
+            bike.size = bike_data.size
+        if bike_data.wheel_size:
+            bike.wheel_size = bike_data.wheel_size
+        if bike_data.type:
+            bike.type = bike_data.type
+
+        bike.save()
+
+        return UpdateBikeMutation(bike=bike)
+
+
+class DeleteBikeMutation(graphene.Mutation):
+    class Arguments:
+        id = graphene.ID(required=True)
+
+    success = graphene.Boolean()
+
+    def mutate(self, info, id):
+        try:
+            Bike.objects.get(pk=id).delete()
+            success = True
+        except ObjectDoesNotExist:
+            success = False
+
+        return DeleteBikeMutation(success=success)

examples/django_mongoengine/bike/schema.py

@@ -3,6 +3,13 @@
 from graphene_mongo.fields import MongoengineConnectionField
 from .models import Shop
 from .types import BikeType, ShopType
+from .mutations import CreateBikeMutation, UpdateBikeMutation, DeleteBikeMutation
+
+
+class Mutations(graphene.ObjectType):
+    create_bike = CreateBikeMutation.Field()
+    update_bike = UpdateBikeMutation.Field()
+    delete_bike = DeleteBikeMutation.Field()
 
 
 class Query(graphene.ObjectType):
@@ -14,4 +21,4 @@ def resolve_shop_list(self, info):
         return Shop.objects.all()
 
 
-schema = graphene.Schema(query=Query, types=[BikeType, ShopType])
+schema = graphene.Schema(query=Query, mutation=Mutations, types=[BikeType, ShopType])

examples/django_mongoengine/bike/tests.py

@@ -1,12 +1,15 @@
+import pytest
+from django.urls import reverse
+from django.test import RequestFactory
 from graphene.test import Client
 from .schema import schema
 from .fixtures import fixtures_data
 
 
-def test_bikes_last_item_query(fixtures_data):
+def test_bikes_first_item_query(fixtures_data):
     query = '''
                {
-               bikes(last: 1){
+               bikes(first: 1){
                    edges {
                    node {
                        name
@@ -27,14 +30,14 @@ def test_bikes_last_item_query(fixtures_data):
                     "edges": [
                         {
                             "node": {
-                                "name": "CAADX ULTEGRA",
-                                "brand": "Cannondale",
-                                "year": '2019',
-                                "size": ['46', '51', '54', '58'],
-                                "wheelSize": 28,
-                                "type": "Gravel"
+                                "name": "Level R",
+                                "brand": "Mondraker",
+                                "year": '2020',
+                                "size": ['S', 'M', 'L', 'XL'],
+                                "wheelSize": 27.5,
+                                "type": "MTB"
                             }
-                        },
+                        }
                     ]
                 }
         }
@@ -48,7 +51,7 @@ def test_bikes_last_item_query(fixtures_data):
 def test_bikes_filter_by_type_item_query(fixtures_data):
     query = '''
                {
-               bikes(first: 2, type: "MTB"){
+               bikes(first: 2, type: "Gravel"){
                    edges {
                    node {
                        name
@@ -69,12 +72,12 @@ def test_bikes_filter_by_type_item_query(fixtures_data):
                     "edges": [
                         {
                             "node": {
-                                "name": "Level R",
-                                "brand": "Mondraker",
-                                "year": '2020',
-                                "size": ['S', 'M', 'L', 'XL'],
-                                "wheelSize": 27.5,
-                                "type": "MTB"
+                                "name": "CAADX ULTEGRA",
+                                "brand": "Cannondale",
+                                "year": '2019',
+                                "size": ['46', '51', '54', '58'],
+                                "wheelSize": 28,
+                                "type": "Gravel"
                             }
                         }
                     ]
@@ -118,3 +121,122 @@ def test_shop_data_query(fixtures_data):
     client = Client(schema)
     result = client.execute(query)
     assert result == expected
+
+
+@pytest.mark.django_db
+def test_create_bike_mutation():
+    query = '''
+                    mutation {
+                        createBike(bikeData:{
+                                name:"Bullhorn",
+                                brand:"Pegas",
+                                year: "2019",
+                                size: ["56", "58" ],
+                                wheelSize: 28,
+                                type: "Fixie"
+                                }) {
+                            bike {
+                                name
+                                brand
+                                year
+                                size
+                                wheelSize
+                                type
+                            }
+                        }
+                    }
+                  '''
+
+    expected = {
+        "data": {
+            "createBike": {
+                "bike":
+                    {
+                        "name": "Bullhorn",
+                        "brand": "Pegas",
+                        "year": "2019",
+                        "size": ["56", "58"],
+                        "wheelSize": 28,
+                        "type": "Fixie",
+                    }
+            }
+        }
+    }
+
+    factory = RequestFactory()
+    request = factory.post(reverse('graphql-query'))
+    client = Client(schema)
+    result = client.execute(query, context=request)
+    assert result == expected
+
+
+@pytest.mark.django_db
+def test_update_bike_mutation():
+    query = '''
+                    mutation {
+                        updateBike(bikeData:{
+                                id: "507f1f77bcf86cd799439011",
+                                name:"Moterra Neo Updated",
+                                year: "2020",
+                                wheelSize: 27.5,
+                                type: "EBike Updated"
+                                }) {
+                            bike {
+                                name
+                                brand
+                                year
+                                size
+                                wheelSize
+                                type
+                            }
+                        }
+                    }
+                  '''
+
+    expected = {
+        "data": {
+            "updateBike": {
+                "bike":
+                    {
+                        "name": "Moterra Neo Updated",
+                        "brand": "Cannondale",
+                        "year": "2020",
+                        "size": ["M", "L", "XL"],
+                        "wheelSize": 27.5,
+                        "type": "EBike Updated"
+                    }
+            }
+        }
+    }
+
+    factory = RequestFactory()
+    request = factory.post(reverse('graphql-query'))
+    client = Client(schema)
+    result = client.execute(query, context=request)
+    print(result)
+    assert result == expected
+
+
+@pytest.mark.django_db
+def test_delete_bike_mutation():
+    query = '''
+                mutation {
+                    deleteBike(id: "507f1f77bcf86cd799439011") {
+                           success
+                    }
+                }
+              '''
+
+    expected = {
+        "data": {
+            "deleteBike": {
+                "success": True
+            }
+        }
+    }
+
+    factory = RequestFactory()
+    request = factory.post(reverse('graphql-query'))
+    client = Client(schema)
+    result = client.execute(query, context=request)
+    assert result == expected

examples/django_mongoengine/bike/urls.py

@@ -3,5 +3,5 @@
 from graphene_django.views import GraphQLView
 
 urlpatterns = [
-    path('graphql', csrf_exempt(GraphQLView.as_view(graphiql=True))),
+    path('graphql', csrf_exempt(GraphQLView.as_view(graphiql=True)), name='graphql-query'),
 ]