/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_ui.py

  • Committer: Martin Pool
  • Date: 2006-10-06 02:04:17 UTC
  • mfrom: (1908.10.1 bench_usecases.merge2)
  • mto: This revision was merged to the branch mainline in revision 2068.
  • Revision ID: mbp@sourcefrog.net-20061006020417-4949ca86f4417a4d
merge additional fix from cfbolz

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_ui.py
19
19
 
20
20
import os
21
21
from StringIO import StringIO
 
22
import re
22
23
import sys
23
24
 
24
 
 
 
25
import bzrlib
25
26
import bzrlib.errors as errors
26
27
from bzrlib.progress import TTYProgressBar, ProgressBarStack
27
28
from bzrlib.tests import TestCase
 
29
from bzrlib.tests.test_progress import _TTYStringIO
28
30
from bzrlib.ui import SilentUIFactory
29
31
from bzrlib.ui.text import TextUIFactory
30
32
 
 
33
 
31
34
class UITests(TestCase):
32
35
 
33
36
    def test_silent_factory(self):
34
 
 
35
37
        ui = SilentUIFactory()
36
38
        pb = ui.nested_progress_bar()
37
39
        try:
62
64
    def test_progress_note(self):
63
65
        stderr = StringIO()
64
66
        stdout = StringIO()
65
 
        ui_factory = TextUIFactory()
66
 
        pb = ui_factory.nested_progress_bar()
67
 
        try:
68
 
            pb.to_messages_file = stdout
69
 
            ui_factory._progress_bar_stack.bottom().to_file = stderr
 
67
        ui_factory = TextUIFactory(bar_type=TTYProgressBar)
 
68
        pb = ui_factory.nested_progress_bar()
 
69
        try:
 
70
            pb.to_messages_file = stdout
 
71
            ui_factory._progress_bar_stack.bottom().to_file = stderr
 
72
            result = pb.note('t')
 
73
            self.assertEqual(None, result)
 
74
            self.assertEqual("t\n", stdout.getvalue())
 
75
            # Since there was no update() call, there should be no clear() call
 
76
            self.failIf(re.search(r'^\r {10,}\r$', stderr.getvalue()) is not None,
 
77
                        'We cleared the stderr without anything to put there')
 
78
        finally:
 
79
            pb.finished()
 
80
 
 
81
    def test_progress_note_clears(self):
 
82
        stderr = StringIO()
 
83
        stdout = StringIO()
 
84
        # The PQM redirects the output to a file, so it
 
85
        # defaults to creating a Dots progress bar. we
 
86
        # need to force it to believe we are a TTY
 
87
        ui_factory = TextUIFactory(bar_type=TTYProgressBar)
 
88
        pb = ui_factory.nested_progress_bar()
 
89
        try:
 
90
            pb.to_messages_file = stdout
 
91
            ui_factory._progress_bar_stack.bottom().to_file = stderr
 
92
            # Create a progress update that isn't throttled
 
93
            pb.start_time -= 10
 
94
            pb.update('x', 1, 1)
70
95
            result = pb.note('t')
71
96
            self.assertEqual(None, result)
72
97
            self.assertEqual("t\n", stdout.getvalue())
73
98
            # the exact contents will depend on the terminal width and we don't
74
99
            # care about that right now - but you're probably running it on at
75
100
            # least a 10-character wide terminal :)
76
 
            self.assertContainsRe(stderr.getvalue(), r'^\r {10,}\r$')
 
101
            self.assertContainsRe(stderr.getvalue(), r'\r {10,}\r$')
77
102
        finally:
78
103
            pb.finished()
79
104
 
107
132
        self.assertRaises(errors.MissingProgressBarFinish, pb1.finished)
108
133
        pb2.finished()
109
134
        pb1.finished()
 
135
 
 
136
    def test_text_factory_setting_progress_bar(self):
 
137
        # we should be able to choose the progress bar type used.
 
138
        factory = bzrlib.ui.text.TextUIFactory(
 
139
            bar_type=bzrlib.progress.DotsProgressBar)
 
140
        bar = factory.nested_progress_bar()
 
141
        bar.finished()
 
142
        self.assertIsInstance(bar, bzrlib.progress.DotsProgressBar)
 
143
 
 
144
    def test_cli_stdin_is_default_stdin(self):
 
145
        factory = bzrlib.ui.CLIUIFactory()
 
146
        self.assertEqual(sys.stdin, factory.stdin)
 
147
 
 
148
    def assert_get_bool_acceptance_of_user_input(self, factory):
 
149
        factory.stdin = StringIO("y\nyes with garbage\nyes\nn\nnot an answer\nno\nfoo\n")
 
150
        factory.stdout = StringIO()
 
151
        # there is no output from the base factory
 
152
        self.assertEqual(True, factory.get_boolean(""))
 
153
        self.assertEqual(True, factory.get_boolean(""))
 
154
        self.assertEqual(False, factory.get_boolean(""))
 
155
        self.assertEqual(False, factory.get_boolean(""))
 
156
        self.assertEqual("foo\n", factory.stdin.read())
 
157
 
 
158
    def test_silent_ui_getbool(self):
 
159
        factory = bzrlib.ui.SilentUIFactory()
 
160
        self.assert_get_bool_acceptance_of_user_input(factory)
 
161
 
 
162
    def test_silent_factory_prompts_silently(self):
 
163
        factory = bzrlib.ui.SilentUIFactory()
 
164
        stdout = StringIO()
 
165
        factory.stdin = StringIO("y\n")
 
166
        self.assertEqual(
 
167
            True,
 
168
            self.apply_redirected(
 
169
                None, stdout, stdout, factory.get_boolean, "foo")
 
170
            )
 
171
        self.assertEqual("", stdout.getvalue())
 
172
        
 
173
    def test_text_ui_getbool(self):
 
174
        factory = bzrlib.ui.text.TextUIFactory()
 
175
        self.assert_get_bool_acceptance_of_user_input(factory)
 
176
 
 
177
    def test_text_factory_prompts_and_clears(self):
 
178
        # a get_boolean call should clear the pb before prompting
 
179
        factory = bzrlib.ui.text.TextUIFactory()
 
180
        factory.stdout = _TTYStringIO()
 
181
        factory.stdin = StringIO("yada\ny\n")
 
182
        pb = self.apply_redirected(
 
183
            factory.stdin, factory.stdout, factory.stdout, factory.nested_progress_bar)
 
184
        pb.start_time = None
 
185
        self.apply_redirected(
 
186
            factory.stdin, factory.stdout, factory.stdout, pb.update, "foo", 0, 1)
 
187
        self.assertEqual(
 
188
            True,
 
189
            self.apply_redirected(
 
190
                None, factory.stdout, factory.stdout, factory.get_boolean, "what do you want")
 
191
            )
 
192
        # use a regular expression so that we don't depend on the particular
 
193
        # screen width - could also set and restore $COLUMN if that has
 
194
        # priority on all platforms, but it doesn't at present.
 
195
        output = factory.stdout.getvalue()
 
196
        if not re.match(
 
197
            "\r/ \\[    *\\] foo 0/1"
 
198
            "\r   *" 
 
199
            "\rwhat do you want\\? \\[y/n\\]:what do you want\\? \\[y/n\\]:", 
 
200
            output):
 
201
            self.fail("didn't match factory output %r, %s" % (factory, output))