Exemplo n.º 1
0
class AddSubcommandTest(TestCase):
    def setUp(self):
        self.cmd = CommandContainer()
        self.sub_cmd_1 = CommandContainer()
        self.sub_cmd_2 = CommandContainer()

        self.cmd.add_command("sub_cmd_1", self.sub_cmd_1)
        self.cmd.add_command("sub_cmd_2", self.sub_cmd_2)
        pass

    def test_it_returns_subcmd_count(self):
        self.assertTrue(len(self.cmd.get_command_names()) == 2)

    def test_it_returns_subcmd_names(self):
        self.assertTrue("sub_cmd_1" in self.cmd.get_command_names())
        self.assertTrue("sub_cmd_2" in self.cmd.get_command_names())
        self.assertFalse("sub_cmd_3" in self.cmd.get_command_names())

    def test_it_finds_subcmd(self):
        self.assertTrue(self.cmd.get_command("sub_cmd_1") == self.sub_cmd_1)

    def test_it_raises_exception_when_subcmd_not_noud(self):
        self.assertRaises(Exception, self.cmd.get_command, "unknown_sub_cmd")
Exemplo n.º 2
0
class AddSubcommandTest(TestCase):

    def setUp(self):
        self.cmd = CommandContainer()
        self.sub_cmd_1 = CommandContainer()
        self.sub_cmd_2 = CommandContainer()

        self.cmd.add_command("sub_cmd_1", self.sub_cmd_1)
        self.cmd.add_command("sub_cmd_2", self.sub_cmd_2)
        pass

    def test_it_returns_subcmd_count(self):
        self.assertTrue(len(self.cmd.get_command_names()) == 2)

    def test_it_returns_subcmd_names(self):
        self.assertTrue("sub_cmd_1" in self.cmd.get_command_names())
        self.assertTrue("sub_cmd_2" in self.cmd.get_command_names())
        self.assertFalse("sub_cmd_3" in self.cmd.get_command_names())

    def test_it_finds_subcmd(self):
        self.assertTrue(self.cmd.get_command("sub_cmd_1") == self.sub_cmd_1)

    def test_it_raises_exception_when_subcmd_not_noud(self):
        self.assertRaises(Exception, self.cmd.get_command, "unknown_sub_cmd")