~bzr-pqm/bzr/bzr.dev

1185.49.22 by John Arbash Meinel
Added get_password to the UIFactory, using it inside of sftp.py
1
# Copyright (C) 2005 by Canonical Ltd
2
#
3
# This program is free software; you can redistribute it and/or modify
4
# it under the terms of the GNU General Public License as published by
5
# the Free Software Foundation; either version 2 of the License, or
6
# (at your option) any later version.
7
#
8
# This program is distributed in the hope that it will be useful,
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11
# GNU General Public License for more details.
12
#
13
# You should have received a copy of the GNU General Public License
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
16
17
"""Tests for the bzrlib ui
18
"""
19
20
import os
21
import sys
22
23
from bzrlib.tests import TestCase
24
from bzrlib.ui import SilentUIFactory
25
from bzrlib.ui.text import TextUIFactory
26
27
class UITests(TestCase):
28
29
    def test_silent_factory(self):
30
31
        ui = SilentUIFactory()
32
        pb = ui.progress_bar()
33
        # TODO: Test that there is no output from SilentUIFactory
34
35
        self.assertEquals(ui.get_password(), None)
36
        self.assertEquals(ui.get_password(u'Hello There \u1234 %(user)s',
37
                                          user=u'some\u1234')
38
                         , None)
39
40
    def test_text_factory(self):
41
        ui = TextUIFactory()
42
        pb = ui.progress_bar()
43
        # TODO: Test the output from TextUIFactory, perhaps by overriding sys.stdout
44
45
        # Unfortunately we can't actually test the ui.get_password() because 
46
        # that would actually prompt the user for a password during the test suite
47
        # This has been tested manually with both LANG=en_US.utf-8 and LANG=C
48
        # print
49
        # self.assertEquals(ui.get_password(u"%(user)s please type 'bogus'",
50
        #                                   user=u'some\u1234')
51
        #                  , 'bogus')
52