test: Add Parent and Child model.

Author: abawchen

graphene_mongo/tests/fixtures.py

@@ -1,4 +1,7 @@
-from .models import Article, Editor, EmbeddedArticle, Player, Reporter
+from .models import (
+    Article, Editor, EmbeddedArticle, Player,
+    Reporter, Parent, Child
+)
 
 
 def setup_fixtures():
@@ -74,3 +77,10 @@ def setup_fixtures():
 
     player2.players = [player1]
     player2.save()
+
+    Child.drop_collection()
+    child1 = Child(bar='BAR', baz='BAZ')
+    child1.save()
+
+    child2 = Child(bar='bar', baz='baz')
+    child2.save()

graphene_mongo/tests/models.py

@@ -45,8 +45,8 @@ class EmbeddedArticle(EmbeddedDocument):
 
 
 class Reporter(Document):
-    meta = {'collection': 'test_repoter'}
 
+    meta = {'collection': 'test_repoter'}
     id = StringField(primary_key=True)
     first_name = StringField(required=True)
     last_name = StringField(required=True)
@@ -65,3 +65,18 @@ class Player(Document):
     articles = ListField(ReferenceField('Article'))
     embedded_list_articles = EmbeddedDocumentListField(EmbeddedArticle)
 
+
+class Parent(Document):
+
+    meta = {
+        'collection': 'test_parent',
+        'allow_inheritance': True
+    }
+    bar = StringField()
+
+
+class Child(Parent):
+
+    meta = {'collection': 'Child'}
+    baz = StringField()
+