Merge pull request #72 from riverfr0zen/master

Add list_length to connection returned by default_resolver in MongoengineConnectionField

Author: abawchen

graphene_mongo/fields.py

@@ -161,6 +161,7 @@ def default_resolver(self, _root, info, **args):
             pageinfo_type=PageInfo,
         )
         connection.iterable = objs
+        connection.list_length = list_length
         return connection
 
     def chained_resolver(self, resolver, root, info, **args):

graphene_mongo/tests/test_fields.py

@@ -1,5 +1,6 @@
 from ..fields import MongoengineConnectionField
 from .types import ArticleNode, PublisherNode, ErroneousModelNode
+from .setup import fixtures
 
 
 def test_field_args():
@@ -35,3 +36,11 @@ def test_default_resolver_with_colliding_objects_field():
 
     connection = field.default_resolver(None, {})
     assert 0 == len(connection.iterable)
+
+
+def test_default_resolver_connection_list_length(fixtures):
+    field = MongoengineConnectionField(ArticleNode)
+
+    connection = field.default_resolver(None, {}, **{'first': 1})
+    assert hasattr(connection, 'list_length')
+    assert connection.list_length == 2

setup.cfg

@@ -8,6 +8,7 @@ per-file-ignores =
     graphene_mongo/tests/test_mutation.py: F401, F811
     graphene_mongo/tests/test_query.py: F401, F811
     graphene_mongo/tests/test_relay_query.py: F401, F811
+    graphene_mongo/tests/test_fields.py: F401, F811
 [coverage:run]
 omit = */tests/*