94a7927cc6
This is buggy in the old email code as well, but it doesn't affect anything there because only the default line length is ever used there.
71 lines
2.0 KiB
Python
71 lines
2.0 KiB
Python
"""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()
|