new: Add test case for type register order

Author: FloatingGhost

graphene_mongo/tests/models.py

@@ -98,3 +98,25 @@ class ProfessorVector(Document):
     meta = {'collection': 'test_professor_vector'}
     vec = ListField(FloatField())
     metadata = EmbeddedDocumentField(ProfessorMetadata)
+
+
+class ParentWithRelationship(Document):
+
+    meta = {'collection': 'test_parent_reference'}
+    before_child = ReferenceField("ChildRegisteredBefore")
+    after_child = ReferenceField("ChildRegisteredAfter")
+    name = StringField()
+
+
+class ChildRegisteredBefore(Document):
+
+    meta = {'collection': 'test_child_before_reference'}
+    parent = ReferenceField(ParentWithRelationship)
+    name = StringField()
+
+
+class ChildRegisteredAfter(Document):
+
+    meta = {'collection': 'test_child_after_reference'}
+    parent = ReferenceField(ParentWithRelationship)
+    name = StringField()

graphene_mongo/tests/setup.py

@@ -2,6 +2,8 @@
 from .models import (
     Article, Editor, EmbeddedArticle, Player,
     Reporter, Child, ProfessorMetadata, ProfessorVector,
+    ChildRegisteredBefore, ChildRegisteredAfter,
+    ParentWithRelationship
 )
 
 
@@ -105,3 +107,22 @@ def fixtures():
         metadata=professor_metadata
     )
     professor_vector.save()
+
+    ParentWithRelationship.drop_collection()
+    ChildRegisteredAfter.drop_collection()
+    ChildRegisteredBefore.drop_collection()
+
+    # This is one messed up family
+
+    # She'd better have presence this time
+    child3 = ChildRegisteredBefore(name="Akari")
+    child4 = ChildRegisteredAfter(name="Kyouko")
+    child3.save()
+    child4.save()
+
+    parent = ParentWithRelationship(name="Yui", before_child=child3, after_child=child4)
+    parent.save()
+
+    child3.parent = child4.parent = parent
+    child3.save()
+    child4.save()

graphene_mongo/tests/test_relay_query.py

@@ -10,7 +10,8 @@
                     EditorNode,
                     PlayerNode,
                     ReporterNode,
-                    ChildNode,)
+                    ChildNode,
+                    ParentWithRelationshipNode)
 from ..fields import MongoengineConnectionField
 
 
@@ -648,3 +649,53 @@ class Query(graphene.ObjectType):
     assert not result.errors
     assert json.dumps(result.data, sort_keys=True) == json.dumps(
         expected, sort_keys=True)
+
+
+def test_should_lazy_reference(fixtures):
+
+    class Query(graphene.ObjectType):
+        node = Node.Field()
+        parents = MongoengineConnectionField(ParentWithRelationshipNode)
+
+    schema = graphene.Schema(query=Query)
+
+    query = """
+    query {
+        parents {
+            edges {
+                node {
+                    beforeChild {
+                        name,
+                        parent { name }
+                    },
+                    afterChild {
+                        name,
+                        parent { name }
+                    }
+                }
+            }
+        }
+    }
+    """
+
+    expected = {
+        "parents": {
+            "edges": [
+                {"node": {
+                    "beforeChild": {
+                        "name": "Akari",
+                        "parent": {"name": "Yui"}
+                    },
+                    "afterChild": {
+                        "name": "Kyouko",
+                        "parent": {"name": "Yui"}
+                    }
+                }}
+            ]
+        }
+    }
+
+    result = schema.execute(query)
+    assert not result.errors
+    assert json.dumps(result.data, sort_keys=True) == json.dumps(
+        expected, sort_keys=True)

graphene_mongo/tests/types.py

@@ -4,6 +4,8 @@
 from .models import (
     Article, Editor, EmbeddedArticle, Player, Reporter,
     Parent, Child, ProfessorMetadata, ProfessorVector,
+    ParentWithRelationship, ChildRegisteredBefore,
+    ChildRegisteredAfter
 )
 
 
@@ -107,3 +109,21 @@ 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, )