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))