mirror of
https://github.com/PR0M3TH3AN/SeedPass.git
synced 2025-09-09 15:58:48 +00:00
90 lines
2.3 KiB
Python
90 lines
2.3 KiB
Python
import time
|
|
from types import SimpleNamespace
|
|
from pathlib import Path
|
|
import pytest
|
|
|
|
import sys
|
|
|
|
sys.path.append(str(Path(__file__).resolve().parents[1]))
|
|
|
|
import main
|
|
|
|
|
|
def test_inactivity_triggers_lock(monkeypatch):
|
|
locked = {"locked": False, "unlocked": False}
|
|
|
|
def update_activity():
|
|
pm.last_activity = time.time()
|
|
|
|
def lock_vault():
|
|
locked["locked"] = True
|
|
|
|
def unlock_vault():
|
|
locked["unlocked"] = True
|
|
update_activity()
|
|
|
|
pm = SimpleNamespace(
|
|
is_dirty=False,
|
|
last_update=time.time(),
|
|
last_activity=time.time() - 1.0,
|
|
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,
|
|
)
|
|
|
|
monkeypatch.setattr(main, "timed_input", lambda *_: "7")
|
|
|
|
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(), "7"])
|
|
|
|
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
|