Merge branch 'feat-export-connection-field'

Author: abawchen

README.md

@@ -35,8 +35,8 @@ from graphene_mongo import MongoengineObjectType
 from .models import User as UserModel
 
 class User(MongoengineObjectType):
-	class Meta:
-		model = UserModel
+    class Meta:
+        model = UserModel
 
 class Query(graphene.ObjectType):
     users = graphene.List(User)

graphene_mongo/__init__.py

@@ -1,6 +1,6 @@
-#from .fields import (
-#    MongoEngineDocumentField
-#)
+from .fields import (
+    MongoengineConnectionField
+)
 
 from .types import (
     MongoengineObjectType,

graphene_mongo/fields.py

@@ -67,7 +67,7 @@ def merge_querysets(cls, default_queryset, queryset):
     def connection_resolver(cls, resolver, connection, model, root, info, **args):
         iterable = resolver(root, info, **args)
         if not iterable:
-            interable = cls.get_query(model, info, **args)
+            iterable = cls.get_query(model, info, **args)
         _len = len(iterable)
         connection = connection_from_list_slice(
             iterable,

graphene_mongo/tests/test_converter.py

@@ -10,7 +10,6 @@
 from .models import Article, Editor, EmbeddedArticle, Reporter
 
 from ..converter import convert_mongoengine_field
-from ..fields import MongoengineConnectionField
 from ..types import MongoengineObjectType
 
 

graphene_mongo/tests/test_query.py

@@ -180,9 +180,51 @@ def resolve_reporter(self, *args, **kwargs):
     assert not result.errors
     assert dict(result.data['reporter']) == expected['reporter']
 
-# TODO:
-def test_should_custom_identifier():
-    pass
+def test_should_connection_field():
+    class EditorNode(MongoengineObjectType):
+
+        class Meta:
+            model = Editor
+            interfaces = (Node,)
+
+    class Query(graphene.ObjectType):
+        node = Node.Field()
+        all_editors = MongoengineConnectionField(EditorNode)
+
+    query = '''
+        query EditorQuery {
+          allEditors {
+            edges {
+                node {
+                    firstName,
+                    lastName
+                }
+            }
+          }
+        }
+    '''
+    expected = {
+        'allEditors': {
+            'edges': [
+                {
+                    'node': {
+                        'firstName': 'Penny',
+                        'lastName': 'Hardaway'
+                    },
+                },
+                {
+                    'node': {
+                        'firstName': 'Grant',
+                        'lastName': 'Hill'
+                    }
+                }
+            ]
+        }
+    }
+    schema = graphene.Schema(query=Query)
+    result = schema.execute(query)
+    assert not result.errors
+    assert dict(result.data['allEditors']) == expected['allEditors']
 
 # TODO:
 def test_should_mutate_well():