From b8bf81918292a7a7fabc1b20a2109eb80766d02f Mon Sep 17 00:00:00 2001
From: thePR0M3TH3AN <53631862+PR0M3TH3AN@users.noreply.github.com>
Date: Tue, 1 Jul 2025 13:23:43 -0400
Subject: [PATCH] Add CLI invalid input tests

---
 src/tests/test_cli_invalid_input.py | 100 ++++++++++++++++++++++++++++
 1 file changed, 100 insertions(+)
 create mode 100644 src/tests/test_cli_invalid_input.py

diff --git a/src/tests/test_cli_invalid_input.py b/src/tests/test_cli_invalid_input.py
new file mode 100644
index 0000000..d4f17b1
--- /dev/null
+++ b/src/tests/test_cli_invalid_input.py
@@ -0,0 +1,100 @@
+import sys
+import time
+from types import SimpleNamespace
+from pathlib import Path
+
+import pytest
+
+sys.path.append(str(Path(__file__).resolve().parents[1]))
+
+import main
+
+
+def _make_pm(called, locked=None):
+    if locked is None:
+        locked = {"lock": 0, "unlock": 0}
+
+    def add():
+        called["add"] = True
+
+    def retrieve():
+        called["retrieve"] = True
+
+    def modify():
+        called["modify"] = True
+
+    def update():
+        pm.last_activity = time.time()
+
+    def lock():
+        locked["lock"] += 1
+
+    def unlock():
+        locked["unlock"] += 1
+        update()
+
+    pm = SimpleNamespace(
+        is_dirty=False,
+        last_update=time.time(),
+        last_activity=time.time(),
+        nostr_client=SimpleNamespace(close_client_pool=lambda: None),
+        handle_add_password=add,
+        handle_retrieve_entry=retrieve,
+        handle_modify_entry=modify,
+        update_activity=update,
+        lock_vault=lock,
+        unlock_vault=unlock,
+    )
+    return pm, locked
+
+
+def test_empty_and_non_numeric_choice(monkeypatch, capsys):
+    called = {"add": False, "retrieve": False, "modify": False}
+    pm, _ = _make_pm(called)
+    inputs = iter(["", "abc", "5"])
+    monkeypatch.setattr("builtins.input", lambda *_: next(inputs))
+    with pytest.raises(SystemExit):
+        main.display_menu(pm, sync_interval=1000, inactivity_timeout=1000)
+    out = capsys.readouterr().out
+    assert "No input detected" in out
+    assert "Invalid choice. Please select a valid option." in out
+    assert not any(called.values())
+
+
+def test_out_of_range_menu(monkeypatch, capsys):
+    called = {"add": False, "retrieve": False, "modify": False}
+    pm, _ = _make_pm(called)
+    inputs = iter(["9", "5"])
+    monkeypatch.setattr("builtins.input", lambda *_: next(inputs))
+    with pytest.raises(SystemExit):
+        main.display_menu(pm, sync_interval=1000, inactivity_timeout=1000)
+    out = capsys.readouterr().out
+    assert "Invalid choice. Please select a valid option." in out
+    assert not any(called.values())
+
+
+def test_invalid_add_entry_submenu(monkeypatch, capsys):
+    called = {"add": False, "retrieve": False, "modify": False}
+    pm, _ = _make_pm(called)
+    inputs = iter(["1", "3", "2", "5"])
+    monkeypatch.setattr("builtins.input", lambda *_: next(inputs))
+    with pytest.raises(SystemExit):
+        main.display_menu(pm, sync_interval=1000, inactivity_timeout=1000)
+    out = capsys.readouterr().out
+    assert "Invalid choice." in out
+    assert not any(called.values())
+
+
+def test_inactivity_timeout_loop(monkeypatch, capsys):
+    called = {"add": False, "retrieve": False, "modify": False}
+    pm, locked = _make_pm(called)
+    pm.last_activity = 0
+    monkeypatch.setattr(time, "time", lambda: 100.0)
+    monkeypatch.setattr("builtins.input", lambda *_: "5")
+    with pytest.raises(SystemExit):
+        main.display_menu(pm, sync_interval=1000, inactivity_timeout=0.1)
+    out = capsys.readouterr().out
+    assert "Session timed out. Vault locked." in out
+    assert locked["lock"] == 1
+    assert locked["unlock"] == 1
+    assert not any(called.values())