added some tests

Author: Victor

graphene_mongo/fields.py

@@ -111,11 +111,11 @@ def get_query(cls, model, info, **args):
 
             # https://github.com/graphql-python/graphene-mongo/issues/21
             if after is not None:
-                _after = from_global_id(after)[-1]
+                _after = int(from_global_id(after)[-1])
                 objs = objs[_after:]
 
             if before is not None:
-                _before = from_global_id(before)[-1]
+                _before = int(from_global_id(before)[-1])
                 objs = objs[:_before]
 
             if first is not None:

graphene_mongo/tests/test_relay_query.py

@@ -364,6 +364,89 @@ class Query(graphene.ObjectType):
     assert all(item in get_nodes(result.data, 'editors') for item in get_nodes(expected, 'editors'))
 
 
+def test_should_after():
+    class Query(graphene.ObjectType):
+
+        players = MongoengineConnectionField(PlayerNode)
+
+    query = '''
+        query EditorQuery {
+            players(after: "YXJyYXljb25uZWN0aW9uOjA=") {
+                edges {
+                    cursor,
+                    node {
+                        firstName
+                    }
+                }
+            }
+        }
+    '''
+    expected = {
+        'players': {
+            'edges': [
+                {
+                    'cursor': 'YXJyYXljb25uZWN0aW9uOjE=',
+                    'node': {
+                        'firstName': 'Magic',
+                    }
+                },
+                {
+                    'cursor': 'YXJyYXljb25uZWN0aW9uOjI=',
+                    'node': {
+                        'firstName': 'Larry'
+                    }
+                }
+            ]
+        }
+    }
+    schema = graphene.Schema(query=Query)
+    result = schema.execute(query)
+
+    assert not result.errors
+    assert json.dumps(result.data, sort_keys=True) == json.dumps(expected, sort_keys=True)
+
+
+def test_should_before():
+    class Query(graphene.ObjectType):
+
+        players = MongoengineConnectionField(PlayerNode)
+
+    query = '''
+        query EditorQuery {
+            players(before: "YXJyYXljb25uZWN0aW9uOjI=") {
+                edges {
+                    cursor,
+                    node {
+                        firstName
+                    }
+                }
+            }
+        }
+    '''
+    expected = {
+        'players': {
+            'edges': [
+                {
+                    'cursor': "YXJyYXljb25uZWN0aW9uOjA=",
+                    'node': {
+                        'firstName': 'Michael',
+                    }
+                },
+                {
+                    'cursor': 'YXJyYXljb25uZWN0aW9uOjE=',
+                    'node': {
+                        'firstName': 'Magic',
+                    }
+                }
+            ]
+        }
+    }
+    schema = graphene.Schema(query=Query)
+    result = schema.execute(query)
+
+    assert not result.errors
+    assert json.dumps(result.data, sort_keys=True) == json.dumps(expected, sort_keys=True)
+
 def test_should_self_reference():
 
     class Query(graphene.ObjectType):
@@ -463,6 +546,7 @@ class Query(graphene.ObjectType):
     assert not result.errors
     assert json.dumps(result.data, sort_keys=True) == json.dumps(expected, sort_keys=True)
 
+
 # TODO:
 def test_should_paging():
     pass