Fixed batch_size in connection_from_ndb_query

ekampf · ekampf · commit a3180b1df93b · 2016-06-01T14:01:52.000-07:00
diff --git a/graphene_gae/ndb/fields.py b/graphene_gae/ndb/fields.py
@@ -31,7 +31,7 @@ def connection_from_ndb_query(query, args={}, connection_type=None,
 
     iter = query.iter(produce_cursors=True, start_cursor=start_cursor, batch_size=batch_size, keys_only=keys_only)
 
-    page_size = first if first else 10
+    page_size = first if first else batch_size
     edges = []
     while len(edges) < page_size:
         try:
