1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
|
# Copyright (C) 2005, 2007 Canonical Ltd
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
from cStringIO import StringIO
from email.Message import Message
import errno
import smtplib
import socket
from bzrlib import (
config,
errors,
)
from bzrlib.email_message import EmailMessage
from bzrlib.errors import NoDestinationAddress
from bzrlib.tests import TestCase
from bzrlib.smtp_connection import SMTPConnection
def connection_refuser():
def connect(server):
raise socket.error(errno.ECONNREFUSED, 'Connection Refused')
smtp = smtplib.SMTP()
smtp.connect = connect
return smtp
class TestSMTPConnection(TestCase):
def get_connection(self, text, smtp_factory=None):
my_config = config.GlobalConfig()
config_file = StringIO(text)
my_config._get_parser(config_file)
return SMTPConnection(my_config, _smtp_factory=smtp_factory)
def test_defaults(self):
conn = self.get_connection('')
self.assertEqual('localhost', conn._smtp_server)
self.assertEqual(None, conn._smtp_username)
self.assertEqual(None, conn._smtp_password)
def test_smtp_server(self):
conn = self.get_connection('[DEFAULT]\nsmtp_server=host:10\n')
self.assertEqual('host:10', conn._smtp_server)
def test_missing_server(self):
conn = self.get_connection('', smtp_factory=connection_refuser)
self.assertRaises(errors.DefaultSMTPConnectionRefused, conn._connect)
conn = self.get_connection('[DEFAULT]\nsmtp_server=smtp.example.com\n',
smtp_factory=connection_refuser)
self.assertRaises(errors.SMTPConnectionRefused, conn._connect)
def test_smtp_username(self):
conn = self.get_connection('')
self.assertIs(None, conn._smtp_username)
conn = self.get_connection('[DEFAULT]\nsmtp_username=joebody\n')
self.assertEqual(u'joebody', conn._smtp_username)
def test_smtp_password(self):
conn = self.get_connection('')
self.assertIs(None, conn._smtp_password)
conn = self.get_connection('[DEFAULT]\nsmtp_password=mypass\n')
self.assertEqual(u'mypass', conn._smtp_password)
def test_get_message_addresses(self):
msg = Message()
from_, to = SMTPConnection.get_message_addresses(msg)
self.assertEqual('', from_)
self.assertEqual([], to)
msg['From'] = '"J. Random Developer" <jrandom@example.com>'
msg['To'] = 'John Doe <john@doe.com>, Jane Doe <jane@doe.com>'
msg['CC'] = u'Pepe P\xe9rez <pperez@ejemplo.com>'
msg['Bcc'] = 'user@localhost'
from_, to = SMTPConnection.get_message_addresses(msg)
self.assertEqual('jrandom@example.com', from_)
self.assertEqual(sorted(['john@doe.com', 'jane@doe.com',
'pperez@ejemplo.com', 'user@localhost']), sorted(to))
# now with bzrlib's EmailMessage
msg = EmailMessage('"J. Random Developer" <jrandom@example.com>', [
'John Doe <john@doe.com>', 'Jane Doe <jane@doe.com>',
u'Pepe P\xe9rez <pperez@ejemplo.com>', 'user@localhost' ],
'subject')
from_, to = SMTPConnection.get_message_addresses(msg)
self.assertEqual('jrandom@example.com', from_)
self.assertEqual(sorted(['john@doe.com', 'jane@doe.com',
'pperez@ejemplo.com', 'user@localhost']), sorted(to))
def test_destination_address_required(self):
class FakeConfig:
def get_user_option(self, option):
return None
msg = Message()
msg['From'] = '"J. Random Developer" <jrandom@example.com>'
self.assertRaises(NoDestinationAddress,
SMTPConnection(FakeConfig()).send_email, msg)
msg = EmailMessage('from@from.com', '', 'subject')
self.assertRaises(NoDestinationAddress,
SMTPConnection(FakeConfig()).send_email, msg)
msg = EmailMessage('from@from.com', [], 'subject')
self.assertRaises(NoDestinationAddress,
SMTPConnection(FakeConfig()).send_email, msg)
|