Fix tests and simplify helper methods

Author: tleonhardt

cmd2/cmd2.py

@@ -3235,12 +3235,12 @@ def _is_tty_session(session: PromptSession[str]) -> bool:
 
     def _add_to_pt_history(self, line: str) -> None:
         """Add a command to the prompt-toolkit UI history if it's a Cmd2History instance."""
-        if hasattr(self, 'main_session') and isinstance(self.main_session.history, Cmd2History):
+        if isinstance(self.main_session.history, Cmd2History):
             self.main_session.history.add_command(line)
 
     def _clear_pt_history(self) -> None:
         """Clear the prompt-toolkit UI history if it's a Cmd2History instance."""
-        if hasattr(self, 'main_session') and isinstance(self.main_session.history, Cmd2History):
+        if isinstance(self.main_session.history, Cmd2History):
             self.main_session.history.clear()
 
     def _read_raw_input(

tests/test_cmd2.py

@@ -1835,6 +1835,7 @@ def test_multiline_complete_statement_without_terminator(multiline_app, monkeypa
     args = 'hello world'
     line = f'{command} {args}'
     statement = multiline_app._complete_statement(line)
+    multiline_app._add_to_pt_history(statement.raw)
     assert statement == args
     assert statement.command == command
     assert statement.multiline_command
@@ -1849,6 +1850,7 @@ def test_multiline_complete_statement_with_unclosed_quotes(multiline_app, monkey
 
     line = 'orate hi "partially open'
     statement = multiline_app._complete_statement(line)
+    multiline_app._add_to_pt_history(statement.raw)
     assert statement == 'hi "partially open\nquotes\n" now closed'
     assert statement.command == 'orate'
     assert statement.multiline_command
@@ -1865,6 +1867,7 @@ def test_multiline_input_line_to_statement(multiline_app, monkeypatch) -> None:
 
     line = 'orate hi'
     statement = multiline_app._input_line_to_statement(line)
+    multiline_app._add_to_pt_history(statement.raw)
     assert statement.raw == 'orate hi\nperson\n\n'
     assert statement == 'hi person'
     assert statement.command == 'orate'
@@ -1933,6 +1936,7 @@ def test_multiline_complete_statement_eof(multiline_app, monkeypatch):
     # This should call _read_command_line, get 'eof', set nextline to '\n',
     # and then parse the line with the newline terminator.
     statement = multiline_app._complete_statement(line)
+    multiline_app._add_to_pt_history(statement.raw)
 
     assert statement.command == command
     assert statement.args == args