fill_gallery generates thumbnail synchronously

uaprom-summer-2015 · Aug 20, 2015 · c6507b7 · c6507b7
1 parent 4663494
commit c6507b7
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/project/gallery.py b/project/gallery.py
@@ -1,12 +1,18 @@
 from werkzeug.datastructures import FileStorage
 from project.models import UploadedImage
+from unittest.mock import patch
+from project.tasks.uploads import celery_make_thumbnail as make_thumbnail
 
 
 def load_images(count=100):
     for _ in range(count):
         with open('testdata/images/face-2.jpg', 'rb') as fp:
             file = FileStorage(fp)
-            UploadedImage.bl.save_image(
-                image=file,
-                img_category=UploadedImage.IMG_CATEGORY.gallery,
-            )
+            with patch(
+                target='project.tasks.uploads.celery_make_thumbnail.delay',
+                new=make_thumbnail,
+            ):
+                UploadedImage.bl.save_image(
+                    image=file,
+                    img_category=UploadedImage.IMG_CATEGORY.gallery,
+                )