test: Add EmbeddedDocument as document field

Author: abawchen

graphene_mongo/tests/fixtures.py

@@ -1,6 +1,6 @@
 from .models import (
     Article, Editor, EmbeddedArticle, Player,
-    Reporter, Child
+    Reporter, Child, ProfessorMetadata, ProfessorVector,
 )
 
 
@@ -84,3 +84,15 @@ def setup_fixtures():
 
     child2 = Child(bar='bar', baz='baz')
     child2.save()
+
+    ProfessorVector.drop_collection()
+    professor_metadata = ProfessorMetadata(
+        first_name='Steven',
+        last_name='Curry',
+        departments=['NBA', 'MLB']
+    )
+    professor_vector = ProfessorVector(
+        vec=[1.0, 2.3],
+        metadata=professor_metadata
+    )
+    professor_vector.save()

graphene_mongo/tests/models.py

@@ -4,7 +4,7 @@
 )
 from mongoengine.fields import (
     DateTimeField, EmailField, EmbeddedDocumentField,
-    EmbeddedDocumentListField, ListField,
+    FloatField, EmbeddedDocumentListField, ListField,
     MapField, ReferenceField, StringField
 )
 
@@ -59,6 +59,8 @@ class Reporter(Document):
 
 
 class Player(Document):
+
+    meta = {'collection': 'test_player'}
     first_name = StringField(required=True)
     last_name = StringField(required=True)
     opponent = ReferenceField('Player')
@@ -78,5 +80,22 @@ class Parent(Document):
 
 class Child(Parent):
 
-    meta = {'collection': 'Child'}
+    meta = {'collection': 'test_child'}
     baz = StringField()
+
+
+class ProfessorMetadata(EmbeddedDocument):
+
+    meta = {'collection': 'test_professor_metadata'}
+    # id = StringField()
+    first_name = StringField()
+    last_name = StringField()
+    departments = ListField(StringField())
+
+
+class ProfessorVector(Document):
+
+    meta = {'collection': 'test_professor_vector'}
+    # id = StringField()
+    vec = ListField(FloatField())
+    metadata = EmbeddedDocumentField(ProfessorMetadata)

graphene_mongo/tests/test_converter.py

@@ -6,8 +6,10 @@
 
 from py.test import raises
 
-from .models import Article, Editor, EmbeddedArticle, Player, Reporter
-
+from .models import (
+    Article, Editor, EmbeddedArticle, Player, Reporter,
+    ProfessorMetadata, ProfessorVector,
+)
 from .. import registry
 from ..converter import convert_mongoengine_field
 from ..fields import MongoengineConnectionField
@@ -96,6 +98,22 @@ class Meta:
     assert graphene_type.type == E
 
 
+def test_should_embedded_convert_dynamic():
+
+    class PM(MongoengineObjectType):
+
+        class Meta:
+            model = ProfessorMetadata
+            interfaces = (Node,)
+
+    dynamic_field = convert_mongoengine_field(
+        ProfessorVector._fields['metadata'], PM._meta.registry)
+    assert isinstance(dynamic_field, Dynamic)
+    graphene_type = dynamic_field.get_type()
+    assert isinstance(graphene_type, graphene.Field)
+    assert graphene_type.type == PM
+
+
 def test_should_convert_none():
     registry.reset_global_registry()
     dynamic_field = convert_mongoengine_field(