60
73
self.assertRaises(errors.BzrCommandError,
61
74
commands.run_bzr, ['log', u'--option\xb5'])
77
def get_command(options):
78
class cmd_foo(commands.Command):
81
takes_options = options
85
def test_help_hidden(self):
86
c = self.get_command([option.Option('foo', hidden=True)])
87
self.assertNotContainsRe(c.get_help_text(), '--foo')
89
def test_help_not_hidden(self):
90
c = self.get_command([option.Option('foo', hidden=False)])
91
self.assertContainsRe(c.get_help_text(), '--foo')
94
class TestInvokedAs(tests.TestCase):
96
def test_invoked_as(self):
97
"""The command object knows the actual name used to invoke it."""
98
commands.install_bzr_command_hooks()
99
commands._register_builtin_commands()
100
# get one from the real get_cmd_object.
101
c = commands.get_cmd_object('ci')
102
self.assertIsInstance(c, builtins.cmd_commit)
103
self.assertEquals(c.invoked_as, 'ci')
64
106
class TestGetAlias(tests.TestCase):
66
108
def _get_config(self, config_text):
67
my_config = config.GlobalConfig()
68
config_file = StringIO(config_text.encode('utf-8'))
69
my_config._parser = my_config._get_parser(file=config_file)
109
my_config = config.GlobalConfig.from_string(config_text)
72
112
def test_simple(self):
101
141
class TestSeeAlso(tests.TestCase):
102
142
"""Tests for the see also functional of Command."""
145
def _get_command_with_see_also(see_also):
146
class ACommand(commands.Command):
147
__doc__ = """A sample command."""
104
151
def test_default_subclass_no_see_also(self):
105
class ACommand(commands.Command):
106
"""A sample command."""
152
command = self._get_command_with_see_also([])
108
153
self.assertEqual([], command.get_see_also())
110
155
def test__see_also(self):
111
156
"""When _see_also is defined, it sets the result of get_see_also()."""
112
class ACommand(commands.Command):
113
_see_also = ['bar', 'foo']
157
command = self._get_command_with_see_also(['bar', 'foo'])
115
158
self.assertEqual(['bar', 'foo'], command.get_see_also())
117
160
def test_deduplication(self):
118
161
"""Duplicates in _see_also are stripped out."""
119
class ACommand(commands.Command):
120
_see_also = ['foo', 'foo']
162
command = self._get_command_with_see_also(['foo', 'foo'])
122
163
self.assertEqual(['foo'], command.get_see_also())
124
165
def test_sorted(self):
125
166
"""_see_also is sorted by get_see_also."""
126
class ACommand(commands.Command):
127
_see_also = ['foo', 'bar']
167
command = self._get_command_with_see_also(['foo', 'bar'])
129
168
self.assertEqual(['bar', 'foo'], command.get_see_also())
131
170
def test_additional_terms(self):
132
171
"""Additional terms can be supplied and are deduped and sorted."""
133
class ACommand(commands.Command):
134
_see_also = ['foo', 'bar']
172
command = self._get_command_with_see_also(['foo', 'bar'])
136
173
self.assertEqual(['bar', 'foo', 'gam'],
137
174
command.get_see_also(['gam', 'bar', 'gam']))
186
225
# commands are registered).
187
226
# when they are simply created.
228
commands.install_bzr_command_hooks()
189
229
commands.Command.hooks.install_named_hook(
190
230
"extend_command", hook_calls.append, None)
191
231
# create a command, should not fire
192
class ACommand(commands.Command):
193
"""A sample command."""
232
class cmd_test_extend_command_hook(commands.Command):
233
__doc__ = """A sample command."""
195
234
self.assertEqual([], hook_calls)
196
235
# -- as a builtin
197
236
# register the command class, should not fire
199
builtins.cmd_test_extend_command_hook = ACommand
238
commands.builtin_command_registry.register(cmd_test_extend_command_hook)
200
239
self.assertEqual([], hook_calls)
201
240
# and ask for the object, should fire
202
241
cmd = commands.get_cmd_object('test-extend-command-hook')
218
257
self.assertEqual([cmd], hook_calls)
220
259
commands.plugin_cmds.remove('fake')
262
class TestGetCommandHook(tests.TestCase):
264
def test_fires_on_get_cmd_object(self):
265
# The get_command(cmd) hook fires when commands are delivered to the
267
commands.install_bzr_command_hooks()
269
class ACommand(commands.Command):
270
__doc__ = """A sample command."""
271
def get_cmd(cmd_or_None, cmd_name):
272
hook_calls.append(('called', cmd_or_None, cmd_name))
273
if cmd_name in ('foo', 'info'):
275
commands.Command.hooks.install_named_hook(
276
"get_command", get_cmd, None)
277
# create a command directly, should not fire
279
self.assertEqual([], hook_calls)
280
# ask by name, should fire and give us our command
281
cmd = commands.get_cmd_object('foo')
282
self.assertEqual([('called', None, 'foo')], hook_calls)
283
self.assertIsInstance(cmd, ACommand)
285
# ask by a name that is supplied by a builtin - the hook should still
286
# fire and we still get our object, but we should see the builtin
287
# passed to the hook.
288
cmd = commands.get_cmd_object('info')
289
self.assertIsInstance(cmd, ACommand)
290
self.assertEqual(1, len(hook_calls))
291
self.assertEqual('info', hook_calls[0][2])
292
self.assertIsInstance(hook_calls[0][1], builtins.cmd_info)
295
class TestGetMissingCommandHook(tests.TestCase):
297
def hook_missing(self):
298
"""Hook get_missing_command for testing."""
300
class ACommand(commands.Command):
301
__doc__ = """A sample command."""
302
def get_missing_cmd(cmd_name):
303
self.hook_calls.append(('called', cmd_name))
304
if cmd_name in ('foo', 'info'):
306
commands.Command.hooks.install_named_hook(
307
"get_missing_command", get_missing_cmd, None)
308
self.ACommand = ACommand
310
def test_fires_on_get_cmd_object(self):
311
# The get_missing_command(cmd) hook fires when commands are delivered to the
314
# create a command directly, should not fire
315
self.cmd = self.ACommand()
316
self.assertEqual([], self.hook_calls)
317
# ask by name, should fire and give us our command
318
cmd = commands.get_cmd_object('foo')
319
self.assertEqual([('called', 'foo')], self.hook_calls)
320
self.assertIsInstance(cmd, self.ACommand)
321
del self.hook_calls[:]
322
# ask by a name that is supplied by a builtin - the hook should not
323
# fire and we still get our object.
324
commands.install_bzr_command_hooks()
325
cmd = commands.get_cmd_object('info')
326
self.assertNotEqual(None, cmd)
327
self.assertEqual(0, len(self.hook_calls))
329
def test_skipped_on_HelpCommandIndex_get_topics(self):
330
# The get_missing_command(cmd_name) hook is not fired when
331
# looking up help topics.
333
topic = commands.HelpCommandIndex()
334
topics = topic.get_topics('foo')
335
self.assertEqual([], self.hook_calls)
338
class TestListCommandHook(tests.TestCase):
340
def test_fires_on_all_command_names(self):
341
# The list_commands() hook fires when all_command_names() is invoked.
343
commands.install_bzr_command_hooks()
344
def list_my_commands(cmd_names):
345
hook_calls.append('called')
346
cmd_names.update(['foo', 'bar'])
348
commands.Command.hooks.install_named_hook(
349
"list_commands", list_my_commands, None)
350
# Get a command, which should not trigger the hook.
351
cmd = commands.get_cmd_object('info')
352
self.assertEqual([], hook_calls)
353
# Get all command classes (for docs and shell completion).
354
cmds = list(commands.all_command_names())
355
self.assertEqual(['called'], hook_calls)
356
self.assertSubset(['foo', 'bar'], cmds)