35
47
def test_silent_factory(self):
36
48
ui = SilentUIFactory()
37
pb = ui.nested_progress_bar()
39
# TODO: Test that there is no output from SilentUIFactory
41
self.assertEquals(ui.get_password(), None)
42
self.assertEquals(ui.get_password(u'Hello There \u1234 %(user)s',
48
def test_text_factory(self):
50
pb = ui.nested_progress_bar()
52
# TODO: Test the output from TextUIFactory, perhaps by overriding sys.stdout
54
# Unfortunately we can't actually test the ui.get_password() because
55
# that would actually prompt the user for a password during the test suite
56
# This has been tested manually with both LANG=en_US.utf-8 and LANG=C
58
# self.assertEquals(ui.get_password(u"%(user)s please type 'bogus'",
50
self.assertEqual(None,
51
self.apply_redirected(None, stdout, stdout,
53
self.assertEqual('', stdout.getvalue())
54
self.assertEqual(None,
55
self.apply_redirected(None, stdout, stdout,
57
u'Hello\u1234 %(user)s',
59
self.assertEqual('', stdout.getvalue())
61
def test_text_factory_ascii_password(self):
62
ui = TestUIFactory(stdin='secret\n', stdout=StringIOWrapper())
63
pb = ui.nested_progress_bar()
65
self.assertEqual('secret',
66
self.apply_redirected(ui.stdin, ui.stdout,
69
# ': ' is appended to prompt
70
self.assertEqual(': ', ui.stdout.getvalue())
71
# stdin should be empty
72
self.assertEqual('', ui.stdin.readline())
76
def test_text_factory_utf8_password(self):
77
"""Test an utf8 password.
79
We can't predict what encoding users will have for stdin, so we force
80
it to utf8 to test that we transport the password correctly.
82
ui = TestUIFactory(stdin=u'baz\u1234'.encode('utf8'),
83
stdout=StringIOWrapper())
84
ui.stdin.encoding = 'utf8'
85
ui.stdout.encoding = ui.stdin.encoding
86
pb = ui.nested_progress_bar()
88
password = self.apply_redirected(ui.stdin, ui.stdout, ui.stdout,
90
u'Hello \u1234 %(user)s',
92
# We use StringIO objects, we need to decode them
93
self.assertEqual(u'baz\u1234', password.decode('utf8'))
94
self.assertEqual(u'Hello \u1234 some\u1234: ',
95
ui.stdout.getvalue().decode('utf8'))
96
# stdin should be empty
97
self.assertEqual('', ui.stdin.readline())
63
101
def test_progress_note(self):
64
102
stderr = StringIO()
65
103
stdout = StringIO()
66
ui_factory = TextUIFactory()
67
pb = ui_factory.nested_progress_bar()
69
pb.to_messages_file = stdout
70
ui_factory._progress_bar_stack.bottom().to_file = stderr
104
ui_factory = TextUIFactory(stdin=StringIO(''),
107
pb = ui_factory.nested_progress_bar()
109
result = pb.note('t')
110
self.assertEqual(None, result)
111
self.assertEqual("t\n", stdout.getvalue())
112
# Since there was no update() call, there should be no clear() call
113
self.failIf(re.search(r'^\r {10,}\r$',
114
stderr.getvalue()) is not None,
115
'We cleared the stderr without anything to put there')
119
def test_progress_note_clears(self):
122
# The PQM redirects the output to a file, so it
123
# defaults to creating a Dots progress bar. we
124
# need to force it to believe we are a TTY
125
ui_factory = TextUIFactory(
127
stdout=stdout, stderr=stderr)
128
pb = ui_factory.nested_progress_bar()
130
# Create a progress update that isn't throttled
71
132
result = pb.note('t')
72
133
self.assertEqual(None, result)
73
134
self.assertEqual("t\n", stdout.getvalue())
74
135
# the exact contents will depend on the terminal width and we don't
75
136
# care about that right now - but you're probably running it on at
76
137
# least a 10-character wide terminal :)
77
self.assertContainsRe(stderr.getvalue(), r'^\r {10,}\r$')
138
self.assertContainsRe(stderr.getvalue(), r'\r {10,}\r$')
81
142
def test_progress_nested(self):
82
143
# test factory based nested and popping.
144
ui = TextUIFactory(None, None, None)
84
145
pb1 = ui.nested_progress_bar()
85
146
pb2 = ui.nested_progress_bar()
86
self.assertRaises(errors.MissingProgressBarFinish, pb1.finished)
147
# We no longer warn about finishing unnested progress bars.
148
warnings, _ = self.callCatchWarnings(pb1.finished)
149
self.assertEqual(len(warnings), 0)
98
161
self.assertFalse(getattr(stack, 'note', False))
99
162
pb1 = stack.get_nested()
100
163
pb2 = stack.get_nested()
101
self.assertRaises(errors.MissingProgressBarFinish, pb1.finished)
164
warnings, _ = self.callCatchWarnings(pb1.finished)
165
self.assertEqual(len(warnings), 1)
104
168
# the text ui factory never actually removes the stack once its setup.
105
169
# we need to be able to nest again correctly from here.
106
170
pb1 = stack.get_nested()
107
171
pb2 = stack.get_nested()
108
self.assertRaises(errors.MissingProgressBarFinish, pb1.finished)
172
warnings, _ = self.callCatchWarnings(pb1.finished)
173
self.assertEqual(len(warnings), 1)
112
def test_text_factory_setting_progress_bar(self):
113
# we should be able to choose the progress bar type used.
114
factory = bzrlib.ui.text.TextUIFactory(
115
bar_type=bzrlib.progress.DotsProgressBar)
116
bar = factory.nested_progress_bar()
118
self.assertIsInstance(bar, bzrlib.progress.DotsProgressBar)
120
def test_cli_stdin_is_default_stdin(self):
121
factory = bzrlib.ui.CLIUIFactory()
122
self.assertEqual(sys.stdin, factory.stdin)
124
177
def assert_get_bool_acceptance_of_user_input(self, factory):
125
factory.stdin = StringIO("y\nyes with garbage\nyes\nn\nnot an answer\nno\nfoo\n")
178
factory.stdin = StringIO("y\nyes with garbage\n"
179
"yes\nn\nnot an answer\n"
126
181
factory.stdout = StringIO()
127
182
# there is no output from the base factory
128
183
self.assertEqual(True, factory.get_boolean(""))
130
185
self.assertEqual(False, factory.get_boolean(""))
131
186
self.assertEqual(False, factory.get_boolean(""))
132
187
self.assertEqual("foo\n", factory.stdin.read())
188
# stdin should be empty
189
self.assertEqual('', factory.stdin.readline())
134
191
def test_silent_ui_getbool(self):
135
factory = bzrlib.ui.SilentUIFactory()
192
factory = SilentUIFactory()
136
193
self.assert_get_bool_acceptance_of_user_input(factory)
138
195
def test_silent_factory_prompts_silently(self):
139
factory = bzrlib.ui.SilentUIFactory()
196
factory = SilentUIFactory()
140
197
stdout = StringIO()
141
198
factory.stdin = StringIO("y\n")
144
self.apply_redirected(
145
None, stdout, stdout, factory.get_boolean, "foo")
199
self.assertEqual(True,
200
self.apply_redirected(None, stdout, stdout,
201
factory.get_boolean, "foo"))
147
202
self.assertEqual("", stdout.getvalue())
203
# stdin should be empty
204
self.assertEqual('', factory.stdin.readline())
149
206
def test_text_ui_getbool(self):
150
factory = bzrlib.ui.text.TextUIFactory()
207
factory = TextUIFactory(None, None, None)
151
208
self.assert_get_bool_acceptance_of_user_input(factory)
153
210
def test_text_factory_prompts_and_clears(self):
154
211
# a get_boolean call should clear the pb before prompting
155
factory = bzrlib.ui.text.TextUIFactory()
156
factory.stdout = StringIO()
157
factory.stdin = StringIO("yada\ny\n")
158
pb = self.apply_redirected(
159
factory.stdin, factory.stdout, factory.stdout, factory.nested_progress_bar)
160
self.apply_redirected(
161
factory.stdin, factory.stdout, factory.stdout, pb.update, "foo", 0, 1)
164
self.apply_redirected(
165
None, factory.stdout, factory.stdout, factory.get_boolean, "what do you want")
167
# use a regular expression so that we don't depend on the particular
168
# screen width - could also set and restore $COLUMN if that has
169
# priority on all platforms, but it doesn't at present.
170
output = factory.stdout.getvalue()
172
"\r/ \\[ *\\] foo 0/1"
174
"\rwhat do you want\\? \\[y/n\\]:what do you want\\? \\[y/n\\]:",
176
self.fail("didn't match factory output %r" % factory)
213
factory = TextUIFactory(stdin=StringIO("yada\ny\n"), stdout=out, stderr=out)
214
pb = factory.nested_progress_bar()
216
pb.show_spinner = False
217
pb.show_count = False
218
pb.update("foo", 0, 1)
219
self.assertEqual(True,
220
self.apply_redirected(None, factory.stdout,
224
output = out.getvalue()
225
self.assertContainsRe(factory.stdout.getvalue(),
227
self.assertContainsRe(factory.stdout.getvalue(),
228
r"what do you want\? \[y/n\]: what do you want\? \[y/n\]: ")
229
# stdin should have been totally consumed
230
self.assertEqual('', factory.stdin.readline())
added
removed
bzrlib/tests/test_ui.py
