Added test for with_context on connectionfield

This commit is contained in:
Markus Padourek 2016-05-20 14:03:47 +01:00
parent 7a5ad21b98
commit e897efb79f

View File

@ -36,16 +36,58 @@ class Query(graphene.ObjectType):
special_node = relay.NodeField(SpecialNode)
all_my_nodes = relay.ConnectionField(
MyNode, connection_type=MyConnection, customArg=graphene.String())
context_nodes = relay.ConnectionField(
MyNode, connection_type=MyConnection, customArg=graphene.String())
def resolve_all_my_nodes(self, args, info):
custom_arg = args.get('customArg')
assert custom_arg == "1"
return [MyNode(name='my')]
@with_context
def resolve_context_nodes(self, args, context, info):
custom_arg = args.get('customArg')
assert custom_arg == "1"
return [MyNode(name='my')]
schema.query = Query
def test_nodefield_query():
query = '''
query RebelsShipsQuery {
contextNodes (customArg:"1") {
edges {
node {
name
}
},
myCustomField
pageInfo {
hasNextPage
}
}
}
'''
expected = {
'allMyNodes': {
'edges': [{
'node': {
'name': 'my'
}
}],
'myCustomField': 'Custom',
'pageInfo': {
'hasNextPage': False,
}
}
}
result = schema.execute(query)
assert not result.errors
assert result.data == expected
def test_connectionfield_context_query():
query = '''
query RebelsShipsQuery {
myNode(id:"TXlOb2RlOjE=") {