feat: Skip EmbeddedDocument field in reference_args

Author: abawchen

graphene_mongo/fields.py

@@ -1,5 +1,6 @@
 from __future__ import absolute_import
 
+import mongoengine
 from collections import OrderedDict
 from functools import partial, reduce
 
@@ -69,7 +70,6 @@ def args(self):
         )
         return args
 
-
     @args.setter
     def args(self, args):
         self._base_args = args
@@ -94,7 +94,7 @@ def reference_args(self):
         def get_reference_field(r, kv):
             if callable(getattr(kv[1], 'get_type', None)):
                 node = kv[1].get_type()._type._meta
-                if not isinstance(kv[1], Dynamic):
+                if not issubclass(node.model, mongoengine.EmbeddedDocument):
                     r.update({kv[0]: node.fields['id']._type.of_type()})
             return r
         return reduce(get_reference_field, self.fields.items(), {})

graphene_mongo/tests/test_relay_query.py

@@ -729,7 +729,7 @@ class Query(graphene.ObjectType):
     assert json.dumps(result.data, sort_keys=True) == json.dumps(
         expected, sort_keys=True)
 
-# @pytest.mark.skip(reason="no way of currently testing this")
+
 def test_should_query_with_embedded_document(fixtures):
 
     class Query(graphene.ObjectType):
@@ -741,7 +741,10 @@ class Query(graphene.ObjectType):
       allProfessors {
         edges {
             node {
-                vec
+                vec,
+                metadata {
+                     firstName
+                }
             }
         }
       }
@@ -752,7 +755,10 @@ class Query(graphene.ObjectType):
             'edges': [
                 {
                     'node': {
-                        'vec': [1.0, 2.3]
+                        'vec': [1.0, 2.3],
+                        'metadata': {
+                             'firstName': 'Steven'
+                        }
                     }
 
                 }
@@ -761,6 +767,5 @@ class Query(graphene.ObjectType):
     }
     schema = graphene.Schema(query=Query)
     result = schema.execute(query)
-    print(result.data)
-    # assert not result.errors
-    # assert dict(result.data['allProfessors']) == expected['allProfessors']
+    assert not result.errors
+    assert dict(result.data['allProfessors']) == expected['allProfessors']