51 lines
1.8 KiB
Python
51 lines
1.8 KiB
Python
|
# Copyright (c) 2014, Menno Smits
|
||
|
# Released subject to the New BSD License
|
||
|
# Please see http://en.wikipedia.org/wiki/BSD_licenses
|
||
|
|
||
|
from __future__ import unicode_literals
|
||
|
|
||
|
from six import text_type, binary_type, int2byte, unichr
|
||
|
|
||
|
from imapclient.imap_utf7 import decode, encode
|
||
|
from imapclient.test.util import unittest
|
||
|
|
||
|
|
||
|
class IMAP4UTF7TestCase(unittest.TestCase):
|
||
|
tests = [
|
||
|
['Foo', b'Foo'],
|
||
|
['Foo Bar', b'Foo Bar'],
|
||
|
['Stuff & Things', b'Stuff &- Things'],
|
||
|
['Hello world', b'Hello world'],
|
||
|
['Hello & world', b'Hello &- world'],
|
||
|
['Hello\xffworld', b'Hello&AP8-world'],
|
||
|
['\xff\xfe\xfd\xfc', b'&AP8A,gD9APw-'],
|
||
|
['~peter/mail/\u65e5\u672c\u8a9e/\u53f0\u5317',
|
||
|
b'~peter/mail/&ZeVnLIqe-/&U,BTFw-'], # example from RFC 2060
|
||
|
['\x00foo', b'&AAA-foo'],
|
||
|
]
|
||
|
|
||
|
def test_encode(self):
|
||
|
for (input, output) in self.tests:
|
||
|
encoded = encode(input)
|
||
|
self.assertIsInstance(encoded, binary_type)
|
||
|
self.assertEqual(encoded, output)
|
||
|
|
||
|
def test_decode(self):
|
||
|
for (input, output) in self.tests:
|
||
|
decoded = decode(output)
|
||
|
self.assertIsInstance(decoded, text_type)
|
||
|
self.assertEqual(input, decoded)
|
||
|
|
||
|
def test_printable_singletons(self):
|
||
|
"""
|
||
|
The IMAP4 modified UTF-7 implementation encodes all printable
|
||
|
characters which are in ASCII using the corresponding ASCII byte.
|
||
|
"""
|
||
|
# All printables represent themselves
|
||
|
for o in list(range(0x20, 0x26)) + list(range(0x27, 0x7f)):
|
||
|
self.assertEqual(int2byte(o), encode(unichr(o)))
|
||
|
self.assertEqual(unichr(o), decode(int2byte(o)))
|
||
|
self.assertEqual(encode('&'), b'&-')
|
||
|
self.assertEqual(encode('&'), b'&-')
|
||
|
self.assertEqual(decode(b'&-'), '&')
|