test: Separate test.nodes from test.types

Author: abawchen

graphene_mongo/tests/nodes.py

@@ -0,0 +1,98 @@
+import graphene
+from graphene.relay import Node
+
+from . import models, types
+from ..types import MongoengineObjectType
+
+
+class PublisherNode(MongoengineObjectType):
+    legal_name = graphene.String()
+    bad_field = graphene.String()
+
+    class Meta:
+        model = models.Publisher
+        only_fields = ('id', 'name')
+        interfaces = (Node, )
+
+
+class ArticleNode(MongoengineObjectType):
+
+    class Meta:
+        model = models.Article
+        interfaces = (Node, )
+
+
+class EditorNode(MongoengineObjectType):
+
+    class Meta:
+        model = models.Editor
+        interfaces = (Node, )
+
+
+class EmbeddedArticleNode(MongoengineObjectType):
+
+    class Meta:
+        model = models.EmbeddedArticle
+        interfaces = (Node, )
+
+
+class PlayerNode(MongoengineObjectType):
+
+    class Meta:
+        model = models.Player
+        interfaces = (Node, )
+
+
+class ReporterNode(MongoengineObjectType):
+
+    class Meta:
+        model = models.Reporter
+        interfaces = (Node, )
+
+
+class ParentNode(MongoengineObjectType):
+
+    class Meta:
+        model = models.Parent
+        interfaces = (Node, )
+
+
+class ChildNode(MongoengineObjectType):
+
+    class Meta:
+        model = models.Child
+        interfaces = (Node, )
+
+
+class ChildRegisteredBeforeNode(MongoengineObjectType):
+
+    class Meta:
+        model = models.ChildRegisteredBefore
+        interfaces = (Node, )
+
+
+class ParentWithRelationshipNode(MongoengineObjectType):
+
+    class Meta:
+        model = models.ParentWithRelationship
+        interfaces = (Node, )
+
+
+class ChildRegisteredAfterNode(MongoengineObjectType):
+
+    class Meta:
+        model = models.ChildRegisteredAfter
+        interfaces = (Node, )
+
+
+class ProfessorVectorNode(MongoengineObjectType):
+
+    class Meta:
+        model = models.ProfessorVector
+        interfaces = (Node, )
+
+
+class ErroneousModelNode(MongoengineObjectType):
+    class Meta:
+        model = models.ErroneousModel
+        interfaces = (Node, )

graphene_mongo/tests/test_fields.py

@@ -1,10 +1,10 @@
-from ..fields import MongoengineConnectionField
-from .types import ArticleNode, PublisherNode, ReporterNode, ErroneousModelNode
+from . import nodes
 from .setup import fixtures
+from ..fields import MongoengineConnectionField
 
 
 def test_article_field_args():
-    field = MongoengineConnectionField(ArticleNode)
+    field = MongoengineConnectionField(nodes.ArticleNode)
 
     field_args = ['id', 'headline', 'pub_date']
     assert set(field.field_args.keys()) == set(field_args)
@@ -18,35 +18,35 @@ def test_article_field_args():
 
 
 def test_reporter_field_args():
-    field = MongoengineConnectionField(ReporterNode)
+    field = MongoengineConnectionField(nodes.ReporterNode)
 
     field_args = ['id', 'first_name', 'last_name', 'email', 'awards']
     assert set(field.field_args.keys()) == set(field_args)
 
 
 def test_field_args_with_property():
-    field = MongoengineConnectionField(PublisherNode)
+    field = MongoengineConnectionField(nodes.PublisherNode)
 
     field_args = ['id', 'name']
     assert set(field.field_args.keys()) == set(field_args)
 
 
 def test_field_args_with_unconverted_field():
-    field = MongoengineConnectionField(PublisherNode)
+    field = MongoengineConnectionField(nodes.PublisherNode)
 
     field_args = ['id', 'name']
     assert set(field.field_args.keys()) == set(field_args)
 
 
 def test_default_resolver_with_colliding_objects_field():
-    field = MongoengineConnectionField(ErroneousModelNode)
+    field = MongoengineConnectionField(nodes.ErroneousModelNode)
 
     connection = field.default_resolver(None, {})
     assert 0 == len(connection.iterable)
 
 
 def test_default_resolver_connection_list_length(fixtures):
-    field = MongoengineConnectionField(ArticleNode)
+    field = MongoengineConnectionField(nodes.ArticleNode)
 
     connection = field.default_resolver(None, {}, **{'first': 1})
     assert hasattr(connection, 'list_length')

graphene_mongo/tests/test_mutation.py

@@ -4,7 +4,7 @@
 
 from .setup import fixtures
 from .models import (Article, Editor)
-from .types import (ArticleNode, EditorNode)
+from .nodes import (ArticleNode, EditorNode)
 
 
 def test_should_create(fixtures):

graphene_mongo/tests/test_query.py

@@ -127,6 +127,7 @@ def resolve_reporter(self, *args, **kwargs):
     assert not result.errors
     assert dict(result.data['reporter']) == expected['reporter']
 
+
 def test_should_custom_kwargs(fixtures):
 
     class Query(graphene.ObjectType):

graphene_mongo/tests/test_relay_query.py

@@ -7,7 +7,7 @@
 
 from .setup import fixtures
 from .models import Article, Reporter
-from .types import (ArticleNode,
+from .nodes import (ArticleNode,
                     EditorNode,
                     PlayerNode,
                     ReporterNode,

graphene_mongo/tests/types.py

@@ -1,169 +1,68 @@
-import graphene
-from graphene.relay import Node
-
+from . import models
 from ..types import MongoengineObjectType
-from .models import (
-    Article, Editor, EmbeddedArticle, Player, Reporter,
-    Parent, Child, ProfessorMetadata, ProfessorVector,
-    ParentWithRelationship, ChildRegisteredBefore,
-    ChildRegisteredAfter, CellTower,
-    Publisher, ErroneousModel)
 
 
 class PublisherType(MongoengineObjectType):
 
     class Meta:
-        model = Publisher
+        model = models.Publisher
 
 
 class EditorType(MongoengineObjectType):
 
     class Meta:
-        model = Editor
+        model = models.Editor
 
 
 class ArticleType(MongoengineObjectType):
 
     class Meta:
-        model = Article
+        model = models.Article
 
 
 class EmbeddedArticleType(MongoengineObjectType):
 
     class Meta:
-        model = EmbeddedArticle
+        model = models.EmbeddedArticle
 
 
 class PlayerType(MongoengineObjectType):
 
     class Meta:
-        model = Player
+        model = models.Player
 
 
 class ReporterType(MongoengineObjectType):
 
     class Meta:
-        model = Reporter
+        model = models.Reporter
 
 
 class ParentType(MongoengineObjectType):
 
     class Meta:
-        model = Parent
+        model = models.Parent
 
 
 class ChildType(MongoengineObjectType):
 
     class Meta:
-        model = Child
+        model = models.Child
 
 
 class CellTowerType(MongoengineObjectType):
 
     class Meta:
-        model = CellTower
+        model = models.CellTower
 
 
 class ProfessorMetadataType(MongoengineObjectType):
 
     class Meta:
-        model = ProfessorMetadata
+        model = models.ProfessorMetadata
 
 
 class ProfessorVectorType(MongoengineObjectType):
 
     class Meta:
-        model = ProfessorVector
-
-
-class PublisherNode(MongoengineObjectType):
-    legal_name = graphene.String()
-    bad_field = graphene.String()
-
-    class Meta:
-        model = Publisher
-        only_fields = ('id', 'name')
-        interfaces = (Node,)
-
-
-class ArticleNode(MongoengineObjectType):
-
-    class Meta:
-        model = Article
-        interfaces = (Node,)
-
-
-class EditorNode(MongoengineObjectType):
-
-    class Meta:
-        model = Editor
-        interfaces = (Node,)
-
-
-class EmbeddedArticleNode(MongoengineObjectType):
-
-    class Meta:
-        model = EmbeddedArticle
-        interfaces = (Node,)
-
-
-class PlayerNode(MongoengineObjectType):
-
-    class Meta:
-        model = Player
-        interfaces = (Node,)
-
-
-class ReporterNode(MongoengineObjectType):
-
-    class Meta:
-        model = Reporter
-        interfaces = (Node,)
-
-
-class ParentNode(MongoengineObjectType):
-
-    class Meta:
-        model = Parent
-        interfaces = (Node,)
-
-
-class ChildNode(MongoengineObjectType):
-
-    class Meta:
-        model = Child
-        interfaces = (Node,)
-
-
-class ChildRegisteredBeforeNode(MongoengineObjectType):
-
-    class Meta:
-        model = ChildRegisteredBefore
-        interfaces = (Node, )
-
-
-class ParentWithRelationshipNode(MongoengineObjectType):
-
-    class Meta:
-        model = ParentWithRelationship
-        interfaces = (Node, )
-
-
-class ChildRegisteredAfterNode(MongoengineObjectType):
-
-    class Meta:
-        model = ChildRegisteredAfter
-        interfaces = (Node, )
-
-
-class ProfessorVectorNode(MongoengineObjectType):
-
-    class Meta:
-        model = ProfessorVector
-        interfaces = (Node, )
-
-
-class ErroneousModelNode(MongoengineObjectType):
-    class Meta:
-        model = ErroneousModel
-        interfaces = (Node,)
+        model = models.ProfessorVector