Added tests for SF patch #597593, syntactically invalid Content-Type: headers.

warsaw · warsaw · commit 3328136e3ccc · 2002-08-20T14:51:10.000Z
diff --git a/Lib/email/test/test_email.py b/Lib/email/test/test_email.py
@@ -428,12 +428,12 @@ def test_get_content_subtype_from_message_text_plain_explicit(self):
     def test_get_content_maintype_error(self):
         msg = Message()
         msg['Content-Type'] = 'no-slash-in-this-string'
-        self.assertRaises(ValueError, msg.get_content_maintype)
+        self.assertEqual(msg.get_content_maintype(), 'text')
 
     def test_get_content_subtype_error(self):
         msg = Message()
         msg['Content-Type'] = 'no-slash-in-this-string'
-        self.assertRaises(ValueError, msg.get_content_subtype)
+        self.assertEqual(msg.get_content_subtype(), 'plain')
 
 
 
@@ -1007,6 +1007,27 @@ def test_multipart_no_boundary(self):
         finally:
             fp.close()
 
+    def test_invalid_content_type(self):
+        eq = self.assertEqual
+        neq = self.ndiffAssertEqual
+        msg = Message()
+        # RFC 2045, $5.2 says invalid yields text/plain
+        msg['Content-Type'] = 'text'
+        eq(msg.get_content_maintype(), 'text')
+        eq(msg.get_content_subtype(), 'plain')
+        eq(msg.get_content_type(), 'text/plain')
+        # Clear the old value and try something /really/ invalid
+        del msg['content-type']
+        msg['Content-Type'] = 'foo'
+        eq(msg.get_content_maintype(), 'text')
+        eq(msg.get_content_subtype(), 'plain')
+        eq(msg.get_content_type(), 'text/plain')
+        # Still, make sure that the message is idempotently generated
+        s = StringIO()
+        g = Generator(s)
+        g.flatten(msg)
+        neq(s.getvalue(), 'Content-Type: foo\n\n')
+
 
 
 # Test RFC 2047 header encoding and decoding
