1 import unittest

     2 from test import test_support

     3 

     4 import string, StringIO, mimetools

     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 = 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()