class ShouldAdaptTests(unittest.TestCase): # tests for ImageAdaptingMiddleware.should_adapt def setUp(self): self.middleware = ImageAdaptingMiddleware(None, {}) def test_no_content_type(self): # docs w/o content type should not be adapted request = Request.blank("http://localhost/test.html") response = request.get_response(wsgi_app_no_content_type) self.assertEqual(self.middleware.should_adapt(response), False) return def test_jpeg_content_type(self): # image/jpeg docs are adapted request = Request.blank("http://localhost/test.html") response = request.get_response(wsgi_app_img_jpg) self.assertEqual(self.middleware.should_adapt(response), True) return def test_gif_content_type(self): # image/gif docs are adapted request = Request.blank("http://localhost/test.html") response = request.get_response(wsgi_app_img_gif) self.assertEqual(self.middleware.should_adapt(response), True) return def test_png_content_type(self): # image/png docs are adapted request = Request.blank("http://localhost/test.html") response = request.get_response(wsgi_app_img_png) self.assertEqual(self.middleware.should_adapt(response), True) return