← 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: Canonical.com Patch Queue Manager
  • Date: 2010-09-01 08:02:42 UTC
  • mfrom: (5390.3.3 faster-revert-593560)
  • Revision ID: pqm@pqm.ubuntu.com-20100901080242-esg62ody4frwmy66
(spiv) Avoid repeatedly calling self.target.all_file_ids() in
 InterTree.iter_changes. (Andrew Bennetts)

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_smtp_connection.py
1
 
# Copyright (C) 2005, 2007, 2009 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
91
91
class TestSMTPConnection(tests.TestCaseInTempDir):
92
92
 
93
93
    def get_connection(self, text, smtp_factory=None):
94
 
        my_config = config.GlobalConfig()
95
 
        config_file = StringIO(text)
96
 
        my_config._get_parser(config_file)
 
94
        my_config = config.GlobalConfig.from_string(text)
97
95
        return smtp_connection.SMTPConnection(my_config,
98
96
                                              _smtp_factory=smtp_factory)
99
97
 
160
158
 
161
159
    def test_authenticate_with_byte_strings(self):
162
160
        user = 'joe'
163
 
        password = 'h\xC3\xACspass'
 
161
        unicode_pass = u'h\xECspass'
 
162
        utf8_pass = unicode_pass.encode('utf-8')
164
163
        factory = WideOpenSMTPFactory()
165
164
        conn = self.get_connection(
166
 
            '[DEFAULT]\nsmtp_username=%s\nsmtp_password=%s\n'
167
 
            % (user, password), smtp_factory=factory)
168
 
        self.assertEqual(u'h\xECspass', conn._smtp_password)
 
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)
169
168
        conn._connect()
170
169
        self.assertEqual([('connect', 'localhost'),
171
170
                          ('ehlo',),
172
171
                          ('has_extn', 'starttls'),
173
 
                          ('login', user, password)], factory._calls)
 
172
                          ('login', user, utf8_pass)], factory._calls)
174
173
        smtp_username, smtp_password = factory._calls[-1][1:]
175
174
        self.assertIsInstance(smtp_username, str)
176
175
        self.assertIsInstance(smtp_password, str)