Function test_update_prog

tests/test_argparse_utils.py:559–620

Test Cmd2ArgumentParser.update_prog() across various scenarios.

()

def test_update_prog() -> None:
    """Test Cmd2ArgumentParser.update_prog() across various scenarios."""

    # Set up a complex parser hierarchy
    old_app = "old_app"
    root = Cmd2ArgumentParser(prog=old_app)

    # Positionals before subcommand
    root.add_argument("pos1")

    # Mutually exclusive group with positionals
    group = root.add_mutually_exclusive_group(required=True)
    group.add_argument("posA", nargs="?")
    group.add_argument("posB", nargs="?")

    # Subparsers with aliases and no help text
    root_subparsers = root.add_subparsers(dest="cmd")

    # Subcommand with aliases
    sub1 = root_subparsers.add_parser("sub1", aliases=["s1", "alias1"], help="help for sub1")

    # Subcommand with no help text
    sub2 = root_subparsers.add_parser("sub2")

    # Nested subparser
    sub2.add_argument("inner_pos")
    sub2_subparsers = sub2.add_subparsers(dest="sub2_cmd")
    leaf = sub2_subparsers.add_parser("leaf", help="leaf help")

    # Save initial prog values
    orig_root_prog = root.prog
    orig_sub1_prog = sub1.prog
    orig_sub2_prog = sub2.prog
    orig_leaf_prog = leaf.prog

    # Perform update
    new_app = "new_app"
    root.update_prog(new_app)

    # Verify updated prog values
    assert root.prog.startswith(new_app)
    assert root.prog == orig_root_prog.replace(old_app, new_app, 1)

    assert sub1.prog.startswith(new_app)
    assert sub1.prog == orig_sub1_prog.replace(old_app, new_app, 1)

    assert sub2.prog.startswith(new_app)
    assert sub2.prog == orig_sub2_prog.replace(old_app, new_app, 1)

    assert leaf.prog.startswith(new_app)
    assert leaf.prog == orig_leaf_prog.replace(old_app, new_app, 1)

    # Verify that action._prog_prefix was updated by adding a new subparser
    sub3 = root_subparsers.add_parser("sub3")
    assert sub3.prog.startswith(new_app)
    assert sub3.prog == root_subparsers._prog_prefix + " sub3"

    # Verify aliases still point to the correct parser