#128 - fixed test

collective · Sep 11, 2017 · 8b5b9c9 · 8b5b9c9
1 parent ea5bece
commit 8b5b9c9
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/sc/social/like/utils.py b/sc/social/like/utils.py
@@ -149,7 +149,11 @@ def validate_image_social(value):
         logger.info(msg.format(type))
         return msg.format(type)
 
-    if value.data.size > 5242880:
+    try:
+        size = value.size
+    except AttributeError:
+        size = value.data.size
+    if size > 5242880:
         msg = u'Image size should be less than 5MB.'
         logger.info(msg)
         return msg