test: test_relay_query pass, but make test failed

Author: abawchen

graphene_mongo/fields.py

@@ -103,6 +103,8 @@ def field_args(self):
 
     @property
     def reference_args(self):
+        from .advanced_types import FsFile
+
         def get_reference_field(r, kv):
             field = kv[1]
             mongo_field = getattr(self.model, kv[0], None)
@@ -112,7 +114,7 @@ def get_reference_field(r, kv):
                 _type = field.get_type()
                 if _type:
                     node = _type._type._meta
-                    if 'id' in node.fields and not issubclass(node.model, mongoengine.EmbeddedDocument):
+                    if 'id' in node.fields and not issubclass(node.model, (mongoengine.EmbeddedDocument, FsFile)):
                         r.update({kv[0]: node.fields['id']._type.of_type()})
             return r
 

graphene_mongo/tests/test_relay_query.py

@@ -141,7 +141,10 @@ class Query(graphene.ObjectType):
                 node {
                     id,
                     firstName,
-                    lastName
+                    lastName,
+                    avatar {
+                        contentType
+                    }
                 }
             }
           }
@@ -154,22 +157,31 @@ class Query(graphene.ObjectType):
                     'node': {
                         'id': 'RWRpdG9yTm9kZTox',
                         'firstName': 'Penny',
-                        'lastName': 'Hardaway'
+                        'lastName': 'Hardaway',
+                        'avatar': {
+                            'contentType': 'image/jpeg'
+                        }
                     }
                 },
                 {
                     'node': {
                         'id': 'RWRpdG9yTm9kZToy',
                         'firstName': 'Grant',
-                        'lastName': 'Hill'
+                        'lastName': 'Hill',
+                        'avatar': {
+                            'contentType': None
+                        }
                     }
 
                 },
                 {
                     'node': {
                         'id': 'RWRpdG9yTm9kZToz',
                         'firstName': 'Dennis',
-                        'lastName': 'Rodman'
+                        'lastName': 'Rodman',
+                        'avatar': {
+                            'contentType': None
+                        }
                     }
                 }
             ]

graphene_mongo/types.py

@@ -163,15 +163,9 @@ def rescan_fields(cls):
     # noqa
     @classmethod
     def is_type_of(cls, root, info):
-        print('cls')
-        print(cls)
-        print('root')
-        print(root.__dict__)
-        print(root)
-        print(type(root))
-        print('*' * 20)
         if isinstance(root, cls):
             return True
+        # XXX: Take care FileField
         if isinstance(root, mongoengine.GridFSProxy):
             return True
         if not is_valid_mongoengine_model(type(root)):