def test_register_commands(self):
"""register subcommands."""
registry = CommandRegistry()
registry.register_commands(globals().values(), parent=MockSubcommand)
self.assertItemsEqual(registry.commands.keys(), ["one", "two", "help"])
self.assertIsInstance(registry.commands['one'], One)
self.assertIsInstance(registry.commands['two'], Two)
def setUp(self):
self.registry = CommandRegistry()
self.one = local_subclass(One)
self.two = local_subclass(Two)
self.registry.register_command(self.one)
self.registry.register_command(self.two)
self.result = argparse.Namespace()
Bcfg2.Options.Subcommands.master_setup = self.result
new_parser()
self.parser = get_parser(namespace=self.result, components=[self])
self.parser.add_options(self.registry.subcommand_options)
