chore: fix test

Author: KelvinChung2000

cmd2/decorators.py

@@ -394,7 +394,6 @@ def team_create(self, name: str): ...
         build_parser_from_function,
         build_subcommand_handler,
     )
-    from .argparse_custom import Cmd2AttributeWrapper
 
     if (help is not None or aliases is not None) and subcommand_to is None:
         raise TypeError("'help' and 'aliases' are only valid with subcommand_to")
@@ -487,13 +486,11 @@ def cmd_wrapper(*args: Any, **kwargs: Any) -> bool | None:
             except SystemExit as exc:
                 raise Cmd2ArgparseError from exc
 
-            ns.cmd2_statement = Cmd2AttributeWrapper(statement)
+            setattr(ns, constants.NS_ATTR_STATEMENT, statement)
             handler = getattr(ns, constants.NS_ATTR_SUBCMD_HANDLER, None)
             if base_command and handler is not None:
                 handler = functools.partial(handler, ns)
-            ns.cmd2_handler = Cmd2AttributeWrapper(handler)
-            if hasattr(ns, constants.NS_ATTR_SUBCMD_HANDLER):
-                delattr(ns, constants.NS_ATTR_SUBCMD_HANDLER)
+            ns.cmd2_handler = handler
 
             func_kwargs = _filtered_namespace_kwargs(ns, accepted=accepted, exclude_subcommand=base_command)
 

docs/features/argument_processing.md

@@ -217,7 +217,7 @@ def do_rawish(self, name: str, _unknown: list[str] | None = None):
 ```py
 @with_annotated(base_command=True)
 def do_manage(self, *, cmd2_handler):
-    handler = cmd2_handler.get()
+    handler = cmd2_handler
     if handler:
         handler()
 
@@ -233,7 +233,7 @@ creates its own subparsers:
 ```py
 @with_annotated(subcommand_to="manage", base_command=True, help="manage projects")
 def manage_project(self, *, cmd2_handler):
-    handler = cmd2_handler.get()
+    handler = cmd2_handler
     if handler:
         handler()
 

tests/test_annotated.py

@@ -1204,7 +1204,7 @@ def do_manage(self, cmd2_handler, verbose: bool = False) -> None:
         """Management command with subcommands."""
         if verbose:
             self.poutput("verbose mode")
-        handler = cmd2_handler.get()
+        handler = cmd2_handler
         if handler:
             handler()
 
@@ -1220,7 +1220,7 @@ def manage_list(self) -> None:
     # Level 2: intermediate subcommand (also a base for level 3)
     @cmd2.with_annotated(subcommand_to='manage', base_command=True, help='manage members')
     def manage_member(self, cmd2_handler) -> None:
-        handler = cmd2_handler.get()
+        handler = cmd2_handler
         if handler:
             handler()
 
@@ -1328,7 +1328,7 @@ def test_subcommand_with_mutually_exclusive_groups(self) -> None:
         class App(cmd2.Cmd):
             @cmd2.with_annotated(base_command=True)
             def do_fmt(self, cmd2_handler) -> None:
-                handler = cmd2_handler.get()
+                handler = cmd2_handler
                 if handler:
                     handler()