Fixed keys_only query results

Author: ekampf

graphene_gae/ndb/fields.py

@@ -39,6 +39,10 @@ def connection_from_ndb_query(query, args={}, connection_type=None,
         except StopIteration:
             break
 
+        if keys_only:
+            # entity is actualy an ndb.Key and we need to create an empty entity to hold it
+            entity = edge_type.node_type._meta.model(key=entity)
+
         edge = edge_type(node=entity, cursor=iter.cursor_after().urlsafe())
         edges.append(edge)
 

tests/_ndb/test_types_relay.py

@@ -176,6 +176,34 @@ def test_connectionField(self):
             self.assertLength(comments, 1)
             self.assertEqual(comments[0]['node']['body'], "c" + article['summary'])
 
+    def test_connectionField_keysOnly(self):
+        a1 = Article(headline="Test1", summary="1").put()
+        a2 = Article(headline="Test2", summary="2").put()
+        a3 = Article(headline="Test3", summary="3").put()
+
+        Comment(parent=a1, body="c1").put()
+        Comment(parent=a2, body="c2").put()
+        Comment(parent=a3, body="c3").put()
+
+        result = schema.execute("""
+            query Articles {
+                articles(keysOnly: true) {
+                    edges {
+                        cursor,
+                        node {
+                            id
+                        }
+                    }
+
+                }
+            }
+            """)
+
+        self.assertEmpty(result.errors)
+
+        articles = result.data.get('articles', {}).get('edges')
+        self.assertLength(articles, 3)
+
     def test_connectionField_empty(self):
         Article(headline="Test1", summary="1").put()