Better checking for image options

1 files changed, 32 insertions, 0 deletions

diff --git a/webapp/polls/tests/structure_tests.py b/webapp/polls/tests/structure_tests.py
index 4f28dd5..e7ee213 100644
--- a/webapp/polls/tests/structure_tests.py
+++ b/webapp/polls/tests/structure_tests.py
@@ -2,6 +2,7 @@
 from polls.models import Poll, Group, Field, Structure
 
 from utils.test import MongoTestCase, mock_in_memory_image
+from polls.tests.poll_tests import MockImageFileInterface
 
 
 class StructureTests(MongoTestCase):
@@ -284,3 +285,34 @@ class StructureTests(MongoTestCase):
 
         self.assertEqual(
             img_name, structure.groups[0].fields[0].get_img_name())
+
+    def test_get_img_options(self):
+        poll = Poll({"name": "poll name"})
+        poll_id = poll.save()
+        poll = poll.get(poll_id)
+        img = "img"
+        data = {
+            'groups': {
+                '0': {
+                    'name': "group name",
+                    'fields': {
+                        '0': {
+                            'name': "field_0_0",
+                            'widget_type': Field.ImageCheckBox,
+                            'options': {
+                                '131212': {
+                                    'img': MockImageFileInterface(),
+                                    'text': "text",
+                                    'order': 0
+                                }
+                            },
+                            'img': img
+                        }
+                    }
+                }
+            }
+        }
+
+        structure = Structure(data=data, poll=poll)
+
+        self.assertEqual(1, len(structure.get_image_options()))