mirror of
https://github.com/Hopiu/django-imagekit.git
synced 2026-03-17 05:40:25 +00:00
f9b79aa67f
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.
78 lines
2.5 KiB
Python
78 lines
2.5 KiB
Python
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))
|