graphql-python/graphene
1
1
Fork 0
mirror of https://github.com/graphql-python/graphene.git synced 2025-02-23 15:10:48 +03:00
Code Issues Packages Projects Releases Wiki Activity

Add totalCount field in ConnectionBase class

Browse Source
This commit is contained in:
Amnisia 2017-07-22 02:09:16 +05:00
parent 557ec44a13
commit 533d514cc9
2 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
graphene/relay/connection.py
View File

@ -82,8 +82,13 @@ class ConnectionMeta(ObjectTypeMeta):
class ConnectionBase(AbstractType):
page_info = Field(PageInfo, name='pageInfo', required=True)
total_count = Int(name='totalCount', required=True)
edges = NonNull(List(edge))
@staticmethod
def resolve_total_count(root, args, context, info):
return root.length
bases = (ConnectionBase, ) + bases
attrs = dict(attrs, _meta=options, Edge=edge)
return ObjectTypeMeta.__new__(cls, name, bases, attrs)

4
graphene/relay/tests/test_connection.py
View File

@ -23,7 +23,7 @@ def test_connection():
assert MyObjectConnection._meta.name == 'MyObjectConnection'
fields = MyObjectConnection._meta.fields
assert list(fields.keys()) == ['page_info', 'edges', 'extra']
assert list(fields.keys()) == ['page_info', 'total_count', 'edges', 'extra']
edge_field = fields['edges']
pageinfo_field = fields['page_info']
@ -48,7 +48,7 @@ def test_connection_inherit_abstracttype():
assert MyObjectConnection._meta.name == 'MyObjectConnection'
fields = MyObjectConnection._meta.fields
assert list(fields.keys()) == ['page_info', 'edges', 'extra']
assert list(fields.keys()) == ['page_info', 'total_count', 'edges', 'extra']
def test_edge():