minor: Simplify test code

Author: abawchen

graphene_mongo/tests/test_relay_query.py

@@ -22,7 +22,6 @@ class Query(graphene.ObjectType):
         reporter = graphene.Field(nodes.ReporterNode)
 
         def resolve_reporter(self, *args, **kwargs):
-            print('abc' * 10)
             return models.Reporter.objects.first()
 
     query = '''
@@ -120,7 +119,7 @@ def resolve_reporter(self, *args, **kwargs):
     schema = graphene.Schema(query=Query)
     result = schema.execute(query)
     assert not result.errors
-    assert result.data['reporter'] == expected['reporter']
+    assert result.data == expected
 
 
 def test_should_query_reporters_with_nested_document(fixtures):
@@ -174,7 +173,7 @@ class Query(graphene.ObjectType):
     schema = graphene.Schema(query=Query)
     result = schema.execute(query)
     assert not result.errors
-    assert result.data['reporters'] == expected['reporters']
+    assert result.data == expected
 
 
 def test_should_query_all_editors(fixtures, fixtures_dirname):
@@ -251,10 +250,10 @@ class Query(graphene.ObjectType):
     schema = graphene.Schema(query=Query)
     result = schema.execute(query)
     assert not result.errors
-    assert result.data['editors'] == expected['editors']
-
+    assert result.data == expected
 
 
+"""
 def test_should_query_editors_with_dataloader(fixtures):
     from promise import Promise
     from promise.dataloader import DataLoader
@@ -274,13 +273,13 @@ def batch_load_fn(self, keys):
     editor_loader = EditorLoader()
 
     class Query(graphene.ObjectType):
-        editors = MongoengineConnectionField(nodes.EditorNode)
+        # editors = MongoengineConnectionField(nodes.EditorNode)
+        editors = graphene.List(types.EditorType)
 
-        """
         def resolve_editors(self, info, **args):
             print(self.__dict__)
             return None
-        """
+
 
     query = '''
         query EditorPromiseQuery {
@@ -299,17 +298,17 @@ def resolve_editors(self, info, **args):
     schema = graphene.Schema(query=Query)
     result = schema.execute(query)
     print(result.data)
+"""
 
 
 def test_should_filter_editors_by_id(fixtures):
 
     class Query(graphene.ObjectType):
-        node = Node.Field()
-        all_editors = MongoengineConnectionField(nodes.EditorNode)
+        editors = MongoengineConnectionField(nodes.EditorNode)
 
     query = '''
         query EditorQuery {
-          allEditors(id: "RWRpdG9yTm9kZToy") {
+          editors(id: "RWRpdG9yTm9kZToy") {
             edges {
                 node {
                     id,
@@ -321,7 +320,7 @@ class Query(graphene.ObjectType):
         }
     '''
     expected = {
-        'allEditors': {
+        'editors': {
             'edges': [
                 {
                     'node': {
@@ -337,13 +336,12 @@ class Query(graphene.ObjectType):
     schema = graphene.Schema(query=Query)
     result = schema.execute(query)
     assert not result.errors
-    assert dict(result.data['allEditors']) == expected['allEditors']
+    assert result.data == expected
 
 
 def test_should_filter(fixtures):
 
     class Query(graphene.ObjectType):
-        node = Node.Field()
         articles = MongoengineConnectionField(nodes.ArticleNode)
 
     query = '''
@@ -385,7 +383,6 @@ class Query(graphene.ObjectType):
 def test_should_filter_by_reference_field(fixtures):
 
     class Query(graphene.ObjectType):
-        node = Node.Field()
         articles = MongoengineConnectionField(nodes.ArticleNode)
 
     query = '''