← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_ui.py

  • Committer: Martin Pool
  • Date: 2009-07-27 06:28:35 UTC
  • mto: This revision was merged to the branch mainline in revision 4587.
  • Revision ID: mbp@sourcefrog.net-20090727062835-o66p8it658tq1sma
Add CountedLock.get_physical_lock_status

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_ui.py
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
"""Tests for the bzrlib ui
18
18
"""
21
21
from StringIO import StringIO
22
22
import re
23
23
import sys
 
24
import time
24
25
 
25
 
import bzrlib
26
 
import bzrlib.errors as errors
27
 
from bzrlib.progress import (
28
 
    DotsProgressBar,
29
 
    ProgressBarStack,
30
 
    TTYProgressBar,
 
26
from bzrlib import (
 
27
    errors,
 
28
    tests,
 
29
    ui as _mod_ui,
31
30
    )
32
31
from bzrlib.symbol_versioning import (
33
32
    deprecated_in,
37
36
    TestUIFactory,
38
37
    StringIOWrapper,
39
38
    )
40
 
from bzrlib.tests.test_progress import _TTYStringIO
 
39
from bzrlib.tests.test_progress import (
 
40
    _NonTTYStringIO,
 
41
    _TTYStringIO,
 
42
    )
41
43
from bzrlib.ui import (
 
44
    CannedInputUIFactory,
42
45
    CLIUIFactory,
43
46
    SilentUIFactory,
44
 
    )
45
 
from bzrlib.ui.text import TextUIFactory
46
 
 
47
 
 
48
 
class UITests(TestCase):
49
 
 
50
 
    def test_silent_factory(self):
51
 
        ui = SilentUIFactory()
52
 
        stdout = StringIO()
53
 
        self.assertEqual(None,
54
 
                         self.apply_redirected(None, stdout, stdout,
55
 
                                               ui.get_password))
56
 
        self.assertEqual('', stdout.getvalue())
57
 
        self.assertEqual(None,
58
 
                         self.apply_redirected(None, stdout, stdout,
59
 
                                               ui.get_password,
60
 
                                               u'Hello\u1234 %(user)s',
61
 
                                               user=u'some\u1234'))
62
 
        self.assertEqual('', stdout.getvalue())
 
47
    UIFactory,
 
48
    make_ui_for_terminal,
 
49
    )
 
50
from bzrlib.ui.text import (
 
51
    NullProgressView,
 
52
    TextProgressView,
 
53
    TextUIFactory,
 
54
    )
 
55
 
 
56
 
 
57
class UITests(tests.TestCase):
63
58
 
64
59
    def test_text_factory_ascii_password(self):
65
 
        ui = TestUIFactory(stdin='secret\n', stdout=StringIOWrapper())
 
60
        ui = tests.TestUIFactory(stdin='secret\n',
 
61
                                 stdout=tests.StringIOWrapper(),
 
62
                                 stderr=tests.StringIOWrapper())
66
63
        pb = ui.nested_progress_bar()
67
64
        try:
68
65
            self.assertEqual('secret',
69
66
                             self.apply_redirected(ui.stdin, ui.stdout,
70
 
                                                   ui.stdout,
 
67
                                                   ui.stderr,
71
68
                                                   ui.get_password))
72
69
            # ': ' is appended to prompt
73
 
            self.assertEqual(': ', ui.stdout.getvalue())
 
70
            self.assertEqual(': ', ui.stderr.getvalue())
 
71
            self.assertEqual('', ui.stdout.readline())
74
72
            # stdin should be empty
75
73
            self.assertEqual('', ui.stdin.readline())
76
74
        finally:
82
80
        We can't predict what encoding users will have for stdin, so we force
83
81
        it to utf8 to test that we transport the password correctly.
84
82
        """
85
 
        ui = TestUIFactory(stdin=u'baz\u1234'.encode('utf8'),
86
 
                           stdout=StringIOWrapper())
87
 
        ui.stdin.encoding = 'utf8'
88
 
        ui.stdout.encoding = ui.stdin.encoding
 
83
        ui = tests.TestUIFactory(stdin=u'baz\u1234'.encode('utf8'),
 
84
                                 stdout=tests.StringIOWrapper(),
 
85
                                 stderr=tests.StringIOWrapper())
 
86
        ui.stderr.encoding = ui.stdout.encoding = ui.stdin.encoding = 'utf8'
89
87
        pb = ui.nested_progress_bar()
90
88
        try:
91
 
            password = self.apply_redirected(ui.stdin, ui.stdout, ui.stdout,
 
89
            password = self.apply_redirected(ui.stdin, ui.stdout, ui.stderr,
92
90
                                             ui.get_password,
93
91
                                             u'Hello \u1234 %(user)s',
94
92
                                             user=u'some\u1234')
95
93
            # We use StringIO objects, we need to decode them
96
94
            self.assertEqual(u'baz\u1234', password.decode('utf8'))
97
95
            self.assertEqual(u'Hello \u1234 some\u1234: ',
98
 
                             ui.stdout.getvalue().decode('utf8'))
99
 
            # stdin should be empty
 
96
                             ui.stderr.getvalue().decode('utf8'))
 
97
            # stdin and stdout should be empty
100
98
            self.assertEqual('', ui.stdin.readline())
 
99
            self.assertEqual('', ui.stdout.readline())
101
100
        finally:
102
101
            pb.finished()
103
102
 
 
103
    def test_progress_construction(self):
 
104
        """TextUIFactory constructs the right progress view.
 
105
        """
 
106
        for (file_class, term, pb, expected_pb_class) in (
 
107
            # on an xterm, either use them or not as the user requests,
 
108
            # otherwise default on
 
109
            (_TTYStringIO, 'xterm', 'none', NullProgressView),
 
110
            (_TTYStringIO, 'xterm', 'text', TextProgressView),
 
111
            (_TTYStringIO, 'xterm', None, TextProgressView),
 
112
            # on a dumb terminal, again if there's explicit configuration do
 
113
            # it, otherwise default off
 
114
            (_TTYStringIO, 'dumb', 'none', NullProgressView),
 
115
            (_TTYStringIO, 'dumb', 'text', TextProgressView),
 
116
            (_TTYStringIO, 'dumb', None, NullProgressView),
 
117
            # on a non-tty terminal, it's null regardless of $TERM
 
118
            (StringIO, 'xterm', None, NullProgressView),
 
119
            (StringIO, 'dumb', None, NullProgressView),
 
120
            # however, it can still be forced on
 
121
            (StringIO, 'dumb', 'text', TextProgressView),
 
122
            ):
 
123
            os.environ['TERM'] = term
 
124
            if pb is None:
 
125
                if 'BZR_PROGRESS_BAR' in os.environ:
 
126
                    del os.environ['BZR_PROGRESS_BAR']
 
127
            else:
 
128
                os.environ['BZR_PROGRESS_BAR'] = pb
 
129
            stdin = file_class('')
 
130
            stderr = file_class()
 
131
            stdout = file_class()
 
132
            uif = make_ui_for_terminal(stdin, stdout, stderr)
 
133
            self.assertIsInstance(uif, TextUIFactory,
 
134
                "TERM=%s BZR_PROGRESS_BAR=%s uif=%r" % (term, pb, uif,))
 
135
            self.assertIsInstance(uif.make_progress_view(),
 
136
                expected_pb_class,
 
137
                "TERM=%s BZR_PROGRESS_BAR=%s uif=%r" % (term, pb, uif,))
 
138
 
 
139
    def test_text_ui_non_terminal(self):
 
140
        """Even on non-ttys, make_ui_for_terminal gives a text ui."""
 
141
        stdin = _NonTTYStringIO('')
 
142
        stderr = _NonTTYStringIO()
 
143
        stdout = _NonTTYStringIO()
 
144
        for term_type in ['dumb', None, 'xterm']:
 
145
            if term_type is None:
 
146
                del os.environ['TERM']
 
147
            else:
 
148
                os.environ['TERM'] = term_type
 
149
            uif = make_ui_for_terminal(stdin, stdout, stderr)
 
150
            self.assertIsInstance(uif, TextUIFactory,
 
151
                'TERM=%r' % (term_type,))
 
152
 
104
153
    def test_progress_note(self):
105
154
        stderr = StringIO()
106
155
        stdout = StringIO()
120
169
            pb.finished()
121
170
 
122
171
    def test_progress_note_clears(self):
123
 
        stderr = StringIO()
124
 
        stdout = StringIO()
125
 
        # The PQM redirects the output to a file, so it
126
 
        # defaults to creating a Dots progress bar. we
127
 
        # need to force it to believe we are a TTY
 
172
        stderr = _TTYStringIO()
 
173
        stdout = _TTYStringIO()
 
174
        # so that we get a TextProgressBar
 
175
        os.environ['TERM'] = 'xterm'
128
176
        ui_factory = TextUIFactory(
129
177
            stdin=StringIO(''),
130
178
            stdout=stdout, stderr=stderr)
 
179
        self.assertIsInstance(ui_factory._progress_view,
 
180
            TextProgressView)
131
181
        pb = ui_factory.nested_progress_bar()
132
182
        try:
133
183
            # Create a progress update that isn't throttled
156
206
        pb2.finished()
157
207
        pb1.finished()
158
208
 
159
 
    def test_progress_stack(self):
160
 
        # test the progress bar stack which the default text factory 
161
 
        # uses.
 
209
    def test_text_ui_get_boolean(self):
 
210
        stdin = StringIO("y\n" # True
 
211
                         "n\n" # False
 
212
                         "yes with garbage\nY\n" # True
 
213
                         "not an answer\nno\n" # False
 
214
                         "I'm sure!\nyes\n" # True
 
215
                         "NO\n" # False
 
216
                         "foo\n")
 
217
        stdout = StringIO()
162
218
        stderr = StringIO()
163
 
        stdout = StringIO()
164
 
        # make a stack, which accepts parameters like a pb.
165
 
        stack = self.applyDeprecated(
166
 
            deprecated_in((1, 12, 0)),
167
 
            ProgressBarStack,
168
 
            to_file=stderr, to_messages_file=stdout)
169
 
        # but is not one
170
 
        self.assertFalse(getattr(stack, 'note', False))
171
 
        pb1 = stack.get_nested()
172
 
        pb2 = stack.get_nested()
173
 
        warnings, _ = self.callCatchWarnings(pb1.finished)
174
 
        self.assertEqual(len(warnings), 1)
175
 
        pb2.finished()
176
 
        pb1.finished()
177
 
        # the text ui factory never actually removes the stack once its setup.
178
 
        # we need to be able to nest again correctly from here.
179
 
        pb1 = stack.get_nested()
180
 
        pb2 = stack.get_nested()
181
 
        warnings, _ = self.callCatchWarnings(pb1.finished)
182
 
        self.assertEqual(len(warnings), 1)
183
 
        pb2.finished()
184
 
        pb1.finished()
185
 
 
186
 
    def assert_get_bool_acceptance_of_user_input(self, factory):
187
 
        factory.stdin = StringIO("y\nyes with garbage\n"
188
 
                                 "yes\nn\nnot an answer\n"
189
 
                                 "no\nfoo\n")
190
 
        factory.stdout = StringIO()
191
 
        # there is no output from the base factory
192
 
        self.assertEqual(True, factory.get_boolean(""))
193
 
        self.assertEqual(True, factory.get_boolean(""))
194
 
        self.assertEqual(False, factory.get_boolean(""))
 
219
        factory = TextUIFactory(stdin, stdout, stderr)
 
220
        self.assertEqual(True, factory.get_boolean(""))
 
221
        self.assertEqual(False, factory.get_boolean(""))
 
222
        self.assertEqual(True, factory.get_boolean(""))
 
223
        self.assertEqual(False, factory.get_boolean(""))
 
224
        self.assertEqual(True, factory.get_boolean(""))
195
225
        self.assertEqual(False, factory.get_boolean(""))
196
226
        self.assertEqual("foo\n", factory.stdin.read())
197
227
        # stdin should be empty
198
228
        self.assertEqual('', factory.stdin.readline())
199
229
 
200
 
    def test_silent_ui_getbool(self):
201
 
        factory = SilentUIFactory()
202
 
        self.assert_get_bool_acceptance_of_user_input(factory)
203
 
 
204
 
    def test_silent_factory_prompts_silently(self):
205
 
        factory = SilentUIFactory()
206
 
        stdout = StringIO()
207
 
        factory.stdin = StringIO("y\n")
208
 
        self.assertEqual(True,
209
 
                         self.apply_redirected(None, stdout, stdout,
210
 
                                               factory.get_boolean, "foo"))
211
 
        self.assertEqual("", stdout.getvalue())
212
 
        # stdin should be empty
213
 
        self.assertEqual('', factory.stdin.readline())
214
 
 
215
 
    def test_text_ui_getbool(self):
216
 
        factory = TextUIFactory(None, None, None)
217
 
        self.assert_get_bool_acceptance_of_user_input(factory)
 
230
    def test_text_factory_prompt(self):
 
231
        # see <https://launchpad.net/bugs/365891>
 
232
        factory = TextUIFactory(StringIO(), StringIO(), StringIO())
 
233
        factory.prompt('foo %2e')
 
234
        self.assertEqual('', factory.stdout.getvalue())
 
235
        self.assertEqual('foo %2e', factory.stderr.getvalue())
218
236
 
219
237
    def test_text_factory_prompts_and_clears(self):
220
238
        # a get_boolean call should clear the pb before prompting
221
239
        out = _TTYStringIO()
 
240
        os.environ['TERM'] = 'xterm'
222
241
        factory = TextUIFactory(stdin=StringIO("yada\ny\n"), stdout=out, stderr=out)
223
242
        pb = factory.nested_progress_bar()
224
243
        pb.show_bar = False
237
256
            r"what do you want\? \[y/n\]: what do you want\? \[y/n\]: ")
238
257
        # stdin should have been totally consumed
239
258
        self.assertEqual('', factory.stdin.readline())
 
259
 
 
260
    def test_text_tick_after_update(self):
 
261
        ui_factory = TextUIFactory(stdout=StringIO(), stderr=StringIO())
 
262
        pb = ui_factory.nested_progress_bar()
 
263
        try:
 
264
            pb.update('task', 0, 3)
 
265
            # Reset the clock, so that it actually tries to repaint itself
 
266
            ui_factory._progress_view._last_repaint = time.time() - 1.0
 
267
            pb.tick()
 
268
        finally:
 
269
            pb.finished()
 
270
 
 
271
    def test_text_ui_getusername(self):
 
272
        factory = TextUIFactory(None, None, None)
 
273
        factory.stdin = StringIO("someuser\n\n")
 
274
        factory.stdout = StringIO()
 
275
        factory.stderr = StringIO()
 
276
        factory.stdout.encoding = "utf8"
 
277
        # there is no output from the base factory
 
278
        self.assertEqual("someuser",
 
279
                         factory.get_username('Hello %(host)s', host='some'))
 
280
        self.assertEquals("Hello some: ", factory.stderr.getvalue())
 
281
        self.assertEquals('', factory.stdout.getvalue())
 
282
        self.assertEqual("", factory.get_username("Gebruiker"))
 
283
        # stdin should be empty
 
284
        self.assertEqual('', factory.stdin.readline())
 
285
 
 
286
    def test_text_ui_getusername_utf8(self):
 
287
        ui = tests.TestUIFactory(stdin=u'someuser\u1234'.encode('utf8'),
 
288
                                 stdout=tests.StringIOWrapper(),
 
289
                                 stderr=tests.StringIOWrapper())
 
290
        ui.stderr.encoding = ui.stdout.encoding = ui.stdin.encoding = "utf8"
 
291
        pb = ui.nested_progress_bar()
 
292
        try:
 
293
            # there is no output from the base factory
 
294
            username = self.apply_redirected(ui.stdin, ui.stdout, ui.stderr,
 
295
                ui.get_username, u'Hello\u1234 %(host)s', host=u'some\u1234')
 
296
            self.assertEquals(u"someuser\u1234", username.decode('utf8'))
 
297
            self.assertEquals(u"Hello\u1234 some\u1234: ",
 
298
                              ui.stderr.getvalue().decode("utf8"))
 
299
            self.assertEquals('', ui.stdout.getvalue())
 
300
        finally:
 
301
            pb.finished()
 
302
 
 
303
 
 
304
class CLIUITests(TestCase):
 
305
 
 
306
    def test_cli_factory_deprecated(self):
 
307
        uif = self.applyDeprecated(deprecated_in((1, 18, 0)),
 
308
            CLIUIFactory,
 
309
            StringIO(), StringIO(), StringIO())
 
310
        self.assertIsInstance(uif, UIFactory)
 
311
 
 
312
 
 
313
class SilentUITests(TestCase):
 
314
 
 
315
    def test_silent_factory_get_password(self):
 
316
        # A silent factory that can't do user interaction can't get a
 
317
        # password.  Possibly it should raise a more specific error but it
 
318
        # can't succeed.
 
319
        ui = SilentUIFactory()
 
320
        stdout = StringIO()
 
321
        self.assertRaises(
 
322
            NotImplementedError,
 
323
            self.apply_redirected,
 
324
            None, stdout, stdout, ui.get_password)
 
325
        # and it didn't write anything out either
 
326
        self.assertEqual('', stdout.getvalue())
 
327
 
 
328
    def test_silent_ui_getbool(self):
 
329
        factory = SilentUIFactory()
 
330
        stdout = StringIO()
 
331
        self.assertRaises(
 
332
            NotImplementedError,
 
333
            self.apply_redirected,
 
334
            None, stdout, stdout, factory.get_boolean, "foo")
 
335
 
 
336
 
 
337
class CannedInputUIFactoryTests(TestCase):
 
338
    
 
339
    def test_canned_input_get_input(self):
 
340
        uif = CannedInputUIFactory([True, 'mbp', 'password'])
 
341
        self.assertEqual(uif.get_boolean('Extra cheese?'), True)
 
342
        self.assertEqual(uif.get_username('Enter your user name'), 'mbp')
 
343
        self.assertEqual(uif.get_password('Password for %(host)s', host='example.com'),
 
344
            'password')
 
345
 
 
346
 
 
347
class TestBoolFromString(tests.TestCase):
 
348
 
 
349
    def assertIsTrue(self, s, accepted_values=None):
 
350
        res = _mod_ui.bool_from_string(s, accepted_values=accepted_values)
 
351
        self.assertEquals(True, res)
 
352
 
 
353
    def assertIsFalse(self, s, accepted_values=None):
 
354
        res = _mod_ui.bool_from_string(s, accepted_values=accepted_values)
 
355
        self.assertEquals(False, res)
 
356
 
 
357
    def assertIsNone(self, s, accepted_values=None):
 
358
        res = _mod_ui.bool_from_string(s, accepted_values=accepted_values)
 
359
        self.assertIs(None, res)
 
360
 
 
361
    def test_know_valid_values(self):
 
362
        self.assertIsTrue('true')
 
363
        self.assertIsFalse('false')
 
364
        self.assertIsTrue('1')
 
365
        self.assertIsFalse('0')
 
366
        self.assertIsTrue('on')
 
367
        self.assertIsFalse('off')
 
368
        self.assertIsTrue('yes')
 
369
        self.assertIsFalse('no')
 
370
        self.assertIsTrue('y')
 
371
        self.assertIsFalse('n')
 
372
        # Also try some case variations
 
373
        self.assertIsTrue('True')
 
374
        self.assertIsFalse('False')
 
375
        self.assertIsTrue('On')
 
376
        self.assertIsFalse('Off')
 
377
        self.assertIsTrue('ON')
 
378
        self.assertIsFalse('OFF')
 
379
        self.assertIsTrue('oN')
 
380
        self.assertIsFalse('oFf')
 
381
 
 
382
    def test_invalid_values(self):
 
383
        self.assertIsNone(None)
 
384
        self.assertIsNone('doubt')
 
385
        self.assertIsNone('frue')
 
386
        self.assertIsNone('talse')
 
387
        self.assertIsNone('42')
 
388
 
 
389
    def test_provided_values(self):
 
390
        av = dict(y=True, n=False, yes=True, no=False)
 
391
        self.assertIsTrue('y', av)
 
392
        self.assertIsTrue('Y', av)
 
393
        self.assertIsTrue('Yes', av)
 
394
        self.assertIsFalse('n', av)
 
395
        self.assertIsFalse('N', av)
 
396
        self.assertIsFalse('No', av)
 
397
        self.assertIsNone('1', av)
 
398
        self.assertIsNone('0', av)
 
399
        self.assertIsNone('on', av)
 
400
        self.assertIsNone('off', av)