/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_commands.py

  • Committer: Robert Collins
  • Date: 2009-05-23 21:01:51 UTC
  • mto: This revision was merged to the branch mainline in revision 4441.
  • Revision ID: robertc@robertcollins.net-20090523210151-69jmrka5l4eh0zf3
Get missing command support sorted out.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_commands.py
163
163
        del sys.modules['bzrlib.tests.fake_command']
164
164
        global lazy_command_imported
165
165
        lazy_command_imported = False
 
166
        commands.install_bzr_command_hooks()
166
167
 
167
168
    @staticmethod
168
169
    def remove_fake():
205
206
        # commands are registered).
206
207
        # when they are simply created.
207
208
        hook_calls = []
 
209
        commands.install_bzr_command_hooks()
208
210
        commands.Command.hooks.install_named_hook(
209
211
            "extend_command", hook_calls.append, None)
210
212
        # create a command, should not fire
244
246
    def test_fires_on_get_cmd_object(self):
245
247
        # The get_command(cmd) hook fires when commands are delivered to the
246
248
        # ui.
 
249
        commands.install_bzr_command_hooks()
247
250
        hook_calls = []
248
251
        class ACommand(commands.Command):
249
252
            """A sample command."""
271
274
        self.assertIsInstance(hook_calls[0][1], builtins.cmd_info)
272
275
 
273
276
 
 
277
class TestGetMissingCommandHook(tests.TestCase):
 
278
 
 
279
    def test_fires_on_get_cmd_object(self):
 
280
        # The get_missing_command(cmd) hook fires when commands are delivered to the
 
281
        # ui.
 
282
        hook_calls = []
 
283
        class ACommand(commands.Command):
 
284
            """A sample command."""
 
285
        def get_missing_cmd(cmd_name):
 
286
            hook_calls.append(('called', cmd_name))
 
287
            if cmd_name in ('foo', 'info'):
 
288
                return ACommand()
 
289
        commands.Command.hooks.install_named_hook(
 
290
            "get_missing_command", get_missing_cmd, None)
 
291
        # create a command directly, should not fire
 
292
        cmd = ACommand()
 
293
        self.assertEqual([], hook_calls)
 
294
        # ask by name, should fire and give us our command
 
295
        cmd = commands.get_cmd_object('foo')
 
296
        self.assertEqual([('called', 'foo')], hook_calls)
 
297
        self.assertIsInstance(cmd, ACommand)
 
298
        del hook_calls[:]
 
299
        # ask by a name that is supplied by a builtin - the hook should not
 
300
        # fire and we still get our object.
 
301
        commands.install_bzr_command_hooks()
 
302
        cmd = commands.get_cmd_object('info')
 
303
        self.assertNotEqual(None, cmd)
 
304
        self.assertEqual(0, len(hook_calls))
 
305
 
 
306
 
274
307
class TestListCommandHook(tests.TestCase):
275
308
 
276
309
    def test_fires_on_all_command_names(self):
277
310
        # The list_commands() hook fires when all_command_names() is invoked.
278
311
        hook_calls = []
 
312
        commands.install_bzr_command_hooks()
279
313
        def list_my_commands(cmd_names):
280
314
            hook_calls.append('called')
281
315
            cmd_names.update(['foo', 'bar'])