diff options
-rw-r--r-- | Lib/test/test_email.py | 18 |
1 files changed, 17 insertions, 1 deletions
diff --git a/Lib/test/test_email.py b/Lib/test/test_email.py index e5ddebb..3134a88 100644 --- a/Lib/test/test_email.py +++ b/Lib/test/test_email.py @@ -10,7 +10,7 @@ from types import StringType import email -from email.Parser import Parser +from email.Parser import Parser, HeaderParser from email.Generator import Generator, DecodedGenerator from email.Message import Message from email.MIMEAudio import MIMEAudio @@ -888,6 +888,21 @@ to reflect upon our own +class TestParsers(unittest.TestCase): + def test_header_parser(self): + eq = self.assertEqual + # Parse only the headers of a complex multipart MIME document + p = HeaderParser() + fp = openfile('msg_02.txt') + msg = p.parse(fp) + eq(msg['from'], 'ppp-request@zzz.org') + eq(msg['to'], 'ppp@zzz.org') + eq(msg.get_type(), 'multipart/mixed') + eq(msg.is_multipart(), 0) + self.failUnless(isinstance(msg.get_payload(), StringType)) + + + def suite(): suite = unittest.TestSuite() suite.addTest(unittest.makeSuite(TestMessageAPI)) @@ -904,6 +919,7 @@ def suite(): suite.addTest(unittest.makeSuite(TestIdempotent)) suite.addTest(unittest.makeSuite(TestMiscellaneous)) suite.addTest(unittest.makeSuite(TestIterators)) + suite.addTest(unittest.makeSuite(TestParsers)) return suite |