check if structure is not read only before save

1 files changed, 7 insertions, 0 deletions

diff --git a/webapp/polls/tests/structure_tests.py b/webapp/polls/tests/structure_tests.py
index a7857f4..aeb8702 100644
--- a/webapp/polls/tests/structure_tests.py
+++ b/webapp/polls/tests/structure_tests.py
@@ -7,6 +7,7 @@ from mock import Mock
 from utils.test import (MongoTestCase, mock_in_memory_image,
                         remove_option_images_dir)
 from polls.tests.poll_tests import MockImageFileInterface
+from polls.exceptions import ReadOnly
 
 
 class StructureTests(MongoTestCase):
@@ -434,3 +435,9 @@ class SaveStructureTest(MongoTestCase):
         structure = Structure(data=self.data, poll=self.poll)
         structure.save()
         self.assertEqual(1, self.db.structures.count())
+
+    def test_it_should_not_save_if_is_read_only(self):
+        poll = self.poll
+        structure = Structure(data=self.data, poll=poll)
+        structure.is_read_only = Mock(return_value=True)
+        self.assertRaises(ReadOnly, structure.save)