import unittest
from test import test_support

import string,StringIO, mimetools

class MimeToolsTest(unittest.TestCase):

    def test_decodeencode(self):
        start = string.ascii_letters + "=" + string.digits + "\n"
        for enc in ['7bit','8bit','base64','quoted-printable']:
            i = StringIO.StringIO(start)
            o = StringIO.StringIO()
            mimetools.encode(i, o, enc)
            i = StringIO.StringIO(o.getvalue())
            o = StringIO.StringIO()
            mimetools.decode(i, o, enc)
            self.assertEqual(o.getvalue(), start)

    def test_boundary(self):
        b = ""
        for i in xrange(10):
            nb = mimetools.choose_boundary()
            self.assertNotEqual(b, nb)
            b = nb

def test_main():
    test_support.run_unittest(MimeToolsTest)

if __name__=="__main__":
    test_main()