← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_smtp_connection.py

  • Committer: Jelmer Vernooij
  • Date: 2010-12-20 11:57:14 UTC
  • mto: This revision was merged to the branch mainline in revision 5577.
  • Revision ID: jelmer@samba.org-20101220115714-2ru3hfappjweeg7q
Don't use no-plugins.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_smtp_connection.py
1
 
# Copyright (C) 2005, 2007 Canonical Ltd
 
1
# Copyright (C) 2007, 2009, 2010 Canonical Ltd
2
2
#
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
12
12
#
13
13
# You should have received a copy of the GNU General Public License
14
14
# along with this program; if not, write to the Free Software
15
 
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
16
 
17
17
from cStringIO import StringIO
18
18
from email.Message import Message
19
19
import errno
20
20
import smtplib
21
21
import socket
22
 
import sys
23
22
 
24
23
from bzrlib import (
25
24
    config,
86
85
    """A fake smtp server that implements login by accepting anybody."""
87
86
 
88
87
    def login(self, user, password):
89
 
        pass
 
88
        self._calls.append(('login', user, password))
90
89
 
91
90
 
92
91
class TestSMTPConnection(tests.TestCaseInTempDir):
93
92
 
94
93
    def get_connection(self, text, smtp_factory=None):
95
 
        my_config = config.GlobalConfig()
96
 
        config_file = StringIO(text)
97
 
        my_config._get_parser(config_file)
 
94
        my_config = config.GlobalConfig.from_string(text)
98
95
        return smtp_connection.SMTPConnection(my_config,
99
96
                                              _smtp_factory=smtp_factory)
100
97
 
137
134
                                   smtp_factory=factory)
138
135
        self.assertIs(None, conn._smtp_password)
139
136
 
140
 
        ui.ui_factory = tests.TestUIFactory(stdin=password + '\n',
141
 
                                            stdout=tests.StringIOWrapper())
 
137
        ui.ui_factory = ui.CannedInputUIFactory([password])
142
138
        conn._connect()
143
139
        self.assertEqual(password, conn._smtp_password)
144
 
        # stdin should be empty (the provided password have been consumed)
145
 
        self.assertEqual('', ui.ui_factory.stdin.readline())
146
140
 
147
141
    def test_smtp_password_from_auth_config(self):
148
142
        user = 'joe'
162
156
        conn._connect()
163
157
        self.assertEqual(password, conn._smtp_password)
164
158
 
 
159
    def test_authenticate_with_byte_strings(self):
 
160
        user = 'joe'
 
161
        unicode_pass = u'h\xECspass'
 
162
        utf8_pass = unicode_pass.encode('utf-8')
 
163
        factory = WideOpenSMTPFactory()
 
164
        conn = self.get_connection(
 
165
            u'[DEFAULT]\nsmtp_username=%s\nsmtp_password=%s\n'
 
166
            % (user, unicode_pass), smtp_factory=factory)
 
167
        self.assertEqual(unicode_pass, conn._smtp_password)
 
168
        conn._connect()
 
169
        self.assertEqual([('connect', 'localhost'),
 
170
                          ('ehlo',),
 
171
                          ('has_extn', 'starttls'),
 
172
                          ('login', user, utf8_pass)], factory._calls)
 
173
        smtp_username, smtp_password = factory._calls[-1][1:]
 
174
        self.assertIsInstance(smtp_username, str)
 
175
        self.assertIsInstance(smtp_password, str)
 
176
 
165
177
    def test_create_connection(self):
166
178
        factory = StubSMTPFactory()
167
179
        conn = self.get_connection('', smtp_factory=factory)