From e897efb79f7d5946c668ddd74ecba17289958668 Mon Sep 17 00:00:00 2001
From: Markus Padourek <markus.padourek@gmail.com>
Date: Fri, 20 May 2016 14:03:47 +0100
Subject: [PATCH] Added test for with_context on connectionfield

---
 graphene/relay/tests/test_query.py | 42 ++++++++++++++++++++++++++++++
 1 file changed, 42 insertions(+)

diff --git a/graphene/relay/tests/test_query.py b/graphene/relay/tests/test_query.py
index 4e94d9fd..70b31936 100644
--- a/graphene/relay/tests/test_query.py
+++ b/graphene/relay/tests/test_query.py
@@ -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=") {