def test_extensionless_gif(self): """If the image file is a GIF without an extension, we can produce a valid thumbnail for it.""" # Note: this test image file was breaking thumbnails if lazythumbs didn't # see the .gif extension. I tried creating a gif on the fly using # PIL but didn't hit the same problem, so it might be something about # this image that's special, maybe that it has a transparent background. # (The error was "cannot write mode P as JPEG"; the symptom was a 404 # response.) MEDIA_ROOT = tempfile.gettempdir() # Need to override MEDIA_ROOT in both django file storage and lazythumbs views # and Django doesn't provide override_settings until 1.4 with patch('django.core.files.storage.settings') as settings1: settings1.MEDIA_ROOT = MEDIA_ROOT with patch('lazythumbs.views.settings') as settings2: settings2.MEDIA_ROOT = MEDIA_ROOT testfile = TEST_IMG_GIF filename = None try: filename = os.path.join(MEDIA_ROOT, "gif_without_extension") shutil.copy(testfile, filename) # Now we have a gif file in a filename that doesn't end in .gif renderer = LazyThumbRenderer() source_path = os.path.relpath(filename, MEDIA_ROOT) rsp = renderer.get( request=Mock(path="/thumbnail/x50/" + source_path), action="thumbnail", geometry="x50", source_path=source_path ) # if you get 404, jpeg encoder is probably missing for Pillow self.assertEqual(200, rsp.status_code) finally: if filename: os.remove(filename)
class GetViewTest(TestCase): """ Test behavior of LazyThumbRenderer.get """ def mc_factory(self, was_404): """ churn out mocked caches with a preset .get(). Also a rapper? """ mc = Mock() ret = was_404 mc.get = Mock(return_value=ret) mc.set = Mock() return mc def setUp(self): self.renderer = LazyThumbRenderer() self.mock_Image = Mock() self.mock_img = Mock() self.mock_Image.open = Mock(return_value=self.mock_img) self.mock_img.size = [1, 1] def test_img_404_warm_cache(self): """ Ensure we go straight to a 404 response without setting anything new in cache or touching filesystem if we encounter a cached 404. """ req = Mock() req.path = "/lt_cache/thumbnail/48/i/p.jpg" self.renderer._render_and_save = Mock() with patch('lazythumbs.views.cache', self.mc_factory(1)) as mc: resp = self.renderer.get(req, 'thumbnail', '48', 'i/p') self.assertEqual(resp.status_code, 404) self.assertEqual(resp['Content-Type'], 'image/jpeg') self.assertTrue('Cache-Control' in resp) self.assertFalse(mc.set.called) def test_img_404_cold_cache(self): """ Basic 404: requested image is not found. Make sure proper response headers are set and the 404 was cached. """ req = Mock() req.path = "/lt_cache/thumbnail/48/i/p.jpg" with patch('lazythumbs.views.cache', MockCache()) as mc: resp = self.renderer.get(req, 'thumbnail', '48', 'i/p') self.assertEqual(resp.status_code, 404) self.assertEqual(resp['Content-Type'], 'image/jpeg') self.assertTrue('Cache-Control' in resp) self.assertEqual(len(mc.cache.keys()), 1) key = mc.cache.keys()[0] cached = mc.cache[key] self.assertEqual(cached, 1) def test_img_200_cold_cache(self): """ Pretend we found the requested rendered image on the filesystem. Ensure proper response headers are set and the rendered path was cached. """ req = Mock() req.path = "/lt_cache/thumbnail/48/i/p.jpg" self.renderer.fs.save = Mock() with patch('lazythumbs.views.Image', self.mock_Image): with patch('lazythumbs.views.cache', MockCache()) as mc: resp = self.renderer.get(req, 'thumbnail', '48', 'i/p') self.assertEqual(resp.status_code, 200) self.assertEqual(resp['Content-Type'], 'image/jpeg') self.assertTrue('Cache-Control' in resp) self.assertEqual(resp.content, '') # empty buffer means raw_data is '' self.assertEqual(len(mc.cache.keys()), 1) key = mc.cache.keys()[0] cached = mc.cache[key] self.assertEqual(cached, False)
class GetViewTest(TestCase): """ Test behavior of LazyThumbRenderer.get """ def mc_factory(self, was_404): """ churn out mocked caches with a preset .get(). Also a rapper? """ mc = Mock() ret = was_404 mc.get = Mock(return_value=ret) mc.set = Mock() return mc def setUp(self): self.renderer = LazyThumbRenderer() self.mock_Image = Mock() self.mock_img = Mock() self.mock_Image.open = Mock(return_value=self.mock_img) self.mock_img.size = [1,1] def test_img_404_warm_cache(self): """ Ensure we go straight to a 404 response without setting anything new in cache or touching filesystem if we encounter a cached 404. """ req = Mock() req.path = "/lt_cache/thumbnail/48/i/p.jpg" self.renderer._render_and_save = Mock() with patch('lazythumbs.views.cache', self.mc_factory(1)) as mc: resp = self.renderer.get(req, 'thumbnail', '48', 'i/p') self.assertEqual(resp.status_code, 404) self.assertEqual(resp['Content-Type'], 'image/jpeg') self.assertTrue('Cache-Control' in resp) self.assertFalse(mc.set.called) def test_img_404_cold_cache(self): """ Basic 404: requested image is not found. Make sure proper response headers are set and the 404 was cached. """ req = Mock() req.path = "/lt_cache/thumbnail/48/i/p.jpg" with patch('lazythumbs.views.cache', MockCache()) as mc: resp = self.renderer.get(req, 'thumbnail', '48', 'i/p') self.assertEqual(resp.status_code, 404) self.assertEqual(resp['Content-Type'], 'image/jpeg') self.assertTrue('Cache-Control' in resp) self.assertEqual(len(mc.cache.keys()), 1) key = mc.cache.keys()[0] cached = mc.cache[key] self.assertEqual(cached, 1) def test_img_200_cold_cache(self): """ Pretend we found the requested rendered image on the filesystem. Ensure proper response headers are set and the rendered path was cached. """ req = Mock() req.path = "/lt_cache/thumbnail/48/i/p.jpg" self.renderer.fs.save = Mock() with patch('lazythumbs.views.Image', self.mock_Image): with patch('lazythumbs.views.cache', MockCache()) as mc: resp = self.renderer.get(req, 'thumbnail', '48', 'i/p') self.assertEqual(resp.status_code, 200) self.assertEqual(resp['Content-Type'], 'image/jpeg') self.assertTrue('Cache-Control' in resp) self.assertEqual(resp.content, '') # empty buffer means raw_data is '' self.assertEqual(len(mc.cache.keys()), 1) key = mc.cache.keys()[0] cached = mc.cache[key] self.assertEqual(cached, False) def test_no_img_should_404(self): """ When save fails with EEXIST error, it will try to read the file again But if it still can't be read, make sure it returns a 404 instead of 0-byte image. """ req = Mock() req.path = "/lt_cache/thumbnail/48/i/p.jpg" self.renderer.fs.save = Mock() err = OSError() err.errno = errno.EEXIST self.renderer.fs.save.side_effect = err with patch('lazythumbs.views.Image', self.mock_Image): with patch('lazythumbs.views.cache', MockCache()): resp = self.renderer.get(req, 'thumbnail', '48', 'i/p') self.assertEqual(resp.status_code, 404) def test_naughty_paths_root(self): resp = self.renderer.get(None, 'thumbnail', '48', '/') self.assertEqual(resp.status_code, 404) def test_naughty_paths_traverse(self): resp = self.renderer.get(None, 'thumbnail', '48', '../') self.assertEqual(resp.status_code, 404) def test_invalid_action(self): resp = self.renderer.get(None, 'invalid_action', '48', 'i/p') self.assertEqual(resp.status_code, 404) def test_invalid_geometry(self): resp = self.renderer.get(None, 'thumbnail', '48x48x48', 'i/p') self.assertEqual(resp.status_code, 404)
class GetViewTest(TestCase): """ Test behavior of LazyThumbRenderer.get """ def mc_factory(self, was_404): """ churn out mocked caches with a preset .get(). Also a rapper? """ mc = Mock() ret = was_404 mc.get = Mock(return_value=ret) mc.set = Mock() return mc def setUp(self): self.renderer = LazyThumbRenderer() self.mock_Image = Mock() self.mock_img = Mock() self.mock_Image.open = Mock(return_value=self.mock_img) self.mock_img.size = [1,1] def test_img_404_warm_cache(self): """ Ensure we go straight to a 404 response without setting anything new in cache or touching filesystem if we encounter a cached 404. """ req = Mock() req.path = "/lt_cache/thumbnail/48/i/p.jpg" self.renderer._render_and_save = Mock() with patch('lazythumbs.views.cache', self.mc_factory(1)) as mc: resp = self.renderer.get(req, 'thumbnail', '48', 'i/p') self.assertEqual(resp.status_code, 404) self.assertEqual(resp['Content-Type'], 'image/jpeg') self.assertTrue('Cache-Control' in resp) self.assertFalse(mc.set.called) def test_img_404_cold_cache(self): """ Basic 404: requested image is not found. Make sure proper response headers are set and the 404 was cached. """ req = Mock() req.path = "/lt_cache/thumbnail/48/i/p.jpg" with patch('lazythumbs.views.cache', MockCache()) as mc: resp = self.renderer.get(req, 'thumbnail', '48', 'i/p') self.assertEqual(resp.status_code, 404) self.assertEqual(resp['Content-Type'], 'image/jpeg') self.assertTrue('Cache-Control' in resp) self.assertEqual(len(mc.cache.keys()), 1) key = mc.cache.keys()[0] cached = mc.cache[key] self.assertEqual(cached, 1) def test_img_200_cold_cache(self): """ Pretend we found the requested rendered image on the filesystem. Ensure proper response headers are set and the rendered path was cached. """ req = Mock() req.path = "/lt_cache/thumbnail/48/i/p.jpg" self.renderer.fs.save = Mock() with patch('lazythumbs.views.Image', self.mock_Image): with patch('lazythumbs.views.cache', MockCache()) as mc: resp = self.renderer.get(req, 'thumbnail', '48', 'i/p') self.assertEqual(resp.status_code, 200) self.assertEqual(resp['Content-Type'], 'image/jpeg') self.assertTrue('Cache-Control' in resp) self.assertEqual(resp.content, '') # empty buffer means raw_data is '' self.assertEqual(len(mc.cache.keys()), 1) key = mc.cache.keys()[0] cached = mc.cache[key] self.assertEqual(cached, False)