Removed the Requirement to Install Python and NodeJS (Now Bundled with Borealis)
This commit is contained in:
78
Dependencies/Python/Lib/test/test_email/test_inversion.py
vendored
Normal file
78
Dependencies/Python/Lib/test/test_email/test_inversion.py
vendored
Normal file
@ -0,0 +1,78 @@
|
||||
"""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
|
||||
"""),),
|
||||
|
||||
'header_with_invalid_date': (dedent(b"""\
|
||||
Date: Tue, 06 Jun 2017 27:39:33 +0600
|
||||
From: abc@xyz.com
|
||||
Subject: timezones
|
||||
|
||||
How do they work even?
|
||||
"""),),
|
||||
|
||||
}
|
||||
|
||||
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()
|
Reference in New Issue
Block a user