Refactor in test

1 files changed, 21 insertions, 12 deletions

diff --git a/webapp/polls/tests/structure_tests.py b/webapp/polls/tests/structure_tests.py
index f11da61..2f793c6 100644
--- a/webapp/polls/tests/structure_tests.py
+++ b/webapp/polls/tests/structure_tests.py
@@ -324,15 +324,9 @@ class StructureTests(MongoTestCase):
 
 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 = {
+    def setUp(self):
+        self.option_id = '131212'
+        self.data = {
             'groups': {
                 '0': {
                     'name': "group name",
@@ -341,8 +335,8 @@ class UploadOptionImagesTest(MongoTestCase):
                             'name': "field_0_0",
                             'widget_type': Field.ImageCheckBox,
                             'options': {
-                                option_id: {
-                                    'img': img,
+                                self.option_id: {
+                                    'img': None,
                                     'text': "text",
                                     'order': 0
                                 }
@@ -354,7 +348,22 @@ class UploadOptionImagesTest(MongoTestCase):
         }
         poll = Poll({"name": "poll name"})
         poll_id = poll.save()
-        poll = poll.get(poll_id)
+        self.poll = poll.get(poll_id)
+
+        img_ext = "jpg"
+        self.img_filename = '.'.join([self.option_id, img_ext])
+        self.img = Mock()
+        self.img.chunks = Mock(return_value=['chunk', 'chunk'])
+        self.img.name = self.img_filename
+
+    def test_it_should_save_option_images_with_many_chunks(self):
+        option_id = self.option_id
+        img = self.img
+        img_filename = self.img_filename
+
+        data = self.data
+        data['groups']['0']['fields']['0']['options'][option_id]['img'] = img
+        poll = self.poll
         structure = Structure(data=data, poll=poll)
 
         structure.save_image_options()