Lib/test/test_mimetools.py

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