425f5553aee99d3ebfebd42e794cfa8e7b304d02
[wolnelektury.git] / apps / sorl / thumbnail / tests / fields.py
1 import os.path
2
3 from django.db import models
4 from django.conf import settings
5 from django.core.files.base import ContentFile
6
7 from sorl.thumbnail.fields import ImageWithThumbnailsField, ThumbnailField
8 from sorl.thumbnail.tests.base import BaseTest, RELATIVE_PIC_NAME, PIC_NAME
9
10 thumbnail = {
11     'size': (50, 50),
12 }
13 extra_thumbnails = {
14     'admin': {
15         'size': (30, 30),
16         'options': ('crop',),
17     }
18 }
19 extension_thumbnail = thumbnail.copy()
20 extension_thumbnail['extension'] = 'png'
21
22
23 # Temporary models for field_tests
24 class TestThumbnailFieldModel(models.Model):
25     avatar = ThumbnailField(upload_to='test', size=(300, 300))
26     photo = ImageWithThumbnailsField(upload_to='test', thumbnail=thumbnail,
27                                      extra_thumbnails=extra_thumbnails)
28
29
30 class TestThumbnailFieldExtensionModel(models.Model):
31     photo = ImageWithThumbnailsField(upload_to='test',
32                                      thumbnail=extension_thumbnail,
33                                      extra_thumbnails=extra_thumbnails)
34
35
36 class TestThumbnailFieldGenerateModel(models.Model):
37     photo = ImageWithThumbnailsField(upload_to='test', thumbnail=thumbnail,
38                                      extra_thumbnails=extra_thumbnails,
39                                      generate_on_save=True)
40
41
42 class FieldTest(BaseTest):
43     """
44     Test the base field functionality. These use an ImageWithThumbnailsField
45     but all the functionality tested is from BaseThumbnailField.
46     """
47     def test_extra_thumbnails(self):
48         model = TestThumbnailFieldModel(photo=RELATIVE_PIC_NAME)
49         self.assertTrue('admin' in model.photo.extra_thumbnails)
50         thumb = model.photo.extra_thumbnails['admin']
51         tag = model.photo.extra_thumbnails_tag['admin']
52         expected_filename = os.path.join(settings.MEDIA_ROOT,
53             'sorl-thumbnail-test_source_jpg_30x30_crop_q85.jpg')
54         self.verify_thumbnail((30, 30), thumb, expected_filename)
55         expected_tag = '<img src="%s" width="30" height="30" alt="" />' % \
56             '/'.join((settings.MEDIA_URL.rstrip('/'),
57                       'sorl-thumbnail-test_source_jpg_30x30_crop_q85.jpg'))
58         self.assertEqual(tag, expected_tag)
59
60     def test_extension(self):
61         model = TestThumbnailFieldExtensionModel(photo=RELATIVE_PIC_NAME)
62         thumb = model.photo.thumbnail
63         tag = model.photo.thumbnail_tag
64         expected_filename = os.path.join(settings.MEDIA_ROOT,
65             'sorl-thumbnail-test_source_jpg_50x50_q85.png')
66         self.verify_thumbnail((50, 37), thumb, expected_filename)
67         expected_tag = '<img src="%s" width="50" height="37" alt="" />' % \
68             '/'.join((settings.MEDIA_URL.rstrip('/'),
69                       'sorl-thumbnail-test_source_jpg_50x50_q85.png'))
70         self.assertEqual(tag, expected_tag)
71
72     def test_delete_thumbnails(self):
73         model = TestThumbnailFieldModel(photo=RELATIVE_PIC_NAME)
74         thumb_file = model.photo.thumbnail.dest
75         open(thumb_file, 'wb').close()
76         self.assert_(os.path.exists(thumb_file))
77         model.photo.delete(save=False)
78         self.assertFalse(os.path.exists(thumb_file))
79
80     def test_generate_on_save(self):
81         main_thumb = os.path.join(settings.MEDIA_ROOT, 'test',
82                         'sorl-thumbnail-test_source_jpg_50x50_q85.jpg')
83         admin_thumb = os.path.join(settings.MEDIA_ROOT, 'test',
84                         'sorl-thumbnail-test_source_jpg_30x30_crop_q85.jpg')
85         self.images_to_delete.add(main_thumb)
86         self.images_to_delete.add(admin_thumb)
87         # Default setting is to only generate when the thumbnail is used.
88         model = TestThumbnailFieldModel()
89         source = ContentFile(open(PIC_NAME).read())
90         model.photo.save(RELATIVE_PIC_NAME, source, save=False)
91         self.images_to_delete.add(model.photo.path)
92         self.assertFalse(os.path.exists(main_thumb))
93         self.assertFalse(os.path.exists(admin_thumb))
94         os.remove(model.photo.path)
95         # But it's easy to set it up the other way...
96         model = TestThumbnailFieldGenerateModel()
97         source = ContentFile(open(PIC_NAME).read())
98         model.photo.save(RELATIVE_PIC_NAME, source, save=False)
99         self.assert_(os.path.exists(main_thumb))
100         self.assert_(os.path.exists(admin_thumb))
101
102
103 class ImageWithThumbnailsFieldTest(BaseTest):
104     def test_thumbnail(self):
105         model = TestThumbnailFieldModel(photo=RELATIVE_PIC_NAME)
106         thumb = model.photo.thumbnail
107         tag = model.photo.thumbnail_tag
108         base_name = RELATIVE_PIC_NAME.replace('.', '_')
109         expected_filename = os.path.join(settings.MEDIA_ROOT,
110                                          '%s_50x50_q85.jpg' % base_name)
111         self.verify_thumbnail((50, 37), thumb, expected_filename)
112         expected_tag = ('<img src="%s" width="50" height="37" alt="" />' %
113                         '/'.join([settings.MEDIA_URL.rstrip('/'),
114                                   '%s_50x50_q85.jpg' % base_name]))
115         self.assertEqual(tag, expected_tag)
116
117
118 class ThumbnailFieldTest(BaseTest):
119     def test_thumbnail(self):
120         model = TestThumbnailFieldModel()
121         source = ContentFile(open(PIC_NAME).read())
122         dest_name = 'sorl-thumbnail-test_dest.jpg'
123         model.avatar.save(dest_name, source, save=False)
124         expected_filename = os.path.join(model.avatar.path)
125         self.verify_thumbnail((300, 225), expected_filename=expected_filename)
126
127         tag = model.avatar.thumbnail_tag
128         expected_tag = ('<img src="%s" width="300" height="225" alt="" />' %
129                         '/'.join([settings.MEDIA_URL.rstrip('/'), 'test',
130                                   dest_name]))
131         self.assertEqual(tag, expected_tag)