tearDown to remove option images

1 files changed, 5 insertions, 1 deletions

diff --git a/webapp/polls/tests/structure_tests.py b/webapp/polls/tests/structure_tests.py
index abb7bd8..f11da61 100644
--- a/webapp/polls/tests/structure_tests.py
+++ b/webapp/polls/tests/structure_tests.py
@@ -4,7 +4,8 @@ import os
 from polls.models import Poll, Group, Field, Structure
 from mock import Mock
 
-from utils.test import MongoTestCase, mock_in_memory_image
+from utils.test import (MongoTestCase, mock_in_memory_image,
+                        remove_option_images_dir)
 from polls.tests.poll_tests import MockImageFileInterface
 
 
@@ -361,3 +362,6 @@ class UploadOptionImagesTest(MongoTestCase):
         images_dir = structure.get_image_options_path()
         saved_on_disk = os.path.exists(os.path.join(images_dir, img_filename))
         self.assertTrue(saved_on_disk)
+
+    def tearDown(self):
+        remove_option_images_dir()