6614.1.3
by Vincent Ladeuil
Fix assertEquals being deprecated by using assertEqual. |
1 |
# Copyright (C) 2005-2012, 2016 Canonical Ltd
|
1185.49.22
by John Arbash Meinel
Added get_password to the UIFactory, using it inside of sftp.py |
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
|
|
4183.7.1
by Sabin Iacob
update FSF mailing address |
15 |
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
1185.49.22
by John Arbash Meinel
Added get_password to the UIFactory, using it inside of sftp.py |
16 |
|
17 |
"""Tests for the bzrlib ui
|
|
18 |
"""
|
|
19 |
||
4017.1.1
by John Arbash Meinel
Get a pb.tick() to work after calling pb.update() |
20 |
import time |
1185.49.22
by John Arbash Meinel
Added get_password to the UIFactory, using it inside of sftp.py |
21 |
|
4797.40.2
by Martin Pool
Add missing import |
22 |
from StringIO import StringIO |
23 |
||
5416.1.11
by Martin Pool
Add ConfirmationUserInterfacePolicy that lets specific confirmations be configured off. |
24 |
from testtools.matchers import * |
25 |
||
4488.1.1
by Vincent Ladeuil
(vila) Cleanup imports in some test files |
26 |
from bzrlib import ( |
5230.1.3
by Martin Pool
Use configured output encoding for make_output_stream |
27 |
config, |
4634.144.5
by Martin Pool
Cleaner presentation and tests for warn_cross_format_fetch |
28 |
remote, |
4488.1.1
by Vincent Ladeuil
(vila) Cleanup imports in some test files |
29 |
tests, |
30 |
ui as _mod_ui, |
|
4449.3.4
by Martin Pool
ProgressTask now talks to ProgressView; easier to test |
31 |
)
|
5230.1.3
by Martin Pool
Use configured output encoding for make_output_stream |
32 |
from bzrlib.tests import ( |
33 |
fixtures, |
|
34 |
)
|
|
4597.3.36
by Vincent Ladeuil
Fix imports and various cleanups in test_ui. |
35 |
from bzrlib.ui import text as _mod_ui_text |
5422.1.5
by Martin Pool
Move ProgressRecordingUIFactory to bzrlib.tests.testui |
36 |
from bzrlib.tests.testui import ( |
5422.1.4
by Martin Pool
Rename CapturingUIFactory to ProgressRecordingUIFactory |
37 |
ProgressRecordingUIFactory, |
5422.1.1
by Martin Pool
Move CapturingUIFactory out of per_workingtree tests into somewhere reusable |
38 |
)
|
1185.49.22
by John Arbash Meinel
Added get_password to the UIFactory, using it inside of sftp.py |
39 |
|
1681.1.2
by Robert Collins
* bzrlib.ui.text.TextUIFactory now accepts a bar_type parameter which |
40 |
|
6437.57.4
by Martin Packman
Test helper spelling changes suggested in review by vila |
41 |
class TTYStringIO(StringIO): |
6437.57.1
by Martin Packman
Move StringIO subclasses from bt.test_progress to bt.test_ui where they're actually used |
42 |
"""A helper class which makes a StringIO look like a terminal"""
|
43 |
||
44 |
def isatty(self): |
|
45 |
return True |
|
46 |
||
47 |
||
6437.57.4
by Martin Packman
Test helper spelling changes suggested in review by vila |
48 |
class NonTTYStringIO(StringIO): |
6437.57.1
by Martin Packman
Move StringIO subclasses from bt.test_progress to bt.test_ui where they're actually used |
49 |
"""Helper that implements isatty() but returns False"""
|
50 |
||
51 |
def isatty(self): |
|
52 |
return False |
|
53 |
||
54 |
||
5230.1.3
by Martin Pool
Use configured output encoding for make_output_stream |
55 |
class TestUIConfiguration(tests.TestCaseWithTransport): |
56 |
||
57 |
def test_output_encoding_configuration(self): |
|
5230.1.5
by Martin Pool
Merge updated test fixtures |
58 |
enc = fixtures.generate_unicode_encodings().next() |
6499.3.8
by Vincent Ladeuil
Update some forgotten uses of GlobalConfig to GlobalStack. |
59 |
config.GlobalStack().set('output_encoding', enc) |
5230.1.3
by Martin Pool
Use configured output encoding for make_output_stream |
60 |
ui = tests.TestUIFactory(stdin=None, |
5230.1.5
by Martin Pool
Merge updated test fixtures |
61 |
stdout=tests.StringIOWrapper(), |
62 |
stderr=tests.StringIOWrapper()) |
|
5416.1.11
by Martin Pool
Add ConfirmationUserInterfacePolicy that lets specific confirmations be configured off. |
63 |
output = ui.make_output_stream() |
6614.1.3
by Vincent Ladeuil
Fix assertEquals being deprecated by using assertEqual. |
64 |
self.assertEqual(output.encoding, enc) |
5230.1.3
by Martin Pool
Use configured output encoding for make_output_stream |
65 |
|
66 |
||
4711.1.6
by Martin Pool
Separate TextUIFactory tests |
67 |
class TestTextUIFactory(tests.TestCase): |
2294.4.1
by Vincent Ladeuil
Add a UIFactory.get_login method, fix tests. |
68 |
|
5416.1.1
by Martin Pool
Use a structured ui_factory.confirm_action rather than just get_boolean |
69 |
def make_test_ui_factory(self, stdin_contents): |
70 |
ui = tests.TestUIFactory(stdin=stdin_contents, |
|
71 |
stdout=tests.StringIOWrapper(), |
|
72 |
stderr=tests.StringIOWrapper()) |
|
73 |
return ui |
|
74 |
||
75 |
def test_text_factory_confirm(self): |
|
76 |
# turns into reading a regular boolean
|
|
77 |
ui = self.make_test_ui_factory('n\n') |
|
6614.1.3
by Vincent Ladeuil
Fix assertEquals being deprecated by using assertEqual. |
78 |
self.assertEqual(ui.confirm_action(u'Should %(thing)s pass?', |
5416.1.1
by Martin Pool
Use a structured ui_factory.confirm_action rather than just get_boolean |
79 |
'bzrlib.tests.test_ui.confirmation', |
80 |
{'thing': 'this'},), |
|
81 |
False) |
|
82 |
||
2294.4.1
by Vincent Ladeuil
Add a UIFactory.get_login method, fix tests. |
83 |
def test_text_factory_ascii_password(self): |
5416.1.1
by Martin Pool
Use a structured ui_factory.confirm_action rather than just get_boolean |
84 |
ui = self.make_test_ui_factory('secret\n') |
2294.4.1
by Vincent Ladeuil
Add a UIFactory.get_login method, fix tests. |
85 |
pb = ui.nested_progress_bar() |
86 |
try: |
|
87 |
self.assertEqual('secret', |
|
88 |
self.apply_redirected(ui.stdin, ui.stdout, |
|
4368.3.1
by Vincent Ladeuil
Use stderr for UI prompt to address bug #376582. |
89 |
ui.stderr, |
2294.4.1
by Vincent Ladeuil
Add a UIFactory.get_login method, fix tests. |
90 |
ui.get_password)) |
91 |
# ': ' is appended to prompt
|
|
4368.3.1
by Vincent Ladeuil
Use stderr for UI prompt to address bug #376582. |
92 |
self.assertEqual(': ', ui.stderr.getvalue()) |
93 |
self.assertEqual('', ui.stdout.readline()) |
|
2363.4.3
by Vincent Ladeuil
Tidy-up tests. |
94 |
# stdin should be empty
|
2363.4.6
by Vincent Ladeuil
Fix tests around stdin emptyness. |
95 |
self.assertEqual('', ui.stdin.readline()) |
2294.4.1
by Vincent Ladeuil
Add a UIFactory.get_login method, fix tests. |
96 |
finally: |
97 |
pb.finished() |
|
98 |
||
99 |
def test_text_factory_utf8_password(self): |
|
6559.2.1
by Vincent Ladeuil
Makes AuthenticationConfig always return unicode user names and passwords. |
100 |
"""Test an utf8 password."""
|
101 |
ui = _mod_ui_text.TextUIFactory(None, None, None) |
|
102 |
ui.stdin = tests.StringIOWrapper(u'baz\u1234'.encode('utf8')) |
|
103 |
ui.stdout = tests.StringIOWrapper() |
|
104 |
ui.stderr = tests.StringIOWrapper() |
|
4368.3.1
by Vincent Ladeuil
Use stderr for UI prompt to address bug #376582. |
105 |
ui.stderr.encoding = ui.stdout.encoding = ui.stdin.encoding = 'utf8' |
6559.2.1
by Vincent Ladeuil
Makes AuthenticationConfig always return unicode user names and passwords. |
106 |
password = ui.get_password(u'Hello \u1234 %(user)s', user=u'some\u1234') |
107 |
self.assertEqual(u'baz\u1234', password) |
|
108 |
self.assertEqual(u'Hello \u1234 some\u1234: ', |
|
109 |
ui.stderr.getvalue().decode('utf8')) |
|
110 |
# stdin and stdout should be empty
|
|
111 |
self.assertEqual('', ui.stdin.readline()) |
|
112 |
self.assertEqual('', ui.stdout.getvalue()) |
|
1534.5.6
by Robert Collins
split out converter logic into per-format objects. |
113 |
|
4449.3.38
by Martin Pool
Cleanup get_boolean tests |
114 |
def test_text_ui_get_boolean(self): |
4773.1.2
by Vincent Ladeuil
Cleanup imports in test_ui |
115 |
stdin = tests.StringIOWrapper("y\n" # True |
116 |
"n\n" # False |
|
6182.2.4
by Benoît Pierre
Fix ui tests. |
117 |
" \n y \n" # True |
118 |
" no \n" # False |
|
4773.1.2
by Vincent Ladeuil
Cleanup imports in test_ui |
119 |
"yes with garbage\nY\n" # True |
120 |
"not an answer\nno\n" # False |
|
121 |
"I'm sure!\nyes\n" # True |
|
122 |
"NO\n" # False |
|
123 |
"foo\n") |
|
4597.3.36
by Vincent Ladeuil
Fix imports and various cleanups in test_ui. |
124 |
stdout = tests.StringIOWrapper() |
125 |
stderr = tests.StringIOWrapper() |
|
126 |
factory = _mod_ui_text.TextUIFactory(stdin, stdout, stderr) |
|
5863.6.1
by Jelmer Vernooij
Require a unicode prompt to be passed into all methods that prompt. |
127 |
self.assertEqual(True, factory.get_boolean(u"")) |
128 |
self.assertEqual(False, factory.get_boolean(u"")) |
|
129 |
self.assertEqual(True, factory.get_boolean(u"")) |
|
130 |
self.assertEqual(False, factory.get_boolean(u"")) |
|
131 |
self.assertEqual(True, factory.get_boolean(u"")) |
|
132 |
self.assertEqual(False, factory.get_boolean(u"")) |
|
6182.2.4
by Benoît Pierre
Fix ui tests. |
133 |
self.assertEqual(True, factory.get_boolean(u"")) |
134 |
self.assertEqual(False, factory.get_boolean(u"")) |
|
1687.1.4
by Robert Collins
Add bzrlib.ui.ui_factory.get_boolean(). |
135 |
self.assertEqual("foo\n", factory.stdin.read()) |
2363.4.4
by Vincent Ladeuil
More tidying-up. |
136 |
# stdin should be empty
|
2363.4.6
by Vincent Ladeuil
Fix tests around stdin emptyness. |
137 |
self.assertEqual('', factory.stdin.readline()) |
6182.2.4
by Benoît Pierre
Fix ui tests. |
138 |
# return false on EOF
|
139 |
self.assertEqual(False, factory.get_boolean(u"")) |
|
1687.1.4
by Robert Collins
Add bzrlib.ui.ui_factory.get_boolean(). |
140 |
|
6182.2.20
by Benoît Pierre
Add some more tests for TextUIFactory.choose. |
141 |
def test_text_ui_choose_bad_parameters(self): |
142 |
stdin = tests.StringIOWrapper() |
|
143 |
stdout = tests.StringIOWrapper() |
|
144 |
stderr = tests.StringIOWrapper() |
|
145 |
factory = _mod_ui_text.TextUIFactory(stdin, stdout, stderr) |
|
146 |
# invalid default index
|
|
147 |
self.assertRaises(ValueError, factory.choose, u"", u"&Yes\n&No", 3) |
|
148 |
# duplicated choice
|
|
149 |
self.assertRaises(ValueError, factory.choose, u"", u"&choice\n&ChOiCe") |
|
150 |
# duplicated shortcut
|
|
151 |
self.assertRaises(ValueError, factory.choose, u"", u"&choice1\nchoi&ce2") |
|
152 |
||
6182.2.22
by Benoît Pierre
More TextUIFactory.choose tests: check prompts. |
153 |
def test_text_ui_choose_prompt(self): |
154 |
stdin = tests.StringIOWrapper() |
|
155 |
stdout = tests.StringIOWrapper() |
|
156 |
stderr = tests.StringIOWrapper() |
|
157 |
factory = _mod_ui_text.TextUIFactory(stdin, stdout, stderr) |
|
158 |
# choices with explicit shortcuts
|
|
159 |
factory.choose(u"prompt", u"&yes\n&No\nmore &info") |
|
160 |
self.assertEqual("prompt ([y]es, [N]o, more [i]nfo): \n", factory.stderr.getvalue()) |
|
161 |
# automatic shortcuts
|
|
162 |
factory.stderr.truncate(0) |
|
163 |
factory.choose(u"prompt", u"yes\nNo\nmore info") |
|
164 |
self.assertEqual("prompt ([y]es, [N]o, [m]ore info): \n", factory.stderr.getvalue()) |
|
165 |
||
6182.2.17
by Benoît Pierre
Add some tests for TextUIFactory.choose. |
166 |
def test_text_ui_choose_return_values(self): |
167 |
choose = lambda: factory.choose(u"", u"&Yes\n&No\nMaybe\nmore &info", 3) |
|
168 |
stdin = tests.StringIOWrapper("y\n" # 0 |
|
169 |
"n\n" # 1 |
|
170 |
" \n" # default: 3 |
|
171 |
" no \n" # 1 |
|
6182.2.25
by Benoît Pierre
Tweak test_text_ui_choose_return_values a little. |
172 |
"b\na\nd \n" # bad shortcuts, all ignored |
6182.2.17
by Benoît Pierre
Add some tests for TextUIFactory.choose. |
173 |
"yes with garbage\nY\n" # 0 |
174 |
"not an answer\nno\n" # 1 |
|
175 |
"info\nmore info\n" # 3 |
|
176 |
"Maybe\n" # 2 |
|
177 |
"foo\n") |
|
178 |
stdout = tests.StringIOWrapper() |
|
179 |
stderr = tests.StringIOWrapper() |
|
180 |
factory = _mod_ui_text.TextUIFactory(stdin, stdout, stderr) |
|
181 |
self.assertEqual(0, choose()) |
|
182 |
self.assertEqual(1, choose()) |
|
183 |
self.assertEqual(3, choose()) |
|
184 |
self.assertEqual(1, choose()) |
|
185 |
self.assertEqual(0, choose()) |
|
186 |
self.assertEqual(1, choose()) |
|
187 |
self.assertEqual(3, choose()) |
|
188 |
self.assertEqual(2, choose()) |
|
189 |
self.assertEqual("foo\n", factory.stdin.read()) |
|
190 |
# stdin should be empty
|
|
191 |
self.assertEqual('', factory.stdin.readline()) |
|
192 |
# return None on EOF
|
|
193 |
self.assertEqual(None, choose()) |
|
194 |
||
6182.2.19
by Benoît Pierre
Add more tests for TextUIFactory.choose. |
195 |
def test_text_ui_choose_no_default(self): |
196 |
stdin = tests.StringIOWrapper(" \n" # no default, invalid! |
|
197 |
" yes \n" # 0 |
|
198 |
"foo\n") |
|
199 |
stdout = tests.StringIOWrapper() |
|
200 |
stderr = tests.StringIOWrapper() |
|
201 |
factory = _mod_ui_text.TextUIFactory(stdin, stdout, stderr) |
|
202 |
self.assertEqual(0, factory.choose(u"", u"&Yes\n&No")) |
|
203 |
self.assertEqual("foo\n", factory.stdin.read()) |
|
204 |
||
4597.3.37
by Vincent Ladeuil
Allows ui factories to query users for an integer. |
205 |
def test_text_ui_get_integer(self): |
206 |
stdin = tests.StringIOWrapper( |
|
207 |
"1\n" |
|
208 |
" -2 \n" |
|
209 |
"hmmm\nwhat else ?\nCome on\nok 42\n4.24\n42\n") |
|
210 |
stdout = tests.StringIOWrapper() |
|
211 |
stderr = tests.StringIOWrapper() |
|
212 |
factory = _mod_ui_text.TextUIFactory(stdin, stdout, stderr) |
|
5863.6.1
by Jelmer Vernooij
Require a unicode prompt to be passed into all methods that prompt. |
213 |
self.assertEqual(1, factory.get_integer(u"")) |
214 |
self.assertEqual(-2, factory.get_integer(u"")) |
|
215 |
self.assertEqual(42, factory.get_integer(u"")) |
|
4597.3.37
by Vincent Ladeuil
Allows ui factories to query users for an integer. |
216 |
|
4300.3.1
by Martin Pool
Fix string expansion in TextUIFactory.prompt |
217 |
def test_text_factory_prompt(self): |
218 |
# see <https://launchpad.net/bugs/365891>
|
|
4597.3.36
by Vincent Ladeuil
Fix imports and various cleanups in test_ui. |
219 |
StringIO = tests.StringIOWrapper |
220 |
factory = _mod_ui_text.TextUIFactory(StringIO(), StringIO(), StringIO()) |
|
5863.6.1
by Jelmer Vernooij
Require a unicode prompt to be passed into all methods that prompt. |
221 |
factory.prompt(u'foo %2e') |
4368.3.1
by Vincent Ladeuil
Use stderr for UI prompt to address bug #376582. |
222 |
self.assertEqual('', factory.stdout.getvalue()) |
223 |
self.assertEqual('foo %2e', factory.stderr.getvalue()) |
|
4300.3.1
by Martin Pool
Fix string expansion in TextUIFactory.prompt |
224 |
|
1687.1.4
by Robert Collins
Add bzrlib.ui.ui_factory.get_boolean(). |
225 |
def test_text_factory_prompts_and_clears(self): |
226 |
# a get_boolean call should clear the pb before prompting
|
|
6437.57.4
by Martin Packman
Test helper spelling changes suggested in review by vila |
227 |
out = TTYStringIO() |
5570.3.9
by Vincent Ladeuil
More use cases for overrideEnv, _cleanEnvironment *may* contain too much variables now. |
228 |
self.overrideEnv('TERM', 'xterm') |
4597.3.36
by Vincent Ladeuil
Fix imports and various cleanups in test_ui. |
229 |
factory = _mod_ui_text.TextUIFactory( |
230 |
stdin=tests.StringIOWrapper("yada\ny\n"), |
|
231 |
stdout=out, stderr=out) |
|
5339.2.6
by Martin Pool
One more UI test needs updates for spinner being at the front |
232 |
factory._avail_width = lambda: 79 |
3882.8.10
by Martin Pool
Fix up test_ui for new progress bars |
233 |
pb = factory.nested_progress_bar() |
234 |
pb.show_bar = False |
|
235 |
pb.show_spinner = False |
|
236 |
pb.show_count = False |
|
237 |
pb.update("foo", 0, 1) |
|
2363.4.4
by Vincent Ladeuil
More tidying-up. |
238 |
self.assertEqual(True, |
239 |
self.apply_redirected(None, factory.stdout, |
|
240 |
factory.stdout, |
|
241 |
factory.get_boolean, |
|
5863.6.1
by Jelmer Vernooij
Require a unicode prompt to be passed into all methods that prompt. |
242 |
u"what do you want")) |
3882.8.10
by Martin Pool
Fix up test_ui for new progress bars |
243 |
output = out.getvalue() |
5339.2.6
by Martin Pool
One more UI test needs updates for spinner being at the front |
244 |
self.assertContainsRe(output, |
245 |
"| foo *\r\r *\r*") |
|
6182.2.4
by Benoît Pierre
Fix ui tests. |
246 |
self.assertContainsString(output, |
247 |
r"what do you want? ([y]es, [n]o): what do you want? ([y]es, [n]o): ") |
|
3882.8.10
by Martin Pool
Fix up test_ui for new progress bars |
248 |
# stdin should have been totally consumed
|
2363.4.6
by Vincent Ladeuil
Fix tests around stdin emptyness. |
249 |
self.assertEqual('', factory.stdin.readline()) |
4017.1.1
by John Arbash Meinel
Get a pb.tick() to work after calling pb.update() |
250 |
|
251 |
def test_text_tick_after_update(self): |
|
4597.3.36
by Vincent Ladeuil
Fix imports and various cleanups in test_ui. |
252 |
ui_factory = _mod_ui_text.TextUIFactory(stdout=tests.StringIOWrapper(), |
253 |
stderr=tests.StringIOWrapper()) |
|
4017.1.1
by John Arbash Meinel
Get a pb.tick() to work after calling pb.update() |
254 |
pb = ui_factory.nested_progress_bar() |
255 |
try: |
|
256 |
pb.update('task', 0, 3) |
|
257 |
# Reset the clock, so that it actually tries to repaint itself
|
|
258 |
ui_factory._progress_view._last_repaint = time.time() - 1.0 |
|
259 |
pb.tick() |
|
260 |
finally: |
|
261 |
pb.finished() |
|
4110.2.15
by Martin Pool
Fix bug in showing task progress and add a test |
262 |
|
4222.2.1
by Jelmer Vernooij
Add get_username() call to the UIFactory. |
263 |
def test_text_ui_getusername(self): |
6559.2.1
by Vincent Ladeuil
Makes AuthenticationConfig always return unicode user names and passwords. |
264 |
ui = _mod_ui_text.TextUIFactory(None, None, None) |
265 |
ui.stdin = tests.StringIOWrapper('someuser\n\n') |
|
266 |
ui.stdout = tests.StringIOWrapper() |
|
267 |
ui.stderr = tests.StringIOWrapper() |
|
268 |
ui.stdout.encoding = 'utf8' |
|
269 |
self.assertEqual('someuser', |
|
270 |
ui.get_username(u'Hello %(host)s', host='some')) |
|
6614.1.3
by Vincent Ladeuil
Fix assertEquals being deprecated by using assertEqual. |
271 |
self.assertEqual('Hello some: ', ui.stderr.getvalue()) |
272 |
self.assertEqual('', ui.stdout.getvalue()) |
|
6559.2.1
by Vincent Ladeuil
Makes AuthenticationConfig always return unicode user names and passwords. |
273 |
self.assertEqual('', ui.get_username(u"Gebruiker")) |
4222.2.1
by Jelmer Vernooij
Add get_username() call to the UIFactory. |
274 |
# stdin should be empty
|
6559.2.1
by Vincent Ladeuil
Makes AuthenticationConfig always return unicode user names and passwords. |
275 |
self.assertEqual('', ui.stdin.readline()) |
4222.2.1
by Jelmer Vernooij
Add get_username() call to the UIFactory. |
276 |
|
4222.2.2
by Jelmer Vernooij
Review from vila: Deal with UTF8 strings in prompts, fix typo. |
277 |
def test_text_ui_getusername_utf8(self): |
6559.2.1
by Vincent Ladeuil
Makes AuthenticationConfig always return unicode user names and passwords. |
278 |
ui = _mod_ui_text.TextUIFactory(None, None, None) |
279 |
ui.stdin = tests.StringIOWrapper(u'someuser\u1234'.encode('utf8')) |
|
280 |
ui.stdout = tests.StringIOWrapper() |
|
281 |
ui.stderr = tests.StringIOWrapper() |
|
4368.3.1
by Vincent Ladeuil
Use stderr for UI prompt to address bug #376582. |
282 |
ui.stderr.encoding = ui.stdout.encoding = ui.stdin.encoding = "utf8" |
6559.2.1
by Vincent Ladeuil
Makes AuthenticationConfig always return unicode user names and passwords. |
283 |
username = ui.get_username(u'Hello %(host)s', host=u'some\u1234') |
6614.1.3
by Vincent Ladeuil
Fix assertEquals being deprecated by using assertEqual. |
284 |
self.assertEqual(u"someuser\u1234", username) |
285 |
self.assertEqual(u"Hello some\u1234: ", |
|
6559.2.1
by Vincent Ladeuil
Makes AuthenticationConfig always return unicode user names and passwords. |
286 |
ui.stderr.getvalue().decode("utf8")) |
6614.1.3
by Vincent Ladeuil
Fix assertEquals being deprecated by using assertEqual. |
287 |
self.assertEqual('', ui.stdout.getvalue()) |
4222.2.2
by Jelmer Vernooij
Review from vila: Deal with UTF8 strings in prompts, fix typo. |
288 |
|
4961.1.3
by Martin Pool
trace quietness now controls whether the progress bar appears |
289 |
def test_quietness(self): |
5570.3.9
by Vincent Ladeuil
More use cases for overrideEnv, _cleanEnvironment *may* contain too much variables now. |
290 |
self.overrideEnv('BZR_PROGRESS_BAR', 'text') |
4961.1.3
by Martin Pool
trace quietness now controls whether the progress bar appears |
291 |
ui_factory = _mod_ui_text.TextUIFactory(None, |
6437.57.4
by Martin Packman
Test helper spelling changes suggested in review by vila |
292 |
TTYStringIO(), |
293 |
TTYStringIO()) |
|
4961.1.3
by Martin Pool
trace quietness now controls whether the progress bar appears |
294 |
self.assertIsInstance(ui_factory._progress_view, |
295 |
_mod_ui_text.TextProgressView) |
|
296 |
ui_factory.be_quiet(True) |
|
297 |
self.assertIsInstance(ui_factory._progress_view, |
|
298 |
_mod_ui_text.NullProgressView) |
|
299 |
||
4634.144.10
by Martin Pool
Update test_ui for warning suppression |
300 |
def test_text_ui_show_user_warning(self): |
4634.144.5
by Martin Pool
Cleaner presentation and tests for warn_cross_format_fetch |
301 |
from bzrlib.repofmt.groupcompress_repo import RepositoryFormat2a |
5757.1.7
by Jelmer Vernooij
Fix more imports. |
302 |
from bzrlib.repofmt.knitpack_repo import RepositoryFormatKnitPack5 |
4634.144.5
by Martin Pool
Cleaner presentation and tests for warn_cross_format_fetch |
303 |
err = StringIO() |
304 |
out = StringIO() |
|
305 |
ui = tests.TextUIFactory(stdin=None, stdout=out, stderr=err) |
|
306 |
remote_fmt = remote.RemoteRepositoryFormat() |
|
307 |
remote_fmt._network_name = RepositoryFormatKnitPack5().network_name() |
|
4634.144.10
by Martin Pool
Update test_ui for warning suppression |
308 |
ui.show_user_warning('cross_format_fetch', from_format=RepositoryFormat2a(), |
309 |
to_format=remote_fmt) |
|
6614.1.3
by Vincent Ladeuil
Fix assertEquals being deprecated by using assertEqual. |
310 |
self.assertEqual('', out.getvalue()) |
311 |
self.assertEqual("Doing on-the-fly conversion from RepositoryFormat2a() to " |
|
4634.144.5
by Martin Pool
Cleaner presentation and tests for warn_cross_format_fetch |
312 |
"RemoteRepositoryFormat(_network_name='Bazaar RepositoryFormatKnitPack5 "
|
313 |
"(bzr 1.6)\\n').\nThis may take some time. Upgrade the repositories to " |
|
314 |
"the same format for better performance.\n", |
|
315 |
err.getvalue()) |
|
4634.144.10
by Martin Pool
Update test_ui for warning suppression |
316 |
# and now with it suppressed please
|
317 |
err = StringIO() |
|
318 |
out = StringIO() |
|
319 |
ui = tests.TextUIFactory(stdin=None, stdout=out, stderr=err) |
|
4634.144.11
by Martin Pool
Rename squelched to suppressed |
320 |
ui.suppressed_warnings.add('cross_format_fetch') |
4634.144.10
by Martin Pool
Update test_ui for warning suppression |
321 |
ui.show_user_warning('cross_format_fetch', from_format=RepositoryFormat2a(), |
322 |
to_format=remote_fmt) |
|
6614.1.3
by Vincent Ladeuil
Fix assertEquals being deprecated by using assertEqual. |
323 |
self.assertEqual('', out.getvalue()) |
324 |
self.assertEqual('', err.getvalue()) |
|
4634.144.5
by Martin Pool
Cleaner presentation and tests for warn_cross_format_fetch |
325 |
|
4110.2.15
by Martin Pool
Fix bug in showing task progress and add a test |
326 |
|
4884.1.1
by Vincent Ladeuil
Cleanup some test imports |
327 |
class TestTextUIOutputStream(tests.TestCase): |
4792.8.1
by Martin Pool
Add TextUIOutputStream coordinated with progress view |
328 |
"""Tests for output stream that synchronizes with progress bar."""
|
329 |
||
330 |
def test_output_clears_terminal(self): |
|
4884.1.1
by Vincent Ladeuil
Cleanup some test imports |
331 |
stdout = tests.StringIOWrapper() |
332 |
stderr = tests.StringIOWrapper() |
|
4792.8.1
by Martin Pool
Add TextUIOutputStream coordinated with progress view |
333 |
clear_calls = [] |
334 |
||
4884.1.1
by Vincent Ladeuil
Cleanup some test imports |
335 |
uif = _mod_ui_text.TextUIFactory(None, stdout, stderr) |
4792.8.1
by Martin Pool
Add TextUIOutputStream coordinated with progress view |
336 |
uif.clear_term = lambda: clear_calls.append('clear') |
337 |
||
4884.1.1
by Vincent Ladeuil
Cleanup some test imports |
338 |
stream = _mod_ui_text.TextUIOutputStream(uif, uif.stdout) |
4792.8.1
by Martin Pool
Add TextUIOutputStream coordinated with progress view |
339 |
stream.write("Hello world!\n") |
340 |
stream.write("there's more...\n") |
|
4792.8.3
by Martin Pool
Add TextUIOutputStream.writelines |
341 |
stream.writelines(["1\n", "2\n", "3\n"]) |
4884.1.1
by Vincent Ladeuil
Cleanup some test imports |
342 |
|
4792.8.1
by Martin Pool
Add TextUIOutputStream coordinated with progress view |
343 |
self.assertEqual(stdout.getvalue(), |
344 |
"Hello world!\n" |
|
4792.8.3
by Martin Pool
Add TextUIOutputStream.writelines |
345 |
"there's more...\n" |
346 |
"1\n2\n3\n") |
|
347 |
self.assertEqual(['clear', 'clear', 'clear'], |
|
4792.8.1
by Martin Pool
Add TextUIOutputStream coordinated with progress view |
348 |
clear_calls) |
349 |
||
4792.8.8
by Martin Pool
Add TextUIOutputStream.flush |
350 |
stream.flush() |
4792.8.1
by Martin Pool
Add TextUIOutputStream coordinated with progress view |
351 |
|
352 |
||
4711.1.6
by Martin Pool
Separate TextUIFactory tests |
353 |
class UITests(tests.TestCase): |
354 |
||
355 |
def test_progress_construction(self): |
|
356 |
"""TextUIFactory constructs the right progress view.
|
|
357 |
"""
|
|
4597.3.36
by Vincent Ladeuil
Fix imports and various cleanups in test_ui. |
358 |
FileStringIO = tests.StringIOWrapper |
4711.1.6
by Martin Pool
Separate TextUIFactory tests |
359 |
for (file_class, term, pb, expected_pb_class) in ( |
360 |
# on an xterm, either use them or not as the user requests,
|
|
361 |
# otherwise default on
|
|
4597.3.36
by Vincent Ladeuil
Fix imports and various cleanups in test_ui. |
362 |
(TTYStringIO, 'xterm', 'none', _mod_ui_text.NullProgressView), |
363 |
(TTYStringIO, 'xterm', 'text', _mod_ui_text.TextProgressView), |
|
364 |
(TTYStringIO, 'xterm', None, _mod_ui_text.TextProgressView), |
|
4711.1.6
by Martin Pool
Separate TextUIFactory tests |
365 |
# on a dumb terminal, again if there's explicit configuration do
|
366 |
# it, otherwise default off
|
|
4597.3.36
by Vincent Ladeuil
Fix imports and various cleanups in test_ui. |
367 |
(TTYStringIO, 'dumb', 'none', _mod_ui_text.NullProgressView), |
368 |
(TTYStringIO, 'dumb', 'text', _mod_ui_text.TextProgressView), |
|
369 |
(TTYStringIO, 'dumb', None, _mod_ui_text.NullProgressView), |
|
4711.1.6
by Martin Pool
Separate TextUIFactory tests |
370 |
# on a non-tty terminal, it's null regardless of $TERM
|
4597.3.36
by Vincent Ladeuil
Fix imports and various cleanups in test_ui. |
371 |
(FileStringIO, 'xterm', None, _mod_ui_text.NullProgressView), |
372 |
(FileStringIO, 'dumb', None, _mod_ui_text.NullProgressView), |
|
4711.1.6
by Martin Pool
Separate TextUIFactory tests |
373 |
# however, it can still be forced on
|
4597.3.36
by Vincent Ladeuil
Fix imports and various cleanups in test_ui. |
374 |
(FileStringIO, 'dumb', 'text', _mod_ui_text.TextProgressView), |
4711.1.6
by Martin Pool
Separate TextUIFactory tests |
375 |
):
|
5570.3.9
by Vincent Ladeuil
More use cases for overrideEnv, _cleanEnvironment *may* contain too much variables now. |
376 |
self.overrideEnv('TERM', term) |
377 |
self.overrideEnv('BZR_PROGRESS_BAR', pb) |
|
4711.1.6
by Martin Pool
Separate TextUIFactory tests |
378 |
stdin = file_class('') |
379 |
stderr = file_class() |
|
380 |
stdout = file_class() |
|
4597.3.36
by Vincent Ladeuil
Fix imports and various cleanups in test_ui. |
381 |
uif = _mod_ui.make_ui_for_terminal(stdin, stdout, stderr) |
382 |
self.assertIsInstance(uif, _mod_ui_text.TextUIFactory, |
|
4711.1.6
by Martin Pool
Separate TextUIFactory tests |
383 |
"TERM=%s BZR_PROGRESS_BAR=%s uif=%r" % (term, pb, uif,)) |
384 |
self.assertIsInstance(uif.make_progress_view(), |
|
385 |
expected_pb_class, |
|
386 |
"TERM=%s BZR_PROGRESS_BAR=%s uif=%r" % (term, pb, uif,)) |
|
387 |
||
388 |
def test_text_ui_non_terminal(self): |
|
389 |
"""Even on non-ttys, make_ui_for_terminal gives a text ui."""
|
|
6437.57.4
by Martin Packman
Test helper spelling changes suggested in review by vila |
390 |
stdin = NonTTYStringIO('') |
391 |
stderr = NonTTYStringIO() |
|
392 |
stdout = NonTTYStringIO() |
|
4711.1.6
by Martin Pool
Separate TextUIFactory tests |
393 |
for term_type in ['dumb', None, 'xterm']: |
5570.3.9
by Vincent Ladeuil
More use cases for overrideEnv, _cleanEnvironment *may* contain too much variables now. |
394 |
self.overrideEnv('TERM', term_type) |
4597.3.36
by Vincent Ladeuil
Fix imports and various cleanups in test_ui. |
395 |
uif = _mod_ui.make_ui_for_terminal(stdin, stdout, stderr) |
396 |
self.assertIsInstance(uif, _mod_ui_text.TextUIFactory, |
|
4711.1.6
by Martin Pool
Separate TextUIFactory tests |
397 |
'TERM=%r' % (term_type,)) |
398 |
||
399 |
||
4597.3.36
by Vincent Ladeuil
Fix imports and various cleanups in test_ui. |
400 |
class SilentUITests(tests.TestCase): |
4449.3.19
by Martin Pool
SilentUIFactory now always errors when asked for input |
401 |
|
4449.3.36
by Martin Pool
Update tests: SilentUIFactory no longer does get_boolean or get_password |
402 |
def test_silent_factory_get_password(self): |
403 |
# A silent factory that can't do user interaction can't get a
|
|
404 |
# password. Possibly it should raise a more specific error but it
|
|
405 |
# can't succeed.
|
|
4597.3.36
by Vincent Ladeuil
Fix imports and various cleanups in test_ui. |
406 |
ui = _mod_ui.SilentUIFactory() |
407 |
stdout = tests.StringIOWrapper() |
|
4449.3.19
by Martin Pool
SilentUIFactory now always errors when asked for input |
408 |
self.assertRaises( |
409 |
NotImplementedError, |
|
410 |
self.apply_redirected, |
|
411 |
None, stdout, stdout, ui.get_password) |
|
412 |
# and it didn't write anything out either
|
|
413 |
self.assertEqual('', stdout.getvalue()) |
|
414 |
||
415 |
def test_silent_ui_getbool(self): |
|
4597.3.36
by Vincent Ladeuil
Fix imports and various cleanups in test_ui. |
416 |
factory = _mod_ui.SilentUIFactory() |
417 |
stdout = tests.StringIOWrapper() |
|
4449.3.19
by Martin Pool
SilentUIFactory now always errors when asked for input |
418 |
self.assertRaises( |
419 |
NotImplementedError, |
|
420 |
self.apply_redirected, |
|
5863.6.1
by Jelmer Vernooij
Require a unicode prompt to be passed into all methods that prompt. |
421 |
None, stdout, stdout, factory.get_boolean, u"foo") |
4449.3.42
by Martin Pool
Add basic test for CannedInputUIFactory |
422 |
|
423 |
||
4597.3.36
by Vincent Ladeuil
Fix imports and various cleanups in test_ui. |
424 |
class TestUIFactoryTests(tests.TestCase): |
4580.2.2
by Martin Pool
Add test for bug 408201 |
425 |
|
426 |
def test_test_ui_factory_progress(self): |
|
427 |
# there's no output; we just want to make sure this doesn't crash -
|
|
5243.1.2
by Martin
Point launchpad links in comments at production server rather than edge |
428 |
# see https://bugs.launchpad.net/bzr/+bug/408201
|
4597.3.36
by Vincent Ladeuil
Fix imports and various cleanups in test_ui. |
429 |
ui = tests.TestUIFactory() |
4580.2.2
by Martin Pool
Add test for bug 408201 |
430 |
pb = ui.nested_progress_bar() |
431 |
pb.update('hello') |
|
432 |
pb.tick() |
|
433 |
pb.finished() |
|
434 |
||
435 |
||
4597.3.36
by Vincent Ladeuil
Fix imports and various cleanups in test_ui. |
436 |
class CannedInputUIFactoryTests(tests.TestCase): |
437 |
||
4449.3.42
by Martin Pool
Add basic test for CannedInputUIFactory |
438 |
def test_canned_input_get_input(self): |
4597.3.37
by Vincent Ladeuil
Allows ui factories to query users for an integer. |
439 |
uif = _mod_ui.CannedInputUIFactory([True, 'mbp', 'password', 42]) |
5863.6.1
by Jelmer Vernooij
Require a unicode prompt to be passed into all methods that prompt. |
440 |
self.assertEqual(True, uif.get_boolean(u'Extra cheese?')) |
441 |
self.assertEqual('mbp', uif.get_username(u'Enter your user name')) |
|
4597.3.37
by Vincent Ladeuil
Allows ui factories to query users for an integer. |
442 |
self.assertEqual('password', |
5863.6.1
by Jelmer Vernooij
Require a unicode prompt to be passed into all methods that prompt. |
443 |
uif.get_password(u'Password for %(host)s', |
4597.3.37
by Vincent Ladeuil
Allows ui factories to query users for an integer. |
444 |
host='example.com')) |
5863.6.1
by Jelmer Vernooij
Require a unicode prompt to be passed into all methods that prompt. |
445 |
self.assertEqual(42, uif.get_integer(u'And all that jazz ?')) |
4110.2.17
by Martin Pool
If one ProgressTask has no count, it passes through that of its child |
446 |
|
4503.2.1
by Vincent Ladeuil
Get a bool from a string. |
447 |
|
448 |
class TestBoolFromString(tests.TestCase): |
|
449 |
||
450 |
def assertIsTrue(self, s, accepted_values=None): |
|
451 |
res = _mod_ui.bool_from_string(s, accepted_values=accepted_values) |
|
6614.1.3
by Vincent Ladeuil
Fix assertEquals being deprecated by using assertEqual. |
452 |
self.assertEqual(True, res) |
4503.2.1
by Vincent Ladeuil
Get a bool from a string. |
453 |
|
454 |
def assertIsFalse(self, s, accepted_values=None): |
|
455 |
res = _mod_ui.bool_from_string(s, accepted_values=accepted_values) |
|
6614.1.3
by Vincent Ladeuil
Fix assertEquals being deprecated by using assertEqual. |
456 |
self.assertEqual(False, res) |
4503.2.1
by Vincent Ladeuil
Get a bool from a string. |
457 |
|
458 |
def assertIsNone(self, s, accepted_values=None): |
|
459 |
res = _mod_ui.bool_from_string(s, accepted_values=accepted_values) |
|
460 |
self.assertIs(None, res) |
|
461 |
||
462 |
def test_know_valid_values(self): |
|
463 |
self.assertIsTrue('true') |
|
464 |
self.assertIsFalse('false') |
|
465 |
self.assertIsTrue('1') |
|
466 |
self.assertIsFalse('0') |
|
467 |
self.assertIsTrue('on') |
|
468 |
self.assertIsFalse('off') |
|
469 |
self.assertIsTrue('yes') |
|
470 |
self.assertIsFalse('no') |
|
471 |
self.assertIsTrue('y') |
|
472 |
self.assertIsFalse('n') |
|
473 |
# Also try some case variations
|
|
474 |
self.assertIsTrue('True') |
|
475 |
self.assertIsFalse('False') |
|
476 |
self.assertIsTrue('On') |
|
477 |
self.assertIsFalse('Off') |
|
478 |
self.assertIsTrue('ON') |
|
479 |
self.assertIsFalse('OFF') |
|
480 |
self.assertIsTrue('oN') |
|
481 |
self.assertIsFalse('oFf') |
|
482 |
||
483 |
def test_invalid_values(self): |
|
484 |
self.assertIsNone(None) |
|
485 |
self.assertIsNone('doubt') |
|
486 |
self.assertIsNone('frue') |
|
487 |
self.assertIsNone('talse') |
|
488 |
self.assertIsNone('42') |
|
489 |
||
490 |
def test_provided_values(self): |
|
491 |
av = dict(y=True, n=False, yes=True, no=False) |
|
492 |
self.assertIsTrue('y', av) |
|
493 |
self.assertIsTrue('Y', av) |
|
494 |
self.assertIsTrue('Yes', av) |
|
495 |
self.assertIsFalse('n', av) |
|
496 |
self.assertIsFalse('N', av) |
|
497 |
self.assertIsFalse('No', av) |
|
498 |
self.assertIsNone('1', av) |
|
499 |
self.assertIsNone('0', av) |
|
500 |
self.assertIsNone('on', av) |
|
501 |
self.assertIsNone('off', av) |
|
5422.1.1
by Martin Pool
Move CapturingUIFactory out of per_workingtree tests into somewhere reusable |
502 |
|
503 |
||
5416.1.11
by Martin Pool
Add ConfirmationUserInterfacePolicy that lets specific confirmations be configured off. |
504 |
class TestConfirmationUserInterfacePolicy(tests.TestCase): |
505 |
||
506 |
def test_confirm_action_default(self): |
|
507 |
base_ui = _mod_ui.NoninteractiveUIFactory() |
|
508 |
for answer in [True, False]: |
|
6614.1.3
by Vincent Ladeuil
Fix assertEquals being deprecated by using assertEqual. |
509 |
self.assertEqual( |
5416.1.11
by Martin Pool
Add ConfirmationUserInterfacePolicy that lets specific confirmations be configured off. |
510 |
_mod_ui.ConfirmationUserInterfacePolicy(base_ui, answer, {}) |
511 |
.confirm_action("Do something?", |
|
512 |
"bzrlib.tests.do_something", {}), |
|
513 |
answer) |
|
514 |
||
515 |
def test_confirm_action_specific(self): |
|
516 |
base_ui = _mod_ui.NoninteractiveUIFactory() |
|
517 |
for default_answer in [True, False]: |
|
518 |
for specific_answer in [True, False]: |
|
519 |
for conf_id in ['given_id', 'other_id']: |
|
520 |
wrapper = _mod_ui.ConfirmationUserInterfacePolicy( |
|
521 |
base_ui, default_answer, dict(given_id=specific_answer)) |
|
522 |
result = wrapper.confirm_action("Do something?", conf_id, {}) |
|
523 |
if conf_id == 'given_id': |
|
6614.1.3
by Vincent Ladeuil
Fix assertEquals being deprecated by using assertEqual. |
524 |
self.assertEqual(result, specific_answer) |
5416.1.11
by Martin Pool
Add ConfirmationUserInterfacePolicy that lets specific confirmations be configured off. |
525 |
else: |
6614.1.3
by Vincent Ladeuil
Fix assertEquals being deprecated by using assertEqual. |
526 |
self.assertEqual(result, default_answer) |
5416.1.11
by Martin Pool
Add ConfirmationUserInterfacePolicy that lets specific confirmations be configured off. |
527 |
|
528 |
def test_repr(self): |
|
529 |
base_ui = _mod_ui.NoninteractiveUIFactory() |
|
530 |
wrapper = _mod_ui.ConfirmationUserInterfacePolicy( |
|
531 |
base_ui, True, dict(a=2)) |
|
532 |
self.assertThat(repr(wrapper), |
|
533 |
Equals("ConfirmationUserInterfacePolicy(" |
|
534 |
"NoninteractiveUIFactory(), True, {'a': 2})")) |
|
5416.2.4
by Martin Pool
resolve against trunk |
535 |
|
536 |
||
5422.1.4
by Martin Pool
Rename CapturingUIFactory to ProgressRecordingUIFactory |
537 |
class TestProgressRecordingUI(tests.TestCase): |
5422.1.1
by Martin Pool
Move CapturingUIFactory out of per_workingtree tests into somewhere reusable |
538 |
"""Test test-oriented UIFactory that records progress updates"""
|
539 |
||
540 |
def test_nested_ignore_depth_beyond_one(self): |
|
541 |
# we only want to capture the first level out progress, not
|
|
542 |
# want sub-components might do. So we have nested bars ignored.
|
|
5422.1.4
by Martin Pool
Rename CapturingUIFactory to ProgressRecordingUIFactory |
543 |
factory = ProgressRecordingUIFactory() |
5422.1.1
by Martin Pool
Move CapturingUIFactory out of per_workingtree tests into somewhere reusable |
544 |
pb1 = factory.nested_progress_bar() |
545 |
pb1.update('foo', 0, 1) |
|
546 |
pb2 = factory.nested_progress_bar() |
|
547 |
pb2.update('foo', 0, 1) |
|
548 |
pb2.finished() |
|
549 |
pb1.finished() |
|
550 |
self.assertEqual([("update", 0, 1, 'foo')], factory._calls) |