Exemplo n.º 1
0
class ImageTest(TestCase):

    def setUp(self):
        self.gallery = GalleryFactory()
        self.image = ImageFactory(gallery=self.gallery)

    def tearDown(self):
        self.image.delete()
        self.gallery.delete()

    def test_unicode(self):
        self.assertEqual(self.image.slug, self.image.__unicode__())

    def test_get_absolute_url(self):
        absolute_url = u'{0}/{1}'.format(settings.MEDIA_URL, self.image.physical)

        self.assertEqual(absolute_url, self.image.get_absolute_url())

    def test_get_extension(self):
        self.assertEqual('jpg', self.image.get_extension())

    def test_save_image(self):
        test_image = ImageFactory(gallery=self.gallery)
        self.assertTrue(os.path.isfile(test_image.physical.path))

        test_image.delete()
        self.assertFalse(os.path.isfile(test_image.physical.path))