Merge pull request #1795 from dbcli/RW/add-ptoolkit-utils-tests

Add tests for `mycli/packages/ptoolkit/utils.py`

Author: rolandwalker

changelog.md

@@ -20,7 +20,7 @@ Internal
 ---------
 * Add an `AGENTS.md`.
 * Refactor `find_matches()` into smaller logical units.
-* Increase test coverage.
+* Greatly increase test coverage.
 * Remove some unused code.
 * Better label Codex PR reviews.
 * Improve gitignored files.

test/pytests/test_ptoolkit_utils.py

@@ -0,0 +1,41 @@
+from dataclasses import dataclass, field
+from typing import Any, cast
+
+from mycli.packages.ptoolkit import utils as ptoolkit_utils
+
+
+@dataclass
+class DummyApp:
+    print_calls: list[str] = field(default_factory=list)
+
+    def print_text(self, text: str) -> None:
+        self.print_calls.append(text)
+
+
+def test_safe_invalidate_display_runs_empty_terminal_print(monkeypatch) -> None:
+    app = DummyApp()
+    callbacks: list[object] = []
+
+    def fake_run_in_terminal(callback) -> None:
+        callbacks.append(callback)
+        callback()
+
+    monkeypatch.setattr(ptoolkit_utils, 'run_in_terminal', fake_run_in_terminal)
+
+    ptoolkit_utils.safe_invalidate_display(cast(Any, app))
+
+    assert len(callbacks) == 1
+    assert app.print_calls == ['']
+
+
+def test_safe_invalidate_display_swallows_runtime_error(monkeypatch) -> None:
+    app = DummyApp()
+
+    def fail_run_in_terminal(_callback) -> None:
+        raise RuntimeError('application is exiting')
+
+    monkeypatch.setattr(ptoolkit_utils, 'run_in_terminal', fail_run_in_terminal)
+
+    ptoolkit_utils.safe_invalidate_display(cast(Any, app))
+
+    assert app.print_calls == []