Updated tests.

Author: kmvanbrunt

tests/test_cmd2.py

@@ -1208,19 +1208,33 @@ def test_ctrl_d_at_prompt(say_app, monkeypatch) -> None:
     reason="Don't have a real Windows console with how we are currently running tests in GitHub Actions",
 )
 @pytest.mark.parametrize(
-    ('msg', 'prompt', 'is_stale'),
+    ('msg', 'prompt', 'is_stale', 'at_continuation_prompt'),
     [
-        ("msg_text", None, False),
-        (None, "new_prompt> ", False),
-        ("msg_text", "new_prompt> ", True),
+        ("msg_text", None, False, False),
+        ("msg_text", "new_prompt> ", False, False),
+        ("msg_text", "new_prompt> ", False, True),
+        ("msg_text", "new_prompt> ", True, False),
+        ("msg_text", "new_prompt> ", True, True),
+        (None, "new_prompt> ", False, False),
+        (None, "new_prompt> ", False, True),
+        (None, "new_prompt> ", True, False),
+        (None, "new_prompt> ", True, True),
         # Blank prompt is acceptable
-        ("msg_text", "", False),
+        ("msg_text", "", False, False),
+        (None, "", False, False),
     ],
 )
-def test_async_alert(base_app, msg, prompt, is_stale) -> None:
+def test_async_alert(base_app, msg, prompt, is_stale, at_continuation_prompt) -> None:
     import time
 
-    with mock.patch('cmd2.cmd2.print_formatted_text') as mock_print:
+    with (
+        mock.patch('cmd2.cmd2.print_formatted_text') as mock_print,
+        mock.patch('cmd2.cmd2.get_app') as mock_get_app,
+    ):
+        # Set up the chained mock: get_app() returns mock_app, which has invalidate()
+        mock_app = mock.MagicMock()
+        mock_get_app.return_value = mock_app
+
         base_app.add_alert(msg=msg, prompt=prompt)
         alert = base_app._alert_queue[0]
 
@@ -1232,6 +1246,8 @@ def test_async_alert(base_app, msg, prompt, is_stale) -> None:
             # In the future
             alert.timestamp = time.monotonic() + 99999999
 
+        base_app._at_continuation_prompt = at_continuation_prompt
+
         with create_pipe_input() as pipe_input:
             base_app.session = PromptSession(
                 input=pipe_input,
@@ -1243,8 +1259,19 @@ def test_async_alert(base_app, msg, prompt, is_stale) -> None:
 
             base_app._cmdloop()
 
+            # If there was a message, patch_stdout handles the redraw (no invalidate)
             if msg:
                 assert msg in str(mock_print.call_args_list[0])
+                mock_app.invalidate.assert_not_called()
+
+            # If there's only a prompt update, we expect invalidate() only if not continuation/stale
+            elif prompt is not None:
+                if is_stale or at_continuation_prompt:
+                    mock_app.invalidate.assert_not_called()
+                else:
+                    mock_app.invalidate.assert_called_once()
+
+            # The state of base_app.prompt should always be correct regardless of redraw
             if prompt is not None:
                 if is_stale:
                     assert base_app.prompt != prompt