mirror of
https://github.com/PR0M3TH3AN/SeedPass.git
synced 2025-09-10 00:09:04 +00:00
Add secret mode toggle command and API
This commit is contained in:
@@ -275,3 +275,28 @@ def test_vault_lock_endpoint(client):
|
||||
api._pm.unlock_vault = lambda: setattr(api._pm, "locked", False)
|
||||
api._pm.unlock_vault()
|
||||
assert api._pm.locked is False
|
||||
|
||||
|
||||
def test_secret_mode_endpoint(client):
|
||||
cl, token = client
|
||||
called = {}
|
||||
|
||||
def set_secret(val):
|
||||
called.setdefault("enabled", val)
|
||||
|
||||
def set_delay(val):
|
||||
called.setdefault("delay", val)
|
||||
|
||||
api._pm.config_manager.set_secret_mode_enabled = set_secret
|
||||
api._pm.config_manager.set_clipboard_clear_delay = set_delay
|
||||
|
||||
headers = {"Authorization": f"Bearer {token}"}
|
||||
res = cl.post(
|
||||
"/api/v1/secret-mode",
|
||||
json={"enabled": True, "delay": 12},
|
||||
headers=headers,
|
||||
)
|
||||
assert res.status_code == 200
|
||||
assert res.json() == {"status": "ok"}
|
||||
assert called["enabled"] is True
|
||||
assert called["delay"] == 12
|
||||
|
44
src/tests/test_cli_toggle_secret_mode.py
Normal file
44
src/tests/test_cli_toggle_secret_mode.py
Normal file
@@ -0,0 +1,44 @@
|
||||
import types
|
||||
from types import SimpleNamespace
|
||||
from typer.testing import CliRunner
|
||||
|
||||
from seedpass.cli import app
|
||||
from seedpass import cli
|
||||
|
||||
runner = CliRunner()
|
||||
|
||||
|
||||
def _make_pm(called, enabled=False, delay=45):
|
||||
cfg = SimpleNamespace(
|
||||
get_secret_mode_enabled=lambda: enabled,
|
||||
get_clipboard_clear_delay=lambda: delay,
|
||||
set_secret_mode_enabled=lambda v: called.setdefault("enabled", v),
|
||||
set_clipboard_clear_delay=lambda v: called.setdefault("delay", v),
|
||||
)
|
||||
pm = SimpleNamespace(
|
||||
config_manager=cfg,
|
||||
secret_mode_enabled=enabled,
|
||||
clipboard_clear_delay=delay,
|
||||
select_fingerprint=lambda fp: None,
|
||||
)
|
||||
return pm
|
||||
|
||||
|
||||
def test_toggle_secret_mode_updates(monkeypatch):
|
||||
called = {}
|
||||
pm = _make_pm(called)
|
||||
monkeypatch.setattr(cli, "PasswordManager", lambda: pm)
|
||||
result = runner.invoke(app, ["config", "toggle-secret-mode"], input="y\n10\n")
|
||||
assert result.exit_code == 0
|
||||
assert called == {"enabled": True, "delay": 10}
|
||||
assert "Secret mode enabled." in result.stdout
|
||||
|
||||
|
||||
def test_toggle_secret_mode_keep(monkeypatch):
|
||||
called = {}
|
||||
pm = _make_pm(called, enabled=True, delay=30)
|
||||
monkeypatch.setattr(cli, "PasswordManager", lambda: pm)
|
||||
result = runner.invoke(app, ["config", "toggle-secret-mode"], input="\n\n")
|
||||
assert result.exit_code == 0
|
||||
assert called == {"enabled": True, "delay": 30}
|
||||
assert "Secret mode enabled." in result.stdout
|
Reference in New Issue
Block a user