easy_thumbnails.test: 63 total statements, 0.0% covered

Generated: Wed 2013-03-13 10:33 CET

Source file: /media/Envs/Envs/filer-gallery/lib/python2.7/site-packages/easy_thumbnails/test.py

Stats: 0 executed, 53 missed, 10 excluded, 61 ignored

1. import shutil
2. import tempfile
3. try:
4. from cStringIO import StringIO
5. except ImportError:
6. from StringIO import StringIO
8. from django.core.files.base import ContentFile
9. from django.core.files.storage import FileSystemStorage
10. from django.test import TestCase
11. try:
12. from PIL import Image
13. except ImportError:
14. import Image
16. from easy_thumbnails.conf import settings
19. class TemporaryStorage(FileSystemStorage):
20. """
21. A storage class useful for tests that uses a temporary location to store
22. all files and provides a method to remove this location when it is finished
23. with.
24. """
26. def __init__(self, location=None, *args, **kwargs):
27. """
28. Create the temporary location.
29. """
30. if location is None:
31. location = tempfile.mkdtemp()
32. self.temporary_location = location
33. super(TemporaryStorage, self).__init__(location=location, *args,
34. **kwargs)
36. def delete_temporary_storage(self):
37. """
38. Delete the temporary directory created during initialisation.
39. This storage class should not be used again after this method is
40. called.
41. """
42. temporary_location = getattr(self, 'temporary_location', None)
43. if temporary_location:
44. shutil.rmtree(temporary_location)
47. class FakeRemoteStorage(TemporaryStorage):
48. """
49. A temporary storage class that acts similar to remote storage.
51. It's not thread safe.
52. """
53. remote_mode = False
55. def path(self, *args, **kwargs):
56. """
57. Raise ``NotImplementedError``, since this is the way that
58. easy-thumbnails determines if a storage is remote.
59. """
60. if self.remote_mode:
61. raise NotImplementedError
62. return super(FakeRemoteStorage, self).path(*args, **kwargs)
64. def exists(self, *args, **kwargs):
65. original_remote_mode = self.remote_mode
66. self.remote_mode = False
67. try:
68. return super(FakeRemoteStorage, self).exists(*args, **kwargs)
69. finally:
70. self.remote_mode = original_remote_mode
72. def save(self, *args, **kwargs):
73. self.remote_mode = False
74. try:
75. return super(FakeRemoteStorage, self).save(*args, **kwargs)
76. finally:
77. self.remote_mode = True
79. def open(self, *args, **kwargs):
80. self.remote_mode = False
81. try:
82. return super(FakeRemoteStorage, self).open(*args, **kwargs)
83. finally:
84. self.remote_mode = True
87. class BaseTest(TestCase):
88. """
89. Remove any customised THUMBNAIL_* settings in a project's ``settings``
90. configuration module before running the tests to ensure there is a
91. consistent test environment.
92. """
94. def setUp(self):
95. """
96. Isolate all settings.
97. """
98. output = super(BaseTest, self).setUp()
99. settings.isolated = True
100. return output
102. def tearDown(self):
103. """
104. Restore settings to their original state.
105. """
106. settings.isolated = False
107. settings.revert()
108. return super(BaseTest, self).tearDown()
110. def create_image(self, storage, filename, size=(800, 600),
111. image_mode='RGB', image_format='JPEG'):
112. """
113. Generate a test image, returning the filename that it was saved as.
115. If ``storage`` is ``None``, the StringIO containing the image data
116. will be passed instead.
117. """
118. data = StringIO()
119. Image.new(image_mode, size).save(data, image_format)
120. data.seek(0)
121. if not storage:
122. return data
123. image_file = ContentFile(data.read())
124. return storage.save(filename, image_file)