From 1d95cf991130df7f4d277ed2f840a397d512146d Mon Sep 17 00:00:00 2001 From: Lucas Hoffmann Date: Thu, 11 Oct 2018 01:22:42 +0200 Subject: Add test for #1301 This adds a test for detecting a malformed content-transfer-encoding (trailing semi-colon). It also changes the raised exception to the more appropriate ValueError. --- tests/db/utils_test.py | 7 +++++++ 1 file changed, 7 insertions(+) (limited to 'tests/db') diff --git a/tests/db/utils_test.py b/tests/db/utils_test.py index 9fd49d76..fe4a1705 100644 --- a/tests/db/utils_test.py +++ b/tests/db/utils_test.py @@ -741,3 +741,10 @@ class TestRemoveCte(unittest.TestCase): # This should not raise an UnicodeDecodeError. utils.remove_cte(mail, as_string=True) self.assertTrue(True) + + def test_malformed_cte_value(self): + with open('tests/static/mail/malformed-header-CTE.eml') as fp: + mail = email.message_from_file(fp) + with self.assertRaisesRegex(ValueError, + "Unknown Content-Transfer-Encoding"): + utils.remove_cte(mail, as_string=True) -- cgit v1.2.3