test: Add corresponding test in test_query and test_relay_query.

Author: abawchen

graphene_mongo/tests/fixtures.py

@@ -1,4 +1,4 @@
-from .models import Article, Editor, Player, Reporter
+from .models import Article, Editor, EmbeddedArticle, Player, Reporter
 
 
 def setup_fixtures():
@@ -20,6 +20,15 @@ def setup_fixtures():
     reporter = Reporter(first_name='Allen', last_name='Iverson',
                         email='[email protected]',  awards=['2010-mvp'])
     reporter.articles = [article1, article2]
+    embedded_article1 = EmbeddedArticle(
+        headline='Real',
+        editor=editor1
+    )
+    embedded_article2 = EmbeddedArticle(
+        headline='World',
+        editor=editor2
+    )
+    reporter.embedded_articles = [embedded_article1, embedded_article2]
     reporter.save()
 
     Player.drop_collection()

graphene_mongo/tests/test_query.py

@@ -76,6 +76,9 @@ def resolve_reporter(self, *args, **kwargs):
                 articles {
                     headline
                 },
+                embeddedArticles {
+                    headline
+                },
                 awards
             }
         }
@@ -89,6 +92,14 @@ def resolve_reporter(self, *args, **kwargs):
                 {'headline': 'Hello'},
                 {'headline': 'World'}
             ],
+            'embeddedArticles': [
+                {
+                    'headline': 'Real'
+                },
+                {
+                    'headline': 'World'
+                }
+            ],
             'awards': ['2010-mvp']
         }
     }

graphene_mongo/tests/test_relay_query.py

@@ -32,22 +32,30 @@ def resolve_reporter(self, *args, **kwargs):
         query ReporterQuery {
             reporter {
                 firstName,
+                lastName,
+                email,
                 articles {
                     edges {
                         node {
                             headline
                         }
                     }
+                },
+                embeddedArticles {
+                    edges {
+                        node {
+                            headline
+                        }
+                    }
                 }
-                lastName,
-                email
             }
         }
     '''
     expected = {
         'reporter': {
             'firstName': 'Allen',
             'lastName': 'Iverson',
+            'email': '[email protected]',
             'articles': {
                 'edges': [
                     {
@@ -62,7 +70,20 @@ def resolve_reporter(self, *args, **kwargs):
                     }
                 ],
             },
-            'email': '[email protected]'
+            'embeddedArticles': {
+                'edges': [
+                    {
+                        'node': {
+                            'headline': 'Real'
+                        }
+                    },
+                    {
+                        'node': {
+                            'headline': 'World'
+                        }
+                    }
+                ],
+            },
         }
     }
 

graphene_mongo/tests/types.py

@@ -1,25 +1,35 @@
 from graphene.relay import Node
 
 from ..types import MongoengineObjectType
-from .models import Article, Editor, Player, Reporter
+from .models import Article, Editor, EmbeddedArticle, Player, Reporter
 
 
 class EditorType(MongoengineObjectType):
+
     class Meta:
         model = Editor
 
 
 class ArticleType(MongoengineObjectType):
+
     class Meta:
         model = Article
 
 
+class EmbeddedArticleType(MongoengineObjectType):
+
+    class Meta:
+        model = EmbeddedArticle
+
+
 class PlayerType(MongoengineObjectType):
+
     class Meta:
         model = Player
 
 
 class ReporterType(MongoengineObjectType):
+
     class Meta:
         model = Reporter
 
@@ -38,6 +48,13 @@ class Meta:
         interfaces = (Node,)
 
 
+class EmbeddedArticleNode(MongoengineObjectType):
+
+    class Meta:
+        model = EmbeddedArticle
+        interfaces = (Node,)
+
+
 class PlayerNode(MongoengineObjectType):
 
     class Meta: