Added new model class without relay

Author: eduarde

examples/django_mongoengine/bike/fixtures.py

@@ -1,9 +1,8 @@
 import pytest
-from .models import Bike
+from .models import Bike, Shop
 
 
-@pytest.fixture(scope='module')
-def fixtures_data():
+def fixture_bike_data():
     Bike.drop_collection()
     bike_one = Bike(
         name='Level R',
@@ -25,4 +24,26 @@ def fixtures_data():
     )
     bike_two.save()
 
+
+def fixture_shop_data():
+    Shop.drop_collection()
+    shop_one = Shop(
+        name="Big Wheel Bicycles",
+        address="2438 Hart Ridge Road",
+        website="https://www.bigwheelbike.test"
+    )
+    shop_one.save()
+    shop_two = Shop(
+        name="Bike Tech",
+        address="2175 Pearl Street",
+        website="https://www.biketech.test"
+    )
+    shop_two.save()
+
+
+@pytest.fixture(scope='module')
+def fixtures_data():
+    fixture_bike_data()
+    fixture_shop_data()
+
     return True

examples/django_mongoengine/bike/models.py

@@ -1,5 +1,12 @@
 from mongoengine import Document
-from mongoengine.fields import FloatField, StringField, ListField
+from mongoengine.fields import FloatField, StringField, ListField, URLField
+
+
+class Shop(Document):
+    meta = {'collection': 'shop'}
+    name = StringField()
+    address = StringField()
+    website = URLField()
 
 
 class Bike(Document):

examples/django_mongoengine/bike/schema.py

@@ -1,12 +1,17 @@
 import graphene
 from graphene.relay import Node
 from graphene_mongo.fields import MongoengineConnectionField
-from .types import BikeType
+from .models import Shop
+from .types import BikeType, ShopType
 
 
 class Query(graphene.ObjectType):
     node = Node.Field()
     bikes = MongoengineConnectionField(BikeType)
+    shop_list = graphene.List(ShopType)
+
+    def resolve_shop_list(self, info):
+        return Shop.objects.all()
 
 
 schema = graphene.Schema(query=Query, types=[BikeType, ])

examples/django_mongoengine/bike/tests.py

@@ -76,7 +76,7 @@ def test_bikes_filter_by_type_item_query(fixtures_data):
                                 "wheelSize": 27.5,
                                 "type": "MTB"
                             }
-                        },
+                        }
                     ]
                 }
         }
@@ -85,3 +85,36 @@ def test_bikes_filter_by_type_item_query(fixtures_data):
     client = Client(schema)
     result = client.execute(query)
     assert result == expected
+
+
+def test_shop_data_query(fixtures_data):
+    query = '''{
+               shopList{
+                    name
+                    address
+                    website
+                    }
+            }'''
+
+    expected = {
+        "data": {
+            "shopList": [
+                {
+
+                    "name": "Big Wheel Bicycles",
+                    "address": "2438 Hart Ridge Road",
+                    "website": "https://www.bigwheelbike.test",
+
+                },
+                {
+                    "name": "Bike Tech",
+                    "address": "2175 Pearl Street",
+                    "website": "https://www.biketech.test",
+                }
+            ]
+        }
+    }
+
+    client = Client(schema)
+    result = client.execute(query)
+    assert result == expected

examples/django_mongoengine/bike/types.py

@@ -1,9 +1,14 @@
 from graphene import relay
 from graphene_mongo import MongoengineObjectType
-from .models import Bike
+from .models import Bike, Shop
 
 
 class BikeType(MongoengineObjectType):
     class Meta:
         model = Bike
         interfaces = (relay.Node,)
+
+
+class ShopType(MongoengineObjectType):
+    class Meta:
+        model = Shop