test: Add test_relay_query.test_should_get_node_by_id

Add Reporter.id as the primary key and assign it in fixtures.

Author: abawchen

graphene_mongo/tests/fixtures.py

@@ -17,7 +17,7 @@ def setup_fixtures():
     article2.save()
 
     Reporter.drop_collection()
-    reporter = Reporter(first_name='Allen', last_name='Iverson',
+    reporter = Reporter(id='1', first_name='Allen', last_name='Iverson',
                         email='[email protected]',  awards=['2010-mvp'])
     reporter.articles = [article1, article2]
     embedded_article1 = EmbeddedArticle(

graphene_mongo/tests/models.py

@@ -45,6 +45,7 @@ class EmbeddedArticle(EmbeddedDocument):
 class Reporter(Document):
     meta = {'collection': 'test_repoter'}
 
+    id = StringField(primary_key=True)
     first_name = StringField(required=True)
     last_name = StringField(required=True)
     email = EmailField()

graphene_mongo/tests/test_relay_query.py

@@ -189,7 +189,7 @@ class Mutation(graphene.ObjectType):
     schema = graphene.Schema(query=Query, mutation=Mutation)
     result = schema.execute(query)
     assert not result.errors
-    # assert result.data == expected
+    assert result.data == expected
 
 def test_should_filter():
 
@@ -231,6 +231,33 @@ class Query(graphene.ObjectType):
     assert result.data == expected
 
 
+def test_should_get_node_by_id():
+    class Query(graphene.ObjectType):
+        node = Node.Field()
+        reporters = MongoengineConnectionField(ReporterNode)
+
+    query = '''
+        query ReportersQuery {
+            reporter: node(id: "UmVwb3J0ZXJOb2RlOjE=") {
+                id,
+                ... on ReporterNode {
+                    firstName
+                }
+            }
+        }
+    '''
+    expected = {
+        'reporter': {
+            'id': 'UmVwb3J0ZXJOb2RlOjE=',
+            'firstName': 'Allen'
+        }
+    }
+    schema = graphene.Schema(query=Query)
+    result = schema.execute(query)
+    assert not result.errors
+    assert result.data == expected
+
+
 def test_should_first_n():
 
     class Query(graphene.ObjectType):