1
# Copyright (C) 2007 Canonical Ltd
1
# Copyright (C) 2007, 2009, 2011, 2014, 2016 Canonical Ltd
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
14
14
# along with this program; if not, write to the Free Software
15
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17
18
from email.Header import decode_header
19
20
from bzrlib import __version__ as _bzrlib_version
20
21
from bzrlib.email_message import EmailMessage
21
22
from bzrlib.errors import BzrBadParameterNotUnicode
22
23
from bzrlib.smtp_connection import SMTPConnection
23
from bzrlib.tests import TestCase
24
from bzrlib import tests
25
26
EMPTY_MESSAGE = '''\
26
27
From: from@from.com
66
67
''' % { 'version': _bzrlib_version, 'boundary': BOUNDARY }
68
SIMPLE_MULTIPART_MESSAGE = _MULTIPART_HEAD + '--%s--' % BOUNDARY
70
COMPLEX_MULTIPART_MESSAGE = _MULTIPART_HEAD + '''\
70
def final_newline_or_not(msg):
71
if sys.version_info >= (2, 7, 6):
72
# Some internals of python's email module changed in an (minor)
73
# incompatible way: a final newline is appended in 2.7.6...
78
def simple_multipart_message():
79
msg = _MULTIPART_HEAD + '--%s--' % BOUNDARY
80
return final_newline_or_not(msg)
83
def complex_multipart_message(typ):
84
msg = _MULTIPART_HEAD + '''\
73
87
Content-Type: text/%%s; charset="us-ascii"; name="lines.txt"
83
97
--%(boundary)s--''' % { 'boundary': BOUNDARY }
86
class TestEmailMessage(TestCase):
98
msg = final_newline_or_not(msg)
102
class TestEmailMessage(tests.TestCase):
88
104
def test_empty_message(self):
89
105
msg = EmailMessage('from@from.com', 'to@to.com', 'subject')
100
116
msg = EmailMessage('from@from.com', 'to@to.com', 'subject', body)
101
117
self.assertEqualDiff(expected, msg.as_string())
103
def test_multipart_message(self):
119
def test_multipart_message_simple(self):
104
120
msg = EmailMessage('from@from.com', 'to@to.com', 'subject')
105
121
msg.add_inline_attachment('body')
106
self.assertEqualDiff(SIMPLE_MULTIPART_MESSAGE, msg.as_string(BOUNDARY))
122
self.assertEqualDiff(simple_multipart_message(),
123
msg.as_string(BOUNDARY))
126
def test_multipart_message_complex(self):
108
127
msg = EmailMessage('from@from.com', 'to@to.com', 'subject', 'body')
109
128
msg.add_inline_attachment(u'a\nb\nc\nd\ne\n', 'lines.txt', 'x-subtype')
110
self.assertEqualDiff(COMPLEX_MULTIPART_MESSAGE % 'x-subtype',
111
msg.as_string(BOUNDARY))
129
self.assertEqualDiff(complex_multipart_message('x-subtype'),
130
msg.as_string(BOUNDARY))
113
132
def test_headers_accept_unicode_and_utf8(self):
114
133
for user in [ u'Pepe P\xe9rez <pperez@ejemplo.com>',
148
167
self.assertEqual('to2@to.com', msg['To'])
149
168
self.assertEqual('cc@cc.com', msg['Cc'])
153
def get_user_option(self, option):
158
def send_as_append(_self, msg):
159
messages.append(msg.as_string(BOUNDARY))
161
old_send_email = SMTPConnection.send_email
163
SMTPConnection.send_email = send_as_append
165
EmailMessage.send(FakeConfig(), 'from@from.com', 'to@to.com',
166
'subject', 'body', u'a\nb\nc\nd\ne\n', 'lines.txt')
167
self.assertEqualDiff(COMPLEX_MULTIPART_MESSAGE % 'plain',
171
EmailMessage.send(FakeConfig(), 'from@from.com', 'to@to.com',
172
'subject', 'body', u'a\nb\nc\nd\ne\n', 'lines.txt',
174
self.assertEqualDiff(COMPLEX_MULTIPART_MESSAGE % 'x-patch',
178
EmailMessage.send(FakeConfig(), 'from@from.com', 'to@to.com',
180
self.assertEqualDiff(SIMPLE_MESSAGE_ASCII , messages[0])
183
SMTPConnection.send_email = old_send_email
185
170
def test_address_to_encoded_header(self):
187
172
"""Convert a RFC2047-encoded string to a unicode string."""
203
188
address = u'Pepe P\xe9rez <pperez@ejemplo.com>' # unicode ok
204
189
encoded = EmailMessage.address_to_encoded_header(address)
205
self.assert_('pperez@ejemplo.com' in encoded) # addr must be unencoded
206
self.assertEquals(address, decode(encoded))
190
self.assertTrue('pperez@ejemplo.com' in encoded) # addr must be unencoded
191
self.assertEqual(address, decode(encoded))
208
193
address = 'Pepe P\xc3\xa9red <pperez@ejemplo.com>' # UTF-8 ok
209
194
encoded = EmailMessage.address_to_encoded_header(address)
210
self.assert_('pperez@ejemplo.com' in encoded)
211
self.assertEquals(address, decode(encoded).encode('utf-8'))
195
self.assertTrue('pperez@ejemplo.com' in encoded)
196
self.assertEqual(address, decode(encoded).encode('utf-8'))
213
198
address = 'Pepe P\xe9rez <pperez@ejemplo.com>' # ISO-8859-1 not ok
214
199
self.assertRaises(BzrBadParameterNotUnicode,
225
210
for string_, pair in pairs.items():
226
211
self.assertEqual(pair, EmailMessage.string_with_encoding(string_))
214
class TestSend(tests.TestCase):
217
super(TestSend, self).setUp()
220
def send_as_append(_self, msg):
221
self.messages.append(msg.as_string(BOUNDARY))
223
self.overrideAttr(SMTPConnection, 'send_email', send_as_append)
227
def send_email(self, attachment=None, attachment_filename=None,
228
attachment_mime_subtype='plain'):
230
def get(self, option):
233
EmailMessage.send(FakeConfig(), 'from@from.com', 'to@to.com',
235
attachment=attachment,
236
attachment_filename=attachment_filename,
237
attachment_mime_subtype=attachment_mime_subtype)
239
def assertMessage(self, expected):
240
self.assertLength(1, self.messages)
241
self.assertEqualDiff(expected, self.messages[0])
243
def test_send_plain(self):
244
self.send_email(u'a\nb\nc\nd\ne\n', 'lines.txt')
245
self.assertMessage(complex_multipart_message('plain'))
247
def test_send_patch(self):
248
self.send_email(u'a\nb\nc\nd\ne\n', 'lines.txt', 'x-patch')
249
self.assertMessage(complex_multipart_message('x-patch'))
251
def test_send_simple(self):
253
self.assertMessage(SIMPLE_MESSAGE_ASCII)