test: Use module scope to separate fixtures

Author: abawchen

graphene_mongo/tests/fixtures.py

@@ -1,10 +1,12 @@
+import pytest
 from .models import (
     Article, Editor, EmbeddedArticle, Player,
     Reporter, Child, ProfessorMetadata, ProfessorVector,
 )
 
-
+@pytest.fixture(scope='module')
 def setup_fixtures():
+    print('setup_fixtures')
     Editor.drop_collection()
     editor1 = Editor(
         id='1',

graphene_mongo/tests/test_mutation.py

@@ -7,10 +7,7 @@
 from .types import (ArticleNode, EditorNode)
 
 
-setup_fixtures()
-
-
-def test_should_create():
+def test_should_create(setup_fixtures):
 
     class CreateArticle(graphene.Mutation):
 
@@ -60,7 +57,7 @@ class Mutation(graphene.ObjectType):
     assert result.data == expected
 
 
-def test_should_update():
+def test_should_update(setup_fixtures):
 
     class UpdateEditor(graphene.Mutation):
 
@@ -71,7 +68,6 @@ class Arguments:
         editor = graphene.Field(EditorNode)
 
         def mutate(self, info, id, first_name):
-            print(id, first_name)
             editor = Editor.objects.get(id=id)
             editor.first_name = first_name
             editor.save()

graphene_mongo/tests/test_query.py

@@ -10,10 +10,8 @@
     EditorType, PlayerType, ReporterType, ProfessorVectorType
 )
 
-setup_fixtures()
 
-
-def test_should_query_editor():
+def test_should_query_editor(setup_fixtures):
 
     class Query(graphene.ObjectType):
 
@@ -65,7 +63,7 @@ def resolve_editors(self, *args, **kwargs):
     assert all(item in result.data['editors'] for item in expected['editors'])
 
 
-def test_should_query_reporter():
+def test_should_query_reporter(setup_fixtures):
 
     class Query(graphene.ObjectType):
         reporter = graphene.Field(ReporterType)
@@ -127,7 +125,7 @@ def resolve_reporter(self, *args, **kwargs):
     assert dict(result.data['reporter']) == expected['reporter']
 
 
-def test_should_custom_kwargs():
+def test_should_custom_kwargs(setup_fixtures):
 
     class Query(graphene.ObjectType):
 
@@ -165,7 +163,7 @@ def resolve_editors(self, *args, **kwargs):
     assert all(item in result.data['editors'] for item in expected['editors'])
 
 
-def test_should_self_reference():
+def test_should_self_reference(setup_fixtures):
 
     class Query(graphene.ObjectType):
 
@@ -229,7 +227,7 @@ def resolve_all_players(self, *args, **kwargs):
     assert json.dumps(result.data, sort_keys=True) == json.dumps(expected, sort_keys=True)
 
 
-def test_should_query_with_embedded_document():
+def test_should_query_with_embedded_document(setup_fixtures):
 
     class Query(graphene.ObjectType):
         professor_vector = graphene.Field(ProfessorVectorType, id=graphene.String())

graphene_mongo/tests/test_relay_query.py

@@ -13,14 +13,12 @@
                     ChildNode,)
 from ..fields import MongoengineConnectionField
 
-setup_fixtures()
-
 
 def get_nodes(data, key):
     return map(lambda edge: edge['node'], data[key]['edges'])
 
 
-def test_should_query_reporter():
+def test_should_query_reporter(setup_fixtures):
 
     class Query(graphene.ObjectType):
         node = Node.Field()
@@ -115,7 +113,7 @@ def resolve_reporter(self, *args, **kwargs):
     assert dict(result.data['reporter']) == expected['reporter']
 
 
-def test_should_query_all_editors():
+def test_should_query_all_editors(setup_fixtures):
 
     class Query(graphene.ObjectType):
         node = Node.Field()
@@ -168,7 +166,7 @@ class Query(graphene.ObjectType):
     assert dict(result.data['allEditors']) == expected['allEditors']
 
 
