mirror of
https://github.com/Hopiu/django-imagekit.git
synced 2026-04-28 16:54:42 +00:00
79 lines
2.5 KiB
Python
79 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))
|