Function test_nested_subcommands

tests/test_commandset.py:673–715

(manual_command_sets_app)

def test_nested_subcommands(manual_command_sets_app) -> None:
    base_cmds = LoadableBase(1)
    pasta_cmds = LoadablePastaStir(1)

    with pytest.raises(CommandSetRegistrationError):
        manual_command_sets_app.register_command_set(pasta_cmds)

    manual_command_sets_app.register_command_set(base_cmds)

    manual_command_sets_app.register_command_set(pasta_cmds)

    with pytest.raises(CommandSetRegistrationError):
        manual_command_sets_app.unregister_command_set(base_cmds)

    class BadNestedSubcommands(cmd2.CommandSet):
        def __init__(self, dummy) -> None:
            super().__init__()
            self._dummy = dummy  # prevents autoload

        stir_pasta_vigor_parser = cmd2.Cmd2ArgumentParser()
        stir_pasta_vigor_parser.add_argument("frequency")

        # stir sauce doesn't exist anywhere, this should fail
        @cmd2.as_subcommand_to("stir sauce", "vigorously", stir_pasta_vigor_parser)
        def stir_pasta_vigorously(self, ns: argparse.Namespace) -> None:
            self._cmd.poutput("stir the pasta vigorously")

    with pytest.raises(CommandSetRegistrationError):
        manual_command_sets_app.register_command_set(BadNestedSubcommands(1))

    fruit_cmds = LoadableFruits(1)
    manual_command_sets_app.register_command_set(fruit_cmds)

    # validates custom namespace provider works correctly. Stir command will fail until
    # the cut command is called
    result = manual_command_sets_app.app_cmd("stir pasta vigorously everyminute")
    assert "Need to cut before stirring" in result.stdout

    result = manual_command_sets_app.app_cmd("cut banana discs")
    assert "cutting banana: discs" in result.stdout

    result = manual_command_sets_app.app_cmd("stir pasta vigorously everyminute")
    assert "stir the pasta vigorously" in result.stdout


class AppWithSubCommands(cmd2.Cmd):