add is_read_only method to structure class

2 files changed, 15 insertions, 0 deletions

diff --git a/webapp/polls/models.py b/webapp/polls/models.py
index a356e92..d254e3b 100644
--- a/webapp/polls/models.py
+++ b/webapp/polls/models.py
@@ -1320,6 +1320,11 @@ class Structure(AbstractObject, ComponentStructure):
                 # TODO: LOG!
                 pass
 
+    def is_read_only(self):
+        """Tells if the structure can be modified."""
+        poll = self.poll
+        return poll.has_result()
+
 
 class NodePollResult(object):
 
diff --git a/webapp/polls/tests/structure_tests.py b/webapp/polls/tests/structure_tests.py
index 8ec1d21..ebb0f7e 100644
--- a/webapp/polls/tests/structure_tests.py
+++ b/webapp/polls/tests/structure_tests.py
@@ -324,6 +324,16 @@ class StructureTests(MongoTestCase):
     def test_it_should_respond_to_ready_only_msg(self):
         self.assertTrue(hasattr(Structure, 'READ_ONLY_MSG'))
 
+    def test_it_should_be_read_only_when_poll_has_results(self):
+        poll = self.poll
+        structure = Structure(data={}, poll=poll)
+        self.assertFalse(poll.has_result())
+        self.assertFalse(structure.is_read_only())
+
+        poll.has_result = Mock(return_value=True)
+        self.assertTrue(poll.has_result())
+        self.assertTrue(structure.is_read_only())
+
 
 class UploadOptionImagesTest(MongoTestCase):