test for bugfix: "images with many chunks

1 files changed, 45 insertions, 0 deletions

diff --git a/webapp/polls/tests/structure_tests.py b/webapp/polls/tests/structure_tests.py
index e7ee213..abb7bd8 100644
--- a/webapp/polls/tests/structure_tests.py
+++ b/webapp/polls/tests/structure_tests.py
@@ -1,5 +1,8 @@
 # -*- encoding: utf-8 -*-
+import os
+
 from polls.models import Poll, Group, Field, Structure
+from mock import Mock
 
 from utils.test import MongoTestCase, mock_in_memory_image
 from polls.tests.poll_tests import MockImageFileInterface
@@ -316,3 +319,45 @@ class StructureTests(MongoTestCase):
         structure = Structure(data=data, poll=poll)
 
         self.assertEqual(1, len(structure.get_image_options()))
+
+
+class UploadOptionImagesTest(MongoTestCase):
+
+    def test_it_should_save_option_images_with_many_chunks(self):
+        option_id = '131212'
+        img_ext = "jpg"
+        img_filename = '.'.join([option_id, img_ext])
+        img = Mock()
+        img.chunks = Mock(return_value=['chunk', 'chunk'])
+        img.name = img_filename
+
+        data = {
+            'groups': {
+                '0': {
+                    'name': "group name",
+                    'fields': {
+                        '0': {
+                            'name': "field_0_0",
+                            'widget_type': Field.ImageCheckBox,
+                            'options': {
+                                option_id: {
+                                    'img': img,
+                                    'text': "text",
+                                    'order': 0
+                                }
+                            }
+                        }
+                    }
+                }
+            }
+        }
+        poll = Poll({"name": "poll name"})
+        poll_id = poll.save()
+        poll = poll.get(poll_id)
+        structure = Structure(data=data, poll=poll)
+
+        structure.save_image_options()
+
+        images_dir = structure.get_image_options_path()
+        saved_on_disk = os.path.exists(os.path.join(images_dir, img_filename))
+        self.assertTrue(saved_on_disk)