Skip to content

Commit 9fe4ed1

Browse files
committed
test: Upgrade mongomock to support gridfs
1 parent 4b2db8a commit 9fe4ed1

File tree

5 files changed

+19
-3
lines changed

5 files changed

+19
-3
lines changed

graphene_mongo/converter.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -92,6 +92,12 @@ def convert_field_to_datetime(field, registry=None):
9292
return DateTime(description=get_field_description(field, registry), required=field.required)
9393

9494

95+
@convert_mongoengine_field.register(mongoengine.FileField)
96+
def convert_field_to_none(field, register=None):
97+
# FIXME
98+
return None
99+
100+
95101
@convert_mongoengine_field.register(mongoengine.ListField)
96102
@convert_mongoengine_field.register(mongoengine.EmbeddedDocumentListField)
97103
def convert_field_to_list(field, registry=None):
45.8 KB
Loading

graphene_mongo/tests/models.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,8 @@
11
import mongoengine
22
from datetime import datetime
3+
from mongomock import gridfs
34

5+
gridfs.enable_gridfs_integration()
46
mongoengine.connect('graphene-mongo-test', host='mongomock://localhost', alias='default')
57

68

@@ -28,6 +30,7 @@ class Editor(mongoengine.Document):
2830
last_name = mongoengine.StringField(required=True, help_text="Editor's last name.")
2931
metadata = mongoengine.MapField(field=mongoengine.StringField(), help_text="Arbitrary metadata.")
3032
company = mongoengine.LazyReferenceField(Publisher)
33+
avatar = mongoengine.FileField()
3134

3235

3336
class Article(mongoengine.Document):

graphene_mongo/tests/setup.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
import os
12
import pytest
23

34
from datetime import datetime
@@ -8,6 +9,7 @@
89
ParentWithRelationship, CellTower,
910
Publisher)
1011

12+
current_dirname = os.path.dirname(os.path.abspath(__file__))
1113

1214
@pytest.fixture(scope='module')
1315
def fixtures():
@@ -23,7 +25,12 @@ def fixtures():
2325
metadata={'age': '20', 'nickname': '$1'},
2426
company=publisher1
2527
)
28+
image_filename = os.path.join(
29+
current_dirname, 'fixtures', 'image.jpg')
30+
with open(image_filename, 'r') as f:
31+
editor1.avatar.put(f, content_type='image/jpeg')
2632
editor1.save()
33+
print('hey:', editor1)
2734
editor2 = Editor(
2835
id='2',
2936
first_name='Grant',

requirements.txt

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,11 @@
11
coveralls==1.2.0
22
flake8==3.5.0
33
flake8-per-file-ignores==0.6
4+
future==0.17.1
45
graphene>=2.1.3,<3
56
iso8601==0.1.12
6-
mongoengine==0.15.0
7-
mongomock==3.8.0
7+
mongoengine==0.16.3
8+
mongomock==3.14.0
89
pytest==3.3.2
910
pytest-cov==2.5.1
1011
singledispatch==3.4.0.3
11-

0 commit comments

Comments
 (0)