fix: Make test_query.test_should_first_n pass.

Author: abawchen

graphene_mongo/tests/test_query.py

@@ -323,39 +323,28 @@ class Query(graphene.ObjectType):
     expected = {
         'editors': {
             'edges': [
-                [
-                    {
-                        'cursor': 'xxx'
-                    },
-                    {
-                        'node': {
-                            'firstName': 'Penny'
-                        }
+                {
+                    'cursor': 'xxx',
+                    'node': {
+                        'firstName': 'Penny'
                     }
-                ],
-                [
-                    {
-                        'cursor': 'xxx'
-                    },
-                    {
-                        'node': {
-                            'firtName': 'Grant'
-                        }
+                },
+                {
+                    'cursor': 'xxx',
+                    'node': {
+                        'firstName': 'Grant'
                     }
-                ]
+                }
             ]
         }
     }
     schema = graphene.Schema(query=Query)
     result = schema.execute(query)
-    edges = result.data['editors']['edges']
-    # nodes = map(lambda edge: edge[1], expected['editors']['edges'])
-    # print(edges)
-    def get_nodes(edges):
-        return map(lambda edge: edge[1], edges)
-
-    print(get_nodes(edges))
-    assert all(item in get_nodes(edges) for item in get_nodes(expected['editors']['edges']))
+
+    def get_nodes(data):
+        return map(lambda edge: edge['node'], data['editors']['edges'])
+
+    assert all(item in get_nodes(result.data) for item in get_nodes(expected))
 
 def test_should_custom_kwargs():