1import unittest 2from test import test_support 3 4import string 5import StringIO 6 7mimetools = test_support.import_module("mimetools", deprecated=True) 8 9msgtext1 = mimetools.Message(StringIO.StringIO( 10"""Content-Type: text/plain; charset=iso-8859-1; format=flowed 11Content-Transfer-Encoding: 8bit 12 13Foo! 14""")) 15 16class MimeToolsTest(unittest.TestCase): 17 18 def test_decodeencode(self): 19 start = string.ascii_letters + "=" + string.digits + "\n" 20 for enc in ['7bit','8bit','base64','quoted-printable', 21 'uuencode', 'x-uuencode', 'uue', 'x-uue']: 22 i = StringIO.StringIO(start) 23 o = StringIO.StringIO() 24 mimetools.encode(i, o, enc) 25 i = StringIO.StringIO(o.getvalue()) 26 o = StringIO.StringIO() 27 mimetools.decode(i, o, enc) 28 self.assertEqual(o.getvalue(), start) 29 30 def test_boundary(self): 31 s = set([""]) 32 for i in xrange(100): 33 nb = mimetools.choose_boundary() 34 self.assertNotIn(nb, s) 35 s.add(nb) 36 37 def test_message(self): 38 msg = mimetools.Message(StringIO.StringIO(msgtext1)) 39 self.assertEqual(msg.gettype(), "text/plain") 40 self.assertEqual(msg.getmaintype(), "text") 41 self.assertEqual(msg.getsubtype(), "plain") 42 self.assertEqual(msg.getplist(), ["charset=iso-8859-1", "format=flowed"]) 43 self.assertEqual(msg.getparamnames(), ["charset", "format"]) 44 self.assertEqual(msg.getparam("charset"), "iso-8859-1") 45 self.assertEqual(msg.getparam("format"), "flowed") 46 self.assertEqual(msg.getparam("spam"), None) 47 self.assertEqual(msg.getencoding(), "8bit") 48 49def test_main(): 50 test_support.run_unittest(MimeToolsTest) 51 52if __name__=="__main__": 53 test_main() 54