Increased test coverage.

Author: kmvanbrunt

cmd2/completion.py

@@ -21,7 +21,7 @@
     overload,
 )
 
-if TYPE_CHECKING:
+if TYPE_CHECKING:  # pragma: no cover
     from .cmd2 import Cmd
     from .command_definition import CommandSet
 

cmd2/pt_utils.py

@@ -22,7 +22,7 @@
     utils,
 )
 
-if TYPE_CHECKING:
+if TYPE_CHECKING:  # pragma: no cover
     from .cmd2 import Cmd
 
 

tests/test_argparse_custom.py

@@ -10,7 +10,10 @@
     Cmd2ArgumentParser,
     constants,
 )
-from cmd2.argparse_custom import generate_range_error
+from cmd2.argparse_custom import (
+    ChoicesCallable,
+    generate_range_error,
+)
 
 from .conftest import run_cmd
 
@@ -75,6 +78,19 @@ def test_apcustom_no_choices_callables_when_nargs_is_0(kwargs) -> None:
     assert 'None of the following parameters can be used on an action that takes no arguments' in str(excinfo.value)
 
 
+def test_apcustom_choices_callables_wrong_property() -> None:
+    """Test using the wrong property when retrieving the to_call value from a ChoicesCallable."""
+    choices_callable = ChoicesCallable(is_completer=True, to_call=fake_func)
+    with pytest.raises(AttributeError) as excinfo:
+        to_call = choices_callable.choices_provider
+    assert 'This instance is configured as a completer' in str(excinfo.value)
+
+    choices_callable = ChoicesCallable(is_completer=False, to_call=fake_func)
+    with pytest.raises(AttributeError) as excinfo:
+        to_call = choices_callable.completer
+    assert 'This instance is configured as a choices_provider' in str(excinfo.value)
+
+
 def test_apcustom_usage() -> None:
     usage = "A custom usage statement"
     parser = Cmd2ArgumentParser(usage=usage)

tests/test_completion.py

@@ -947,7 +947,7 @@ def test_no_completer(cmd2_app) -> None:
     assert completions.to_strings() == Completions.from_values(expected).to_strings()
 
 
-def test_wordbreak_in_command(cmd2_app) -> None:
+def test_word_break_in_command(cmd2_app) -> None:
     text = ''
     line = f'"{text}'
     endidx = len(line)
@@ -983,6 +983,17 @@ def test_complete_multiline_on_multiple_lines(cmd2_app) -> None:
     assert completions.to_strings() == Completions.from_values(expected).to_strings()
 
 
+def test_completions_iteration() -> None:
+    items = [CompletionItem(1), CompletionItem(2)]
+    completions = Completions(items)
+
+    # Test __iter__
+    assert list(completions) == items
+
+    # Test __reversed__
+    assert list(reversed(completions)) == items[::-1]
+
+
 # Used by redirect_complete tests
 class RedirCompType(enum.Enum):
     SHELL_CMD = (1,)