middleware: check if content-type is defined

closes #98
cobrateam · Mar 21, 2016 · 3aa4916 · 3aa4916
1 parent 3cb2e19
commit 3aa4916
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/htmlmin/middleware.py b/htmlmin/middleware.py
@@ -29,7 +29,7 @@ def can_minify_response(self, request, response):
                     req_ok = False
                     break
 
-        resp_ok = 'text/html' in response['Content-Type']
+        resp_ok = 'text/html' in response.get('Content-Type', '')
         if hasattr(response, 'minify_response'):
             resp_ok = resp_ok and response.minify_response
         return req_ok and resp_ok

diff --git a/htmlmin/tests/test_middleware.py b/htmlmin/tests/test_middleware.py
@@ -44,6 +44,16 @@ def test_should_minify_with_any_charset(self):
         minified = "<html><head></head><body>some text here</body></html>"
         self.assertEqual(minified, response.content)
 
+    def test_should_not_minify_without_content(self):
+        response_mock = ResponseMock()
+        del response_mock['Content-Type']
+        response = HtmlMinifyMiddleware().process_response(
+            RequestMock(), response_mock,
+        )
+
+        html_not_minified = "<html>   <body>some text here</body>    </html>"
+        self.assertEqual(html_not_minified, response.content)
+
     def test_should_not_minify_not_html_content(self):
         response_mock = ResponseMock()
         response_mock['Content-Type'] = 'application/json'