14710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm# Copyright (C) 2002-2006 Python Software Foundation 24710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm# Contact: email-sig@python.org 34710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm# email package unit tests for (optional) Asian codecs 44710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 54710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylmimport unittest 64710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylmfrom test.test_support import run_unittest 74710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 84710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylmfrom email.test.test_email import TestEmailBase 94710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylmfrom email.charset import Charset 104710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylmfrom email.header import Header, decode_header 114710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylmfrom email.message import Message 124710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 134710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm# We're compatible with Python 2.3, but it doesn't have the built-in Asian 144710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm# codecs, so we have to skip all these tests. 154710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylmtry: 164710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm unicode('foo', 'euc-jp') 174710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylmexcept LookupError: 184710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm raise unittest.SkipTest 194710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 204710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 214710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 224710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylmclass TestEmailAsianCodecs(TestEmailBase): 234710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm def test_japanese_codecs(self): 244710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm eq = self.ndiffAssertEqual 254710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm j = Charset("euc-jp") 264710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm g = Charset("iso-8859-1") 274710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm h = Header("Hello World!") 284710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm jhello = '\xa5\xcf\xa5\xed\xa1\xbc\xa5\xef\xa1\xbc\xa5\xeb\xa5\xc9\xa1\xaa' 294710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm ghello = 'Gr\xfc\xdf Gott!' 304710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm h.append(jhello, j) 314710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm h.append(ghello, g) 324710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # BAW: This used to -- and maybe should -- fold the two iso-8859-1 334710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # chunks into a single encoded word. However it doesn't violate the 344710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # standard to have them as two encoded chunks and maybe it's 354710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # reasonable <wink> for each .append() call to result in a separate 364710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # encoded word. 374710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm eq(h.encode(), """\ 384710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylmHello World! =?iso-2022-jp?b?GyRCJU8lbSE8JW8hPCVrJUkhKhsoQg==?= 394710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm =?iso-8859-1?q?Gr=FC=DF?= =?iso-8859-1?q?_Gott!?=""") 404710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm eq(decode_header(h.encode()), 414710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm [('Hello World!', None), 424710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm ('\x1b$B%O%m!<%o!<%k%I!*\x1b(B', 'iso-2022-jp'), 434710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm ('Gr\xfc\xdf Gott!', 'iso-8859-1')]) 444710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm long = 'test-ja \xa4\xd8\xc5\xea\xb9\xc6\xa4\xb5\xa4\xec\xa4\xbf\xa5\xe1\xa1\xbc\xa5\xeb\xa4\xcf\xbb\xca\xb2\xf1\xbc\xd4\xa4\xce\xbe\xb5\xc7\xa7\xa4\xf2\xc2\xd4\xa4\xc3\xa4\xc6\xa4\xa4\xa4\xde\xa4\xb9' 454710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm h = Header(long, j, header_name="Subject") 464710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # test a very long header 474710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm enc = h.encode() 484710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # TK: splitting point may differ by codec design and/or Header encoding 494710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm eq(enc , """\ 504710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm=?iso-2022-jp?b?dGVzdC1qYSAbJEIkWEVqOUYkNSRsJD8lYSE8JWskTztKGyhC?= 514710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm =?iso-2022-jp?b?GyRCMnE8VCROPjVHJyRyQlQkQyRGJCQkXiQ5GyhC?=""") 524710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm # TK: full decode comparison 534710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm eq(h.__unicode__().encode('euc-jp'), long) 544710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 554710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm def test_payload_encoding(self): 564710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm jhello = '\xa5\xcf\xa5\xed\xa1\xbc\xa5\xef\xa1\xbc\xa5\xeb\xa5\xc9\xa1\xaa' 574710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm jcode = 'euc-jp' 584710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm msg = Message() 594710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm msg.set_payload(jhello, jcode) 604710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm ustr = unicode(msg.get_payload(), msg.get_content_charset()) 614710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm self.assertEqual(jhello, ustr.encode(jcode)) 624710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 634710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 644710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 654710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylmdef suite(): 664710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm suite = unittest.TestSuite() 674710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm suite.addTest(unittest.makeSuite(TestEmailAsianCodecs)) 684710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm return suite 694710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 704710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 714710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylmdef test_main(): 724710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm run_unittest(TestEmailAsianCodecs) 734710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 744710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 754710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm 764710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylmif __name__ == '__main__': 774710c53dcad1ebf3755f3efb9e80ac24bd72a9b2darylm unittest.main(defaultTest='suite') 78