Fix projections on connection query

ekampf · ekampf · commit e398c346796f · 2016-06-09T16:05:24.000-07:00
diff --git a/graphene_gae/ndb/fields.py b/graphene_gae/ndb/fields.py
@@ -30,7 +30,7 @@ def connection_from_ndb_query(query, args={}, connection_type=None,
     page_size = first if first else full_args.get('page_size', 20)
     start_cursor = ndb.Cursor(urlsafe=after) if after else None
 
-    iter = query.iter(produce_cursors=True, start_cursor=start_cursor, batch_size=batch_size, keys_only=keys_only)
+    iter = query.iter(produce_cursors=True, start_cursor=start_cursor, batch_size=batch_size, keys_only=keys_only, projection=query.projection)
 
     edges = []
     while len(edges) < page_size:
