From f9b79aa67f6b05b07e1db693a5d654a3a6894c4e Mon Sep 17 00:00:00 2001
From: Bryan Veloso <bryan@revyver.com>
Date: Fri, 9 Sep 2011 17:57:18 -0700
Subject: [PATCH] Tests now pass and don't leave any directory cruft. Fixes
 #28.

Tests were failing because they assumed that the file was deleted upon
teardown. This isn't the case after Django 1.3, where this no longer
happens. This fix removes any orphaned files (and directories) that were
created via the testing process.
---
 tests/core/tests.py | 78 +++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 78 insertions(+)
 create mode 100644 tests/core/tests.py

diff --git a/tests/core/tests.py b/tests/core/tests.py
new file mode 100644
index 0000000..e96af80
--- /dev/null
+++ b/tests/core/tests.py
@@ -0,0 +1,78 @@
+import os
+import tempfile
+import unittest
+
+from django.conf import settings
+from django.core.files.base import ContentFile
+from django.test import TestCase
+
+from imagekit.lib import Image
+
+from core.models import TestPhoto
+
+
+class IKTest(TestCase):
+    """ Base TestCase class. """
+    def generate_image(self):
+        tmp = tempfile.TemporaryFile()
+        Image.new('RGB', (800, 600)).save(tmp, 'JPEG')
+        tmp.seek(0)
+        return tmp
+
+    def setUp(self):
+        self.p = TestPhoto()
+        img = self.generate_image()
+        self.p.save_image('test.jpeg', ContentFile(img.read()))
+        self.p.save()
+        img.close()
+
+    def test_save_image(self):
+        img = self.generate_image()
+        path = self.p.image.path
+        self.p.save_image('test2.jpeg', ContentFile(img.read()))
+        self.failIf(os.path.isfile(path))
+        path = self.p.image.path
+        img.seek(0)
+        self.p.save_image('test.jpeg', ContentFile(img.read()))
+        self.failIf(os.path.isfile(path))
+        img.close()
+
+    def test_setup(self):
+        self.assertEqual(self.p.image.width, 800)
+        self.assertEqual(self.p.image.height, 600)
+
+    def test_to_width(self):
+        self.assertEqual(self.p.to_width.width, 100)
+        self.assertEqual(self.p.to_width.height, 75)
+
+    def test_to_height(self):
+        self.assertEqual(self.p.to_height.width, 133)
+        self.assertEqual(self.p.to_height.height, 100)
+
+    def test_crop(self):
+        self.assertEqual(self.p.cropped.width, 100)
+        self.assertEqual(self.p.cropped.height, 100)
+
+    def test_url(self):
+        tup = (settings.MEDIA_URL, self.p._ik.cache_dir,
+               'images/test_to_width.jpeg')
+        self.assertEqual(self.p.to_width.url, "%s%s/%s" % tup)
+
+    def tearDown(self):
+        # ImageKit doesn't delete processed files unless you clear the cache.
+        # We also attempt to remove the cache directory as to not clutter up
+        # your filesystem.
+        self.p._clear_cache()
+        try:
+            os.removedirs(os.path.join(settings.MEDIA_ROOT, self.p._ik.cache_dir, 'images'))
+        except OSError:
+            pass
+
+        # As of Django 1.3, FileFields no longer delete the underlying image
+        # when you delete the model. For the sanity of these tests, we have
+        # to do this ourselves.
+        path = self.p.image.path
+        os.remove(os.path.join(settings.MEDIA_ROOT, path))
+        os.removedirs(os.path.join(settings.MEDIA_ROOT, 'images'))
+        self.p.delete()
+        self.failIf(os.path.isfile(path))