Added tests.

Author: kmvanbrunt

tests/test_cmd2.py

@@ -1203,6 +1203,67 @@ def test_ctrl_d_at_prompt(say_app, monkeypatch) -> None:
     assert out == 'hello\n\n'
 
 
+@pytest.mark.parametrize(
+    ('msg', 'prompt', 'is_stale'),
+    [
+        ("msg_text", None, False),
+        (None, "new_prompt> ", False),
+        ("msg_text", "new_prompt> ", True),
+        # Blank prompt is acceptable
+        ("msg_text", "", False),
+    ],
+)
+def test_async_alert(base_app, msg, prompt, is_stale) -> None:
+    import time
+
+    with mock.patch('cmd2.cmd2.print_formatted_text') as mock_print:
+        base_app.add_alert(msg=msg, prompt=prompt)
+        alert = base_app._alert_queue.get()
+
+        # Stale means alert was created before the current prompt.
+        if is_stale:
+            # In the past
+            alert.timestamp = 0.0
+        else:
+            # In the future
+            alert.timestamp = time.monotonic() + 99999999
+
+        base_app._alert_queue.put(alert)
+
+        with create_pipe_input() as pipe_input:
+            base_app.session = PromptSession(
+                input=pipe_input,
+                output=DummyOutput(),
+                history=base_app.session.history,
+                completer=base_app.session.completer,
+            )
+            pipe_input.send_text("quit\n")
+
+            base_app._cmdloop()
+
+            if msg:
+                assert msg in str(mock_print.call_args_list[0])
+            if prompt is not None:
+                if is_stale:
+                    assert base_app.prompt != prompt
+                else:
+                    assert base_app.prompt == prompt
+
+
+def test_add_alert(base_app) -> None:
+    orig_num_alerts = base_app._alert_queue.qsize()
+
+    # Nothing is added when both are None
+    base_app.add_alert(msg=None, prompt=None)
+    assert base_app._alert_queue.qsize() == orig_num_alerts
+
+    # Now test valid alert arguments
+    base_app.add_alert(msg="Hello", prompt=None)
+    base_app.add_alert(msg="Hello", prompt="prompt> ")
+    base_app.add_alert(msg=None, prompt="prompt> ")
+    assert base_app._alert_queue.qsize() == orig_num_alerts + 3
+
+
 class ShellApp(cmd2.Cmd):
     def __init__(self, *args, **kwargs) -> None:
         super().__init__(*args, **kwargs)

tests/test_pt_utils.py

@@ -18,8 +18,13 @@
     stylize,
     utils,
 )
+from cmd2 import rich_utils as ru
+from cmd2 import string_utils as su
 from cmd2.history import HistoryItem
 from cmd2.parsing import Statement
+from cmd2.pt_utils import pt_filter_style
+
+from .conftest import with_ansi_style
 
 
 # Mock for cmd2.Cmd
@@ -47,6 +52,48 @@ def mock_cmd_app() -> MockCmd:
     return MockCmd()
 
 
+@with_ansi_style(ru.AllowStyle.ALWAYS)
+def test_pt_filter_style_always() -> None:
+    """This should preserve all styles and return ANSI."""
+    unstyled = "unstyled"
+    result = pt_filter_style(unstyled)
+    assert isinstance(result, ANSI)
+    assert result.value == unstyled
+
+    styled = stylize("styled", Cmd2Style.COMMAND_LINE)
+    result = pt_filter_style(styled)
+    assert isinstance(result, ANSI)
+    assert result.value == styled
+
+
+@with_ansi_style(ru.AllowStyle.TERMINAL)
+def test_pt_filter_style_terminal() -> None:
+    """This should preserve all styles and return ANSI."""
+    unstyled = "unstyled"
+    result = pt_filter_style(unstyled)
+    assert isinstance(result, ANSI)
+    assert result.value == unstyled
+
+    styled = stylize("styled", Cmd2Style.COMMAND_LINE)
+    result = pt_filter_style(styled)
+    assert isinstance(result, ANSI)
+    assert result.value == styled
+
+
+@with_ansi_style(ru.AllowStyle.NEVER)
+def test_pt_filter_style_never() -> None:
+    """This should strip all styles and return str."""
+    unstyled = "unstyled"
+    result = pt_filter_style(unstyled)
+    assert isinstance(result, str)
+    assert result == unstyled
+
+    styled = stylize("styled", Cmd2Style.COMMAND_LINE)
+    result = pt_filter_style(styled)
+    assert isinstance(result, str)
+    assert result == su.strip_style(styled)
+
+
 class TestCmd2Lexer:
     def test_lex_document_command(self, mock_cmd_app):
         """Test lexing a command name."""