Added a test for properly auto-determining mime types and setting con…

…tent encoding header
miguelgrinberg · Aug 13, 2024 · 58dd6c0 · 58dd6c0
1 parent 81caeb0
commit 58dd6c0
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/tests/files/test.txt.gz b/tests/files/test.txt.gz
@@ -0,0 +1 @@
+foo
diff --git a/tests/test_response.py b/tests/test_response.py
@@ -280,6 +280,16 @@ def test_send_file_compressed(self):
                          'application/octet-stream')
         self.assertEqual(res.headers['Content-Encoding'], 'gzip')
 
+    def test_send_file_with_correct_mime_type_and_content_encoding(self):
+        res = Response.send_file('tests/files/test.txt')
+        self.assertEqual(res.status_code, 200)
+        self.assertEqual(res.headers['Content-Type'], 'text/plain')
+
+        res = Response.send_file('tests/files/test.txt.gz')
+        self.assertEqual(res.status_code, 200)
+        self.assertEqual(res.headers['Content-Type'], 'text/plain')
+        self.assertEqual(res.headers['Content-Encoding'], 'gzip')
+
     def test_default_content_type(self):
         original_content_type = Response.default_content_type
         res = Response('foo')