copy test_save to his own test class

1 files changed, 35 insertions, 0 deletions

diff --git a/webapp/polls/tests/structure_tests.py b/webapp/polls/tests/structure_tests.py
index ebb0f7e..9e499a8 100644
--- a/webapp/polls/tests/structure_tests.py
+++ b/webapp/polls/tests/structure_tests.py
@@ -413,3 +413,38 @@ class UploadOptionImagesTest(MongoTestCase):
 
     def tearDown(self):
         remove_option_images_dir()
+
+
+class SaveStructureTest(MongoTestCase):
+
+    def setUp(self):
+        poll = Poll(data={'name': 'name'})
+        poll_id = poll.save()
+
+        self.poll = Poll.get(id=poll_id)
+
+        self.data = {
+            'poll_id': 'POLL_ID',
+            'groups': {
+                '0': {
+                    'name': 'group_0',
+                    'fields': {
+                        '0': {
+                            'widget_type': Field.TextInput,
+                            'name': 'field_0_0'
+                        }
+                    }
+                }
+            }
+        }
+
+    def test_save(self):
+
+        structure = Structure(data=self.data)
+
+        self.assertRaises(structure.ValidationError, structure.save)
+
+        structure = Structure(data=self.data, poll=self.poll)
+        structure.save()
+
+        self.assertEqual(1, self.db.structures.count())