-def test_should_filter_editors_by_id():
+def test_should_filter_editors_by_id(setup_fixtures):
 
     class Query(graphene.ObjectType):
         node = Node.Field()
@@ -207,57 +205,7 @@ class Query(graphene.ObjectType):
     assert dict(result.data['allEditors']) == expected['allEditors']
 
 
-def test_should_mutate():
-
-    class CreateArticle(graphene.Mutation):
-
-        class Arguments:
-
-            headline = graphene.String()
-
-        article = graphene.Field(ArticleNode)
-
-        def mutate(self, info, headline):
-            article = Article(
-                headline=headline
-            )
-            article.save()
-
-            return CreateArticle(article=article)
-
-    class Query(graphene.ObjectType):
-
-        node = Node.Field()
-
-    class Mutation(graphene.ObjectType):
-
-        create_article = CreateArticle.Field()
-
-    query = '''
-        mutation ArticleCreator {
-            createArticle(
-                headline: "My Article"
-            ) {
-                article {
-                    headline
-                }
-            }
-        }
-    '''
-    expected = {
-        'createArticle': {
-            'article': {
-                'headline': 'My Article'
-            }
-        }
-    }
-    schema = graphene.Schema(query=Query, mutation=Mutation)
-    result = schema.execute(query)
-    assert not result.errors
-    assert result.data == expected
-
-
-def test_should_filter():
+def test_should_filter(setup_fixtures):
 
     class Query(graphene.ObjectType):
         node = Node.Field()
@@ -297,7 +245,7 @@ class Query(graphene.ObjectType):
     assert result.data == expected
 
 
-def test_should_filter_by_reference_field():
+def test_should_filter_by_reference_field(setup_fixtures):
 
     class Query(graphene.ObjectType):
         node = Node.Field()
@@ -337,7 +285,7 @@ class Query(graphene.ObjectType):
     assert result.data == expected
 
 
-def test_should_filter_through_inheritance():
+def test_should_filter_through_inheritance(setup_fixtures):
 
     class Query(graphene.ObjectType):
         node = Node.Field()
@@ -374,7 +322,7 @@ class Query(graphene.ObjectType):
         expected, sort_keys=True)
 
 
-def test_should_get_node_by_id():
+def test_should_get_node_by_id(setup_fixtures):
     # Notes: https://goo.gl/hMNRgs
     class Query(graphene.ObjectType):
         reporter = Node.Field(ReporterNode)
@@ -400,7 +348,7 @@ class Query(graphene.ObjectType):
     assert result.data == expected
 
 
-def test_should_first_n():
+def test_should_first_n(setup_fixtures):
 
     class Query(graphene.ObjectType):
 
@@ -456,7 +404,7 @@ class Query(graphene.ObjectType):
                for item in get_nodes(expected, 'editors'))
 
 
-def test_should_after():
+def test_should_after(setup_fixtures):
     class Query(graphene.ObjectType):
 
         players = MongoengineConnectionField(PlayerNode)
@@ -499,7 +447,7 @@ class Query(graphene.ObjectType):
         expected, sort_keys=True)
 
 
-def test_should_before():
+def test_should_before(setup_fixtures):
     class Query(graphene.ObjectType):
 
         players = MongoengineConnectionField(PlayerNode)
@@ -542,7 +490,7 @@ class Query(graphene.ObjectType):
         expected, sort_keys=True)
 
 
-def test_should_last_n():
+def test_should_last_n(setup_fixtures):
     class Query(graphene.ObjectType):
         players = MongoengineConnectionField(PlayerNode)
 
@@ -584,7 +532,7 @@ class Query(graphene.ObjectType):
         expected, sort_keys=True)
 
 
-def test_should_self_reference():
+def test_should_self_reference(setup_fixtures):
 
     class Query(graphene.ObjectType):
 
@@ -683,8 +631,3 @@ 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