From 1137487689b23439c51318f8594c8c221a11dbb7 Mon Sep 17 00:00:00 2001 From: Matthew Tretter Date: Fri, 2 Dec 2011 02:19:12 -0500 Subject: [PATCH] Separate create_photo method --- tests/core/tests.py | 16 ++++++++++------ 1 file changed, 10 insertions(+), 6 deletions(-) diff --git a/tests/core/tests.py b/tests/core/tests.py index 3988dc4..8e91530 100644 --- a/tests/core/tests.py +++ b/tests/core/tests.py @@ -45,16 +45,20 @@ class IKTest(TestCase): Image.open(lennafile).save(tmp, 'JPEG') tmp.seek(0) return tmp - - def setUp(self): - self.photo = Photo() + + def create_photo(self, name): + photo = Photo() img = self.generate_lenna() file = ContentFile(img.read()) - self.photo.original_image = file - self.photo.original_image.save('test.jpeg', file) - self.photo.save() + photo.original_image = file + photo.original_image.save(name, file) + photo.save() img.close() + return photo + def setUp(self): + self.photo = self.create_photo('test.jpg') + def test_save_image(self): photo = Photo.objects.get(id=self.photo.id) self.assertTrue(os.path.isfile(photo.original_image.path))