test: Add with_local_registry to separate registry in test_query.

Author: abawchen

graphene_mongo/tests/test_query.py

@@ -5,6 +5,7 @@
 from graphene.relay import Node
 
 from .models import Article, Editor, EmbeddedArticle, Reporter
+from .utils import with_local_registry
 from ..fields import MongoengineConnectionField
 from ..types import MongoengineObjectType
 
@@ -27,6 +28,7 @@ def setup_fixtures():
 setup_fixtures()
 
 
+@with_local_registry
 def test_should_query_editor_well():
     class EditorType(MongoengineObjectType):
         class Meta:
@@ -73,6 +75,7 @@ def resolve_editors(self, *args, **kwargs):
     assert all(item in result.data['editors'] for item in expected['editors'])
 
 
+@with_local_registry
 def test_should_query_reporter_well():
     class ArticleType(MongoengineObjectType):
         class Meta:
@@ -119,13 +122,17 @@ def resolve_reporter(self, *args, **kwargs):
     assert not result.errors
     assert dict(result.data['reporter']) == expected['reporter']
 
+
+@with_local_registry
 def test_should_node():
     class ArticleNode(MongoengineObjectType):
+
         class Meta:
             model = Article
             interfaces = (Node,)
 
     class ReporterNode(MongoengineObjectType):
+
         class Meta:
             model = Reporter
             interfaces = (Node,)
@@ -180,6 +187,8 @@ def resolve_reporter(self, *args, **kwargs):
     assert not result.errors
     assert dict(result.data['reporter']) == expected['reporter']
 
+
+@with_local_registry
 def test_should_connection_field():
     class EditorNode(MongoengineObjectType):
 
@@ -227,6 +236,7 @@ class Query(graphene.ObjectType):
     assert dict(result.data['allEditors']) == expected['allEditors']
 
 
+@with_local_registry
 def test_should_mutate_well():
     class ArticleNode(MongoengineObjectType):
 

graphene_mongo/tests/test_types.py

@@ -7,6 +7,7 @@
 from ..types import MongoengineObjectType
 from .models import Article
 from .models import Reporter
+from .utils import with_local_registry
 
 registry.reset_global_registry()
 
@@ -80,21 +81,6 @@ class Meta:
     assert 'valid Mongoengine Model' in str(excinfo.value)
 
 
-def with_local_registry(func):
-    def inner(*args, **kwargs):
-        old = registry.get_global_registry()
-        registry.reset_global_registry()
-        try:
-            retval = func(*args, **kwargs)
-        except Exception as e:
-            registry.registry = old
-            raise e
-        else:
-            registry.registry = old
-            return retval
-    return inner
-
-
 @with_local_registry
 def test_mongoengine_objecttype_only_fields():
     class A(MongoengineObjectType):

graphene_mongo/tests/utils.py

@@ -0,0 +1,17 @@
+from .. import registry
+
+
+def with_local_registry(func):
+    def inner(*args, **kwargs):
+        old = registry.get_global_registry()
+        registry.reset_global_registry()
+        try:
+            retval = func(*args, **kwargs)
+        except Exception as e:
+            registry.registry = old
+            raise e
+        else:
+            registry.registry = old
+            return retval
+    return inner
+