← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_progress.py

  • Committer: Canonical.com Patch Queue Manager
  • Date: 2009-03-16 13:18:16 UTC
  • mfrom: (4149.1.1 bzr.integration)
  • Revision ID: pqm@pqm.ubuntu.com-20090316131816-p0a3ugbpmbqm3a04
(vila,
        jfroy) Provides all request parameters to authentication providers

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_progress.py
1
 
# Copyright (C) 2006 by Canonical Ltd
 
1
# Copyright (C) 2006, 2007, 2009 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
19
19
 
20
20
from bzrlib import errors
21
21
from bzrlib.progress import (
22
 
        DummyProgress, ChildProgress,
 
22
        DummyProgress,
 
23
        ChildProgress,
23
24
        TTYProgressBar,
24
25
        DotsProgressBar,
25
26
        ProgressBarStack,
 
27
        InstrumentedProgress,
26
28
        )
27
29
from bzrlib.tests import TestCase
 
30
from bzrlib.symbol_versioning import (
 
31
    deprecated_in,
 
32
    )
28
33
 
29
34
 
30
35
class FakeStack:
 
36
 
31
37
    def __init__(self, top):
32
38
        self.__top = top
33
39
 
34
40
    def top(self):
35
41
        return self.__top
36
42
 
37
 
class InstrumentedProgress(TTYProgressBar):
38
 
    """TTYProgress variant that tracks outcomes"""
39
 
 
40
 
    def __init__(self, *args, **kwargs):
41
 
        self.always_throttled = True
42
 
        TTYProgressBar.__init__(self, *args, **kwargs)
43
 
 
44
 
    def throttle(self, old_message):
45
 
        result = TTYProgressBar.throttle(self, old_message)
46
 
        if result is False:
47
 
            self.always_throttled = False
48
 
        
49
43
 
50
44
class _TTYStringIO(StringIO):
51
45
    """A helper class which makes a StringIO look like a terminal"""
62
56
 
63
57
 
64
58
class TestProgress(TestCase):
 
59
 
65
60
    def setUp(self):
66
61
        q = DummyProgress()
67
62
        self.top = ChildProgress(_stack=FakeStack(q))
96
91
        self.assertEqual(self.top.child_fraction, 1)
97
92
 
98
93
    def test_implementations(self):
99
 
        for implementation in (TTYProgressBar, DotsProgressBar, 
 
94
        for implementation in (TTYProgressBar, DotsProgressBar,
100
95
                               DummyProgress):
101
96
            self.check_parent_handling(implementation)
102
97
 
114
109
        self.check_stack(DummyProgress, DummyProgress)
115
110
 
116
111
    def check_stack(self, parent_class, child_class):
117
 
        stack = ProgressBarStack(klass=parent_class, to_file=StringIO())
 
112
        stack = self.applyDeprecated(
 
113
            deprecated_in((1, 12, 0)),
 
114
            ProgressBarStack,
 
115
            klass=parent_class, to_file=StringIO())
118
116
        parent = stack.get_nested()
119
117
        try:
120
118
            self.assertIs(parent.__class__, parent_class)
220
218
 
221
219
        self.addCleanup(reset)
222
220
 
223
 
        stack = ProgressBarStack(to_file=outf)
 
221
        stack = self.applyDeprecated(
 
222
            deprecated_in((1, 12, 0)),
 
223
            ProgressBarStack,
 
224
            to_file=outf)
224
225
        pb = stack.get_nested()
225
226
        pb.start_time -= 1 # Make sure it is ready to write
226
227
        pb.width = 20 # And it is of reasonable size
243
244
                         '\r                   \r',
244
245
                         out.getvalue())
245
246
 
 
247
    def test_noninteractive_progress(self):
 
248
        out = _NonTTYStringIO()
 
249
        pb = self.get_nested(out, 'xterm')
 
250
        self.assertIsInstance(pb, DummyProgress)
 
251
        try:
 
252
            pb.update('foo', 1, 2)
 
253
            pb.update('bar', 2, 2)
 
254
        finally:
 
255
            pb.finished()
 
256
        self.assertEqual('', out.getvalue())
 
257
 
246
258
    def test_dots_progress(self):
247
 
        # Make sure the ProgressBarStack thinks it is
248
 
        # not writing out to a terminal, and thus uses a 
249
 
        # DotsProgressBar
 
259
        # make sure we get the right progress bar when not on a terminal
250
260
        out = _NonTTYStringIO()
251
 
        pb = self.get_nested(out, 'xterm')
 
261
        pb = self.get_nested(out, 'xterm', 'dots')
252
262
        self.assertIsInstance(pb, DotsProgressBar)
253
263
        try:
254
264
            pb.update('foo', 1, 2)
255
265
            pb.update('bar', 2, 2)
256
266
        finally:
257
267
            pb.finished()
258
 
 
259
268
        self.assertEqual('foo: .'
260
269
                         '\nbar: .'
261
270
                         '\n',
267
276
        out = cStringIO.StringIO()
268
277
        pb = self.get_nested(out, 'xterm')
269
278
        pb.finished()
270
 
        self.assertIsInstance(pb, DotsProgressBar)
 
279
        self.assertIsInstance(pb, DummyProgress)
271
280
 
272
281
    def test_dumb_progress(self):
273
 
        # Make sure the ProgressBarStack thinks it is writing out to a 
274
 
        # terminal, but it is the emacs 'dumb' terminal, so it uses
275
 
        # Dots
 
282
        # using a terminal that can't do cursor movement
276
283
        out = _TTYStringIO()
277
284
        pb = self.get_nested(out, 'dumb')
278
285
        pb.finished()
279
 
        self.assertIsInstance(pb, DotsProgressBar)
 
286
        self.assertIsInstance(pb, DummyProgress)
280
287
 
281
288
    def test_progress_env_tty(self):
282
289
        # The environ variable BZR_PROGRESS_BAR controls what type of
290
297
        # Even though we are not a tty, the env_var will override
291
298
        self.assertIsInstance(pb, TTYProgressBar)
292
299
 
293
 
    def test_progress_env_dots(self):
294
 
        # Even though we are in a tty, the env_var will override
295
 
        out = _TTYStringIO()
296
 
        pb = self.get_nested(out, 'xterm', 'dots')
297
 
        pb.finished()
298
 
        self.assertIsInstance(pb, DotsProgressBar)
299
 
 
300
300
    def test_progress_env_none(self):
301
301
        # Even though we are in a valid tty, no progress
302
302
        out = _TTYStringIO()