def test_register_command(self):
commands = get_commands()
before = len(commands)
class Foo:
name = 'foo'
register_command(Foo)
commands = get_commands()
after = len(commands)
try:
self.assertEqual(before + 1, after)
c_dict = {c.name: c for c in commands}
self.assertIn('foo', c_dict)
self.assertIs(c_dict['foo'], Foo)
finally:
remove_command(Foo)
commands = get_commands()
after2 = len(commands)
self.assertEqual(before, after2)
c_dict = {c.name: c for c in commands}
self.assertNotIn('foo', c_dict)
def test_available_commands(self):
commands = get_commands()
self.assertEqual(len(COMMAND_NAMES), len(commands))
c_list = [c.name for c in commands]
self.assertListEqual(COMMAND_NAMES, c_list)
def test_auto_register(self):
commands = get_commands()
before = len(commands)
class Foo(BaseCommand):
name = "foo"
def handle_xml(self, xml):
pass
after = len(commands)
try:
self.assertEqual(before + 1, after)
c_dict = {c.name: c for c in commands}
self.assertIn('foo', c_dict)
self.assertIs(c_dict['foo'], Foo)
finally:
remove_command(Foo)