fix: Set relationship types to list

Author: FloatingGhost

graphene_mongo/tests/models.py

@@ -103,8 +103,8 @@ class ProfessorVector(Document):
 class ParentWithRelationship(Document):
 
     meta = {'collection': 'test_parent_reference'}
-    before_child = ReferenceField("ChildRegisteredBefore")
-    after_child = ReferenceField("ChildRegisteredAfter")
+    before_child = ListField(ReferenceField("ChildRegisteredBefore"))
+    after_child = ListField(ReferenceField("ChildRegisteredAfter"))
     name = StringField()
 
 

graphene_mongo/tests/setup.py

@@ -120,7 +120,12 @@ def fixtures():
     child3.save()
     child4.save()
 
-    parent = ParentWithRelationship(name="Yui", before_child=child3, after_child=child4)
+    parent = ParentWithRelationship(
+        name="Yui",
+        before_child=[child3],
+        after_child=[child4]
+    )
+
     parent.save()
 
     child3.parent = child4.parent = parent

graphene_mongo/tests/test_relay_query.py

@@ -665,12 +665,20 @@ class Query(graphene.ObjectType):
             edges {
                 node {
                     beforeChild {
-                        name,
-                        parent { name }
+                        edges {
+                            node {
+                                name,
+                                parent { name }
+                            }
+                        }
                     },
                     afterChild {
-                        name,
-                        parent { name }
+                        edges {
+                            node {
+                                name,
+                                parent { name }
+                            }
+                        }
                     }
                 }
             }
@@ -683,12 +691,20 @@ class Query(graphene.ObjectType):
             "edges": [
                 {"node": {
                     "beforeChild": {
-                        "name": "Akari",
-                        "parent": {"name": "Yui"}
+                        "edges": [
+                            {"node": {
+                                "name": "Akari",
+                                "parent": {"name": "Yui"}
+                            }}
+                        ]
                     },
                     "afterChild": {
-                        "name": "Kyouko",
-                        "parent": {"name": "Yui"}
+                        "edges": [
+                            {"node": {
+                                "name": "Kyouko",
+                                "parent": {"name": "Yui"}
+                            }}
+                        ]
                     }
                 }}
             ]