Add timed input for inactivity and tests

src/tests/test_inactivity_lock.py

@@ -36,10 +36,54 @@ def test_inactivity_triggers_lock(monkeypatch):
         unlock_vault=unlock_vault,
     )
 
-    monkeypatch.setattr("builtins.input", lambda _: "5")
+    monkeypatch.setattr(main, "timed_input", lambda *_: "5")
 
     with pytest.raises(SystemExit):
         main.display_menu(pm, sync_interval=1000, inactivity_timeout=0.1)
 
     assert locked["locked"]
     assert locked["unlocked"]
+
+
+def test_input_timeout_triggers_lock(monkeypatch):
+    """Ensure locking occurs if no input is provided before timeout."""
+    locked = {"locked": 0, "unlocked": 0}
+
+    def update_activity():
+        pm.last_activity = time.time()
+
+    def lock_vault():
+        locked["locked"] += 1
+
+    def unlock_vault():
+        locked["unlocked"] += 1
+        update_activity()
+
+    pm = SimpleNamespace(
+        is_dirty=False,
+        last_update=time.time(),
+        last_activity=time.time(),
+        nostr_client=SimpleNamespace(close_client_pool=lambda: None),
+        handle_add_password=lambda: None,
+        handle_retrieve_entry=lambda: None,
+        handle_modify_entry=lambda: None,
+        update_activity=update_activity,
+        lock_vault=lock_vault,
+        unlock_vault=unlock_vault,
+    )
+
+    responses = iter([TimeoutError(), "5"])
+
+    def fake_input(*_args, **_kwargs):
+        val = next(responses)
+        if isinstance(val, Exception):
+            raise val
+        return val
+
+    monkeypatch.setattr(main, "timed_input", fake_input)
+
+    with pytest.raises(SystemExit):
+        main.display_menu(pm, sync_interval=1000, inactivity_timeout=0.1)
+
+    assert locked["locked"] == 1
+    assert locked["unlocked"] == 1