wip: test_converter and test_query pass

Author: abawchen

graphene_mongo/converter.py

@@ -65,20 +65,25 @@ def convert_field_to_float(field, registry=None):
     return Float(description=get_field_description(field, registry), required=field.required)
 
 
+@convert_mongoengine_field.register(mongoengine.DateTimeField)
+def convert_field_to_datetime(field, registry=None):
+    return DateTime(description=get_field_description(field, registry), required=field.required)
+
+
 @convert_mongoengine_field.register(mongoengine.DictField)
 @convert_mongoengine_field.register(mongoengine.MapField)
 def convert_field_to_jsonstring(field, registry=None):
     return JSONString(description=get_field_description(field, registry), required=field.required)
 
 
 @convert_mongoengine_field.register(mongoengine.PointField)
-def convert_point_to_field(field, register=None):
+def convert_point_to_field(field, registry=None):
     from . import advanced_types
     return Field(advanced_types.PointFieldType)
 
 
 @convert_mongoengine_field.register(mongoengine.PolygonField)
-def convert_polygon_to_field(field, register=None):
+def convert_polygon_to_field(field, registry=None):
     from . import advanced_types
     return Field(advanced_types.PolygonFieldType)
 
@@ -89,17 +94,26 @@ def convert_multipolygon_to_field(field, register=None):
     return Field(advanced_types.MultiPolygonFieldType)
 
 
-@convert_mongoengine_field.register(mongoengine.DateTimeField)
-def convert_field_to_datetime(field, registry=None):
-    return DateTime(description=get_field_description(field, registry), required=field.required)
-
-
 @convert_mongoengine_field.register(mongoengine.FileField)
-def convert_field_to_none(field, register=None):
-    from .advanced_types import FsFileType
-    print(FsFileType)
-    # FIXME
-    return None
+def convert_file_to_field(field, registry=None):
+    from . import advanced_types
+    """
+    from .advanced_models import FsFile
+
+    Meta = type(
+        'Meta',
+        (object,),
+        {'model': FsFile, 'interfaces': (Node,)}
+    )
+    _ = type(
+        '_FsFileType',
+        (MongoengineObjectType,),
+        {'Meta': Meta}
+    )
+    field = mongoengine.ReferenceField(FsFile)
+    """
+    field = mongoengine.ReferenceField(advanced_types.FsFile)
+    return convert_mongoengine_field(field, registry)
 
 
 @convert_mongoengine_field.register(mongoengine.ListField)

graphene_mongo/tests/models.py

@@ -3,7 +3,8 @@
 from mongomock import gridfs
 
 gridfs.enable_gridfs_integration()
-mongoengine.connect('graphene-mongo-test', host='mongomock://localhost', alias='default')
+# mongoengine.connect('graphene-mongo-test', host='mongomock://localhost', alias='default')
+mongoengine.connect('graphene-mongo-test', host='mongodb://192.168.15.91/graphene-mongo-dev')
 
 
 class Publisher(mongoengine.Document):

graphene_mongo/tests/setup.py

@@ -27,10 +27,16 @@ def fixtures():
     )
     image_filename = os.path.join(
         current_dirname, 'fixtures', 'image.jpg')
-    with open(image_filename, 'r') as f:
+    with open(image_filename, 'rb') as f:
         editor1.avatar.put(f, content_type='image/jpeg')
     editor1.save()
 
+    """
+    e = Editor.objects(first_name='Penny').first()
+    avatar = e.avatar.read()
+    print(e.avatar.content_type)
+    """
+
     editor2 = Editor(
         id='2',
         first_name='Grant',

graphene_mongo/tests/test_converter.py

@@ -98,6 +98,16 @@ def test_should_multipolygon_convert_field():
     assert isinstance(graphene_type.type.coordinates, graphene.List)
 
 
+def test_should_file_convert_field():
+    from ..advanced_types import FsFileType
+    dynamic_field = convert_mongoengine_field(
+        mongoengine.FileField(), registry.get_global_registry())
+    assert isinstance(dynamic_field, graphene.Dynamic)
+    graphene_type = dynamic_field.get_type()
+    assert isinstance(graphene_type, graphene.Field)
+    assert graphene_type.type == FsFileType
+
+
 def test_should_field_convert_list():
     assert_conversion(mongoengine.ListField, graphene.List, field=mongoengine.StringField())
 
@@ -297,8 +307,6 @@ class Meta:
 
 
 def test_should_generic_reference_convert_union():
-    pass
-    """
     class A(MongoengineObjectType):
 
         class Meta:
@@ -318,4 +326,3 @@ class Meta:
         Reporter._fields['generic_reference'], registry.get_global_registry())
     assert isinstance(generic_reference_field, graphene.Field)
     assert isinstance(generic_reference_field.type(), graphene.Union)
-    """

graphene_mongo/tests/test_query.py

@@ -31,6 +31,12 @@ def resolve_editors(self, *args, **kwargs):
                 metadata,
                 company {
                     name
+                },
+                avatar {
+                    contentType,
+                    chunkSize,
+                    length,
+                    md5
                 }
             }
             editors {
@@ -42,7 +48,13 @@ def resolve_editors(self, *args, **kwargs):
     expected = {
         'editor': {
             'firstName': 'Penny',
-            'company': {'name': 'Newsco'}
+            'company': {'name': 'Newsco'},
+            'avatar': {
+                'contentType': 'image/jpeg',
+                'chunkSize': 261120,
+                'length': 46928,
+                'md5': 'f3c657fd472fdc4bc2ca9056a1ae6106'
+            }
         },
         'editors': [{
             'firstName': 'Penny',

graphene_mongo/types.py

@@ -163,8 +163,17 @@ 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
+        if isinstance(root, mongoengine.GridFSProxy):
+            return True
         if not is_valid_mongoengine_model(type(root)):
             raise Exception((
                 'Received incompatible instance "{}".'