test: Add test_query.test_should_self_reference

Author: abawchen

graphene_mongo/tests/models.py

@@ -46,7 +46,7 @@ class Reporter(Document):
     meta = {'collection': 'test_repoter'}
 
     first_name = StringField(required=True)
-    last_name = StringField(requred=True)
+    last_name = StringField(required=True)
     email = EmailField()
     articles = ListField(ReferenceField(Article))
     # FIXME
@@ -55,3 +55,9 @@ class Reporter(Document):
     # custom_map = MapField(field=StringField())
     awards = ListField(StringField())
 
+
+class Player(Document):
+    first_name = StringField(required=True)
+    last_name = StringField(required=True)
+    opponent = ReferenceField('Player')
+

graphene_mongo/tests/test_query.py

@@ -26,15 +26,22 @@ def setup_fixtures():
     article1.save()
     article2 = Article(headline='World', editor=editor2)
     article2.save()
+
     reporter.articles = [article1, article2]
     reporter.save()
 
-    player = Player(first_name='Michael', last_name='Jordan')
-    player.save()
+    player1 = Player(first_name='Michael', last_name='Jordan')
+    player1.save()
+    player2 = Player(first_name='Magic', last_name='Johnson', opponent=player1)
+    player2.save()
 
 setup_fixtures()
 
 
+def get_nodes(data, key):
+    return map(lambda edge: edge['node'], data[key]['edges'])
+
+
 def test_should_query_editor_well():
 
     class Query(graphene.ObjectType):
@@ -351,11 +358,8 @@ class Query(graphene.ObjectType):
     schema = graphene.Schema(query=Query)
     result = schema.execute(query)
 
-    def get_nodes(data):
-        return map(lambda edge: edge['node'], data['editors']['edges'])
-
     assert not result.errors
-    assert all(item in get_nodes(result.data) for item in get_nodes(expected))
+    assert all(item in get_nodes(result.data, 'editors') for item in get_nodes(expected, 'editors'))
 
 def test_should_custom_kwargs():
 
@@ -408,7 +412,9 @@ class Query(graphene.ObjectType):
                     cursor,
                     node {
                         firstName,
-
+                        opponent {
+                            firstName
+                        }
                     }
                 }
             }
@@ -420,29 +426,27 @@ class Query(graphene.ObjectType):
                 {
                     'cursor': 'xxx',
                     'node': {
-                        'firstName': 'Penny'
+                        'firstName': 'Michael',
+                        'opponent': None
                     }
                 },
                 {
                     'cursor': 'xxx',
                     'node': {
-                        'firstName': 'Grant'
+                        'firstName': 'Magic',
+                        'opponent': {
+                            'firstName': 'Michael'
+                        }
                     }
                 }
             ]
         }
     }
     schema = graphene.Schema(query=Query)
     result = schema.execute(query)
-    print(result.data)
 
     assert not result.errors
-    '''
-    def get_nodes(data):
-        return map(lambda edge: edge['node'], data['editors']['edges'])
-
-    assert all(item in get_nodes(result.data) for item in get_nodes(expected))
-    '''
+    assert all(item in get_nodes(result.data, 'players') for item in get_nodes(expected, 'players'))
 
 
 # TODO:

graphene_mongo/tests/types.py

@@ -1,7 +1,7 @@
 from graphene.relay import Node
 
-from .models import Article, Editor, Reporter
 from ..types import MongoengineObjectType
+from .models import Article, Editor, Player, Reporter
 
 
 class EditorType(MongoengineObjectType):
@@ -13,6 +13,12 @@ class ArticleType(MongoengineObjectType):
     class Meta:
         model = Article
 
+
+class PlayerType(MongoengineObjectType):
+    class Meta:
+        model = Player
+
+
 class ReporterType(MongoengineObjectType):
     class Meta:
         model = Reporter
@@ -24,13 +30,20 @@ class Meta:
         model = Article
         interfaces = (Node,)
 
+
 class EditorNode(MongoengineObjectType):
 
     class Meta:
         model = Editor
         interfaces = (Node,)
 
 
+class PlayerNode(MongoengineObjectType):
+
+    class Meta:
+        model = Player
+        interfaces = (Node,)
+
 class ReporterNode(MongoengineObjectType):
 
     class Meta: