diff options
Diffstat (limited to 'test/test_umessage.py')
| -rw-r--r-- | test/test_umessage.py | 39 |
1 files changed, 20 insertions, 19 deletions
diff --git a/test/test_umessage.py b/test/test_umessage.py index e70e386..7ae13e6 100644 --- a/test/test_umessage.py +++ b/test/test_umessage.py @@ -23,35 +23,36 @@ from os.path import join, dirname, abspath from logilab.common.testlib import TestCase, unittest_main from logilab.common.umessage import UMessage, decode_QP, message_from_string -DATA = join(dirname(abspath(__file__)), 'data') +DATA = join(dirname(abspath(__file__)), "data") -class UMessageTC(TestCase): +class UMessageTC(TestCase): def setUp(self): if sys.version_info >= (3, 2): import io - msg1 = email.message_from_file(io.open(join(DATA, 'test1.msg'), encoding='utf8')) - msg2 = email.message_from_file(io.open(join(DATA, 'test2.msg'), encoding='utf8')) + + msg1 = email.message_from_file(io.open(join(DATA, "test1.msg"), encoding="utf8")) + msg2 = email.message_from_file(io.open(join(DATA, "test2.msg"), encoding="utf8")) else: - msg1 = email.message_from_file(open(join(DATA, 'test1.msg'))) - msg2 = email.message_from_file(open(join(DATA, 'test2.msg'))) + msg1 = email.message_from_file(open(join(DATA, "test1.msg"))) + msg2 = email.message_from_file(open(join(DATA, "test2.msg"))) self.umessage1 = UMessage(msg1) self.umessage2 = UMessage(msg2) def test_get_subject(self): - subj = self.umessage2.get('Subject') + subj = self.umessage2.get("Subject") self.assertEqual(type(subj), str) - self.assertEqual(subj, u'À LA MER') + self.assertEqual(subj, "À LA MER") def test_get_all(self): - to = self.umessage2.get_all('To') + to = self.umessage2.get_all("To") self.assertEqual(type(to[0]), str) - self.assertEqual(to, [u'élément à accents <alf@logilab.fr>']) + self.assertEqual(to, ["élément à accents <alf@logilab.fr>"]) def test_get_payload_no_multi(self): payload = self.umessage1.get_payload() self.assertEqual(type(payload), str) - + def test_get_payload_decode(self): msg = """\ MIME-Version: 1.0 @@ -67,26 +68,26 @@ Date: now dW4gcGV0aXQgY8O2dWNvdQ== """ msg = message_from_string(msg) - self.assertEqual(msg.get_payload(decode=True), u'un petit cöucou') + self.assertEqual(msg.get_payload(decode=True), "un petit cöucou") def test_decode_QP(self): - test_line = '=??b?UmFwaGHrbA==?= DUPONT<raphael.dupont@societe.fr>' + test_line = "=??b?UmFwaGHrbA==?= DUPONT<raphael.dupont@societe.fr>" test = decode_QP(test_line) self.assertEqual(type(test), str) - self.assertEqual(test, u'Raphaël DUPONT<raphael.dupont@societe.fr>') + self.assertEqual(test, "Raphaël DUPONT<raphael.dupont@societe.fr>") def test_decode_QP_utf8(self): - test_line = '=?utf-8?q?o=C3=AEm?= <oim@logilab.fr>' + test_line = "=?utf-8?q?o=C3=AEm?= <oim@logilab.fr>" test = decode_QP(test_line) self.assertEqual(type(test), str) - self.assertEqual(test, u'oîm <oim@logilab.fr>') + self.assertEqual(test, "oîm <oim@logilab.fr>") def test_decode_QP_ascii(self): - test_line = 'test <test@logilab.fr>' + test_line = "test <test@logilab.fr>" test = decode_QP(test_line) self.assertEqual(type(test), str) - self.assertEqual(test, u'test <test@logilab.fr>') + self.assertEqual(test, "test <test@logilab.fr>") -if __name__ == '__main__': +if __name__ == "__main__": unittest_main() |