home *** CD-ROM | disk | FTP | other *** search
/ PC World 2005 June / PCWorld_2005-06_cd.bin / software / vyzkuste / firewally / firewally.exe / framework-2.3.exe / test_mimetools.py < prev    next >
Encoding:
Python Source  |  2003-12-30  |  1.7 KB  |  51 lines
  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()
  51.