test: Add corresponding test cases.

Author: abawchen

graphene_mongo/tests/test_relay_query.py

@@ -9,7 +9,8 @@
 from .types import (ArticleNode, ArticleType,
                     EditorNode, EditorType,
                     PlayerNode, PlayerType,
-                    ReporterNode, ReporterType)
+                    ReporterNode, ReporterType,
+                    ChildNode, ChildType)
 from ..fields import MongoengineConnectionField
 
 setup_fixtures()
@@ -294,6 +295,43 @@ class Query(graphene.ObjectType):
     assert result.data == expected
 
 
+def test_should_filter_through_inheritance():
+
+    class Query(graphene.ObjectType):
+        node = Node.Field()
+        children = MongoengineConnectionField(ChildNode)
+
+    query = '''
+        query ChildrenQuery {
+            children(bar: "bar") {
+                edges {
+                    node {
+                        bar,
+                        baz
+                    }
+                }
+            }
+        }
+    '''
+    expected = {
+        'children': {
+            'edges': [
+                {
+                    'node': {
+                        'bar': 'bar',
+                        'baz': 'baz',
+                    }
+                }
+            ]
+        }
+    }
+    schema = graphene.Schema(query=Query)
+    result = schema.execute(query)
+    assert not result.errors
+    assert json.dumps(result.data, sort_keys=True) == json.dumps(
+        expected, sort_keys=True)
+
+
 def test_should_get_node_by_id():
     # Notes: https://goo.gl/hMNRgs
     class Query(graphene.ObjectType):

graphene_mongo/tests/test_types.py

@@ -6,6 +6,7 @@
 from .. import registry
 from ..types import MongoengineObjectType
 from .models import Article, EmbeddedArticle, Reporter
+from .models import Parent, Child
 from .utils import with_local_registry
 
 registry.reset_global_registry()
@@ -28,11 +29,26 @@ class Meta:
         interfaces = (Node,)
 
 class Character(MongoengineObjectType):
+
     class Meta:
         model = Reporter
         registry = registry.get_global_registry()
 
 
+class Dad(MongoengineObjectType):
+
+    class Meta:
+        model = Parent
+        registry = registry.get_global_registry()
+
+
+class Son(MongoengineObjectType):
+
+    class Meta:
+        model = Child
+        registry = registry.get_global_registry()
+
+
 def test_mongoengine_interface():
     assert issubclass(Node, Interface)
     assert issubclass(Node, Node)
@@ -53,6 +69,9 @@ def test_objecttype_registered():
     ])
 
 
+def test_mongoengine_inheritance():
+    assert issubclass(Son._meta.model, Dad._meta.model)
+
 def test_node_replacedfield():
     idfield = Human._meta.fields['pub_date']
     assert isinstance(idfield, Field)

graphene_mongo/tests/types.py

@@ -1,7 +1,10 @@
 from graphene.relay import Node
 
 from ..types import MongoengineObjectType
-from .models import Article, Editor, EmbeddedArticle, Player, Reporter
+from .models import (
+    Article, Editor, EmbeddedArticle, Player, Reporter,
+    Parent, Child
+)
 
 
 class EditorType(MongoengineObjectType):
@@ -34,6 +37,18 @@ class Meta:
         model = Reporter
 
 
+class ParentType(MongoengineObjectType):
+
+    class Meta:
+        model = Parent
+
+
+class ChildType(MongoengineObjectType):
+
+    class Meta:
+        model = Child
+
+
 class ArticleNode(MongoengineObjectType):
 
     class Meta:
@@ -68,3 +83,15 @@ class Meta:
         model = Reporter
         interfaces = (Node,)
 
+
+class ParentNode(MongoengineObjectType):
+
+    class Meta:
+        model = Parent
+        interfaces = (Node,)
+
+
+class ChildNode(MongoengineObjectType):
+    class Meta:
+        model = Child
+        interfaces = (Node,)