wip: Add test_should_query_reporters_with_nested_document

Author: abawchen

graphene_mongo/fields.py

@@ -135,6 +135,7 @@ def get_queryset(self, model, info, **args):
                     reference_obj = get_node_from_global_id(reference_fields[arg_name], info, args.pop(arg_name))
                     hydrated_references[arg_name] = reference_obj
             args.update(hydrated_references)
+
         if self._get_queryset:
             queryset_or_filters = self._get_queryset(model, info, **args)
             if isinstance(queryset_or_filters, mongoengine.QuerySet):

graphene_mongo/tests/setup.py

@@ -57,6 +57,9 @@ def fixtures():
     article2 = Article(headline='World', editor=editor2, pub_date=pub_date)
     article2.save()
 
+    article3 = Article(headline='Bye', editor=editor2, pub_date=pub_date)
+    article3.save()
+
     Reporter.drop_collection()
     reporter1 = Reporter(
         id='1',

graphene_mongo/tests/test_relay_query.py

@@ -126,7 +126,60 @@ def resolve_reporter(self, *args, **kwargs):
     schema = graphene.Schema(query=Query)
     result = schema.execute(query)
     assert not result.errors
-    assert dict(result.data['reporter']) == expected['reporter']
+    assert json.dumps(result.data['reporter'], sort_keys=True) \
+        == json.dumps(expected['reporter'], sort_keys=True)
+    # assert dict(result.data['reporter']) == expected['reporter']
+
+
+def test_should_query_reporters_with_nested_document(fixtures):
+
+    class Query(graphene.ObjectType):
+        node = Node.Field()
+        reporters = MongoengineConnectionField(ReporterNode)
+
+    query = '''
+        query ReporterQuery {
+            reporters(firstName: "Allen") {
+                edges {
+                    node {
+                        firstName,
+                        lastName,
+                        email,
+                        articles(headline: "Hello") {
+                             edges {
+                                  node {
+                                       headline
+                                  }
+                             }
+                        }
+                    }
+                }
+            }
+        }
+    '''
+    expected = {
+        'reporter': {
+            'firstName': 'Allen',
+            'lastName': 'Iverson',
+            'email': '[email protected]',
+            'articles': {
+                'edges': [
+                    {
+                        'node': {
+                            'headline': 'Hello'
+                        }
+                    }
+                ]
+            }
+        }
+    }
+
+    schema = graphene.Schema(query=Query)
+    result = schema.execute(query)
+    # assert not result.errors
+    # print(dict(result.data['reporters']))
+    # assert json.dumps(result.data['reporter'], sort_keys=True) \
+    #     == json.dumps(expected['reporter'], sort_keys=True)
 
 
 def test_should_query_all_editors(fixtures, fixtures_dirname):

graphene_mongo/tests/types.py

@@ -1,4 +1,13 @@
+<<<<<<< HEAD
 from . import models
+||||||| merged common ancestors
+from graphene.relay import Node
+
+=======
+from graphene.relay import Node
+
+from ..fields import MongoengineConnectionField
+>>>>>>> wip: Add test_should_query_reporters_with_nested_document
 from ..types import MongoengineObjectType
 
 
@@ -57,7 +66,6 @@ class Meta:
 
 
 class ProfessorMetadataType(MongoengineObjectType):
-
     class Meta:
         model = models.ProfessorMetadata