Function test_subcommand_attachment

tests/test_cmd2.py:4521–4567

()

def test_subcommand_attachment() -> None:
    import argparse

    class SubcmdApp(cmd2.Cmd):
        def __init__(self) -> None:
            super().__init__()

        root_parser = cmd2.Cmd2ArgumentParser()
        root_parser.add_subparsers()

        @cmd2.with_argparser(root_parser)
        def do_root(self, _args: argparse.Namespace) -> None:
            pass

    app = SubcmdApp()

    # Verify root exists and uses argparse
    root_parser = app.command_parsers.get(app.do_root)
    assert root_parser is not None

    # Attach child to root
    child_parser = cmd2.Cmd2ArgumentParser(prog="child")
    child_parser.add_subparsers()
    app.attach_subcommand("root", "child", child_parser, help="child help")

    # Verify child was attached
    root_subparsers_action = root_parser.get_subparsers_action()
    assert "child" in root_subparsers_action._name_parser_map
    assert root_subparsers_action._name_parser_map["child"] is child_parser

    # Attach grandchild to child
    grandchild_parser = cmd2.Cmd2ArgumentParser(prog="grandchild")
    app.attach_subcommand("root child", "grandchild", grandchild_parser)

    # Verify grandchild was attached
    child_subparsers_action = child_parser.get_subparsers_action()
    assert "grandchild" in child_subparsers_action._name_parser_map

    # Detach grandchild
    detached_grandchild = app.detach_subcommand("root child", "grandchild")
    assert detached_grandchild is grandchild_parser
    assert "grandchild" not in child_subparsers_action._name_parser_map

    # Detach child
    detached_child = app.detach_subcommand("root", "child")
    assert detached_child is child_parser
    assert "child" not in root_subparsers_action._name_parser_map


def test_subcommand_attachment_errors() -> None: