Beta Shell v2.0

File Manager

~ /usr/local/python-3.5/lib/python3.5/test/test_email

Name	Size	Perms	Modified	Actions
[ .. / .. ]
__init__.py	6.19 KB	-rw-r--r--	2022-02-22 15:51:03	Edit Del
__main__.py	72 B	-rw-r--r--	2022-02-22 15:51:03	Edit Del
__pycache__	-	drwxr-xr-x	2022-02-22 15:51:22	Del
data	-	drwxr-xr-x	2022-02-22 15:51:03	Del
test__encoded_words.py	6.24 KB	-rw-r--r--	2022-02-22 15:51:03	Edit Del
test__header_value_parser.py	114.47 KB	-rw-r--r--	2022-02-22 15:51:03	Edit Del
test_asian_codecs.py	3.11 KB	-rw-r--r--	2022-02-22 15:51:03	Edit Del
test_contentmanager.py	32.75 KB	-rw-r--r--	2022-02-22 15:51:03	Edit Del
test_defect_handling.py	11.06 KB	-rw-r--r--	2022-02-22 15:51:03	Edit Del
test_email.py	202.03 KB	-rw-r--r--	2022-02-22 15:51:03	Edit Del
test_generator.py	9.44 KB	-rw-r--r--	2022-02-22 15:51:03	Edit Del
test_headerregistry.py	56.43 KB	-rw-r--r--	2022-02-22 15:51:03	Edit Del
test_inversion.py	2.03 KB	-rw-r--r--	2022-02-22 15:51:03	Edit Del
test_message.py	25.74 KB	-rw-r--r--	2022-02-22 15:51:03	Edit Del
test_parser.py	2.74 KB	-rw-r--r--	2022-02-22 15:51:03	Edit Del
test_pickleable.py	2.49 KB	-rw-r--r--	2022-02-22 15:51:03	Edit Del
test_policy.py	13.41 KB	-rw-r--r--	2022-02-22 15:51:03	Edit Del
test_utils.py	6.42 KB	-rw-r--r--	2022-02-22 15:51:03	Edit Del
torture_test.py	3.52 KB	-rw-r--r--	2022-02-22 15:51:03	Edit Del

Editing: test_inversion.py (2.03 KB)

"""Test the parser and generator are inverses.

Note that this is only strictly true if we are parsing RFC valid messages and
producing RFC valid messages.
"""

import io
import unittest
from email import policy, message_from_bytes
from email.message import EmailMessage
from email.generator import BytesGenerator
from test.test_email import TestEmailBase, parameterize

# This is like textwrap.dedent for bytes, except that it uses \r\n for the line
# separators on the rebuilt string.
def dedent(bstr):
    lines = bstr.splitlines()
    if not lines[0].strip():
        raise ValueError("First line must contain text")
    stripamt = len(lines[0]) - len(lines[0].lstrip())
    return b'\r\n'.join(
        [x[stripamt:] if len(x)>=stripamt else b''
            for x in lines])


@parameterize
class TestInversion(TestEmailBase):

    policy = policy.default
    message = EmailMessage

    def msg_as_input(self, msg):
        m = message_from_bytes(msg, policy=policy.SMTP)
        b = io.BytesIO()
        g = BytesGenerator(b)
        g.flatten(m)
        self.assertEqual(b.getvalue(), msg)

    # XXX: spaces are not preserved correctly here yet in the general case.
    msg_params = {
        'header_with_one_space_body': (dedent(b"""\
            From: abc@xyz.com
            X-Status:\x20
            Subject: test

            foo
            """),),

            }

    payload_params = {
        'plain_text': dict(payload='This is a test\n'*20),
        'base64_text': dict(payload=(('xy a'*40+'\n')*5), cte='base64'),
        'qp_text': dict(payload=(('xy a'*40+'\n')*5), cte='quoted-printable'),
        }

    def payload_as_body(self, payload, **kw):
        msg = self._make_message()
        msg['From'] = 'foo'
        msg['To'] = 'bar'
        msg['Subject'] = 'payload round trip test'
        msg.set_content(payload, **kw)
        b = bytes(msg)
        msg2 = message_from_bytes(b, policy=self.policy)
        self.assertEqual(bytes(msg2), b)
        self.assertEqual(msg2.get_content(), payload)


if __name__ == '__main__':
    unittest.main()