1 import unittest 2 import textwrap 3 import copy 4 import pickle 5 import email 6 import email.message 7 from email import policy 8 from email.headerregistry import HeaderRegistry 9 from test.test_email import TestEmailBase, parameterize 10 11 12 @parameterize 13 class TestPickleCopyHeader(TestEmailBase): 14 15 header_factory = HeaderRegistry() 16 17 unstructured = header_factory('subject', 'this is a test') 18 19 header_params = { 20 'subject': ('subject', 'this is a test'), 21 'from': ('from', 'frodo (at] mordor.net'), 22 'to': ('to', 'a: k (at] b.com, y (at] z.com;, j (at] f.com'), 23 'date': ('date', 'Tue, 29 May 2012 09:24:26 +1000'), 24 } 25 26 def header_as_deepcopy(self, name, value): 27 header = self.header_factory(name, value) 28 h = copy.deepcopy(header) 29 self.assertEqual(str(h), str(header)) 30 31 def header_as_pickle(self, name, value): 32 header = self.header_factory(name, value) 33 for proto in range(pickle.HIGHEST_PROTOCOL + 1): 34 p = pickle.dumps(header, proto) 35 h = pickle.loads(p) 36 self.assertEqual(str(h), str(header)) 37 38 39 @parameterize 40 class TestPickleCopyMessage(TestEmailBase): 41 42 # Message objects are a sequence, so we have to make them a one-tuple in 43 # msg_params so they get passed to the parameterized test method as a 44 # single argument instead of as a list of headers. 45 msg_params = {} 46 47 # Note: there will be no custom header objects in the parsed message. 48 msg_params['parsed'] = (email.message_from_string(textwrap.dedent("""\ 49 Date: Tue, 29 May 2012 09:24:26 +1000 50 From: frodo (at] mordor.net 51 To: bilbo (at] underhill.org 52 Subject: help 53 54 I think I forgot the ring. 55 """), policy=policy.default),) 56 57 msg_params['created'] = (email.message.Message(policy=policy.default),) 58 msg_params['created'][0]['Date'] = 'Tue, 29 May 2012 09:24:26 +1000' 59 msg_params['created'][0]['From'] = 'frodo (at] mordor.net' 60 msg_params['created'][0]['To'] = 'bilbo (at] underhill.org' 61 msg_params['created'][0]['Subject'] = 'help' 62 msg_params['created'][0].set_payload('I think I forgot the ring.') 63 64 def msg_as_deepcopy(self, msg): 65 msg2 = copy.deepcopy(msg) 66 self.assertEqual(msg2.as_string(), msg.as_string()) 67 68 def msg_as_pickle(self, msg): 69 for proto in range(pickle.HIGHEST_PROTOCOL + 1): 70 p = pickle.dumps(msg, proto) 71 msg2 = pickle.loads(p) 72 self.assertEqual(msg2.as_string(), msg.as_string()) 73 74 75 if __name__ == '__main__': 76 unittest.main() 77
