mirror of
https://github.com/PR0M3TH3AN/SeedPass.git
synced 2025-09-08 07:18:47 +00:00
196 lines
7.3 KiB
Python
196 lines
7.3 KiB
Python
import sys
|
|
from pathlib import Path
|
|
from tempfile import TemporaryDirectory
|
|
from types import SimpleNamespace
|
|
import queue
|
|
|
|
import pytest
|
|
|
|
from helpers import create_vault, TEST_SEED, TEST_PASSWORD
|
|
|
|
sys.path.append(str(Path(__file__).resolve().parents[1]))
|
|
|
|
from seedpass.core.entry_management import EntryManager
|
|
from seedpass.core.backup import BackupManager
|
|
from seedpass.core.config_manager import ConfigManager
|
|
from seedpass.core.manager import PasswordManager, EncryptionMode
|
|
from seedpass.core.entry_types import EntryType
|
|
|
|
|
|
def setup_entry_mgr(tmp_path: Path) -> EntryManager:
|
|
vault, _ = create_vault(tmp_path, TEST_SEED, TEST_PASSWORD)
|
|
cfg_mgr = ConfigManager(vault, tmp_path)
|
|
backup_mgr = BackupManager(tmp_path, cfg_mgr)
|
|
return EntryManager(vault, backup_mgr)
|
|
|
|
|
|
def test_archive_restore_affects_listing_and_search():
|
|
with TemporaryDirectory() as tmpdir:
|
|
tmp_path = Path(tmpdir)
|
|
em = setup_entry_mgr(tmp_path)
|
|
idx = em.add_entry("example.com", 8, "alice")
|
|
|
|
assert em.list_entries() == [(idx, "example.com", "alice", "", False)]
|
|
assert em.search_entries("example") == [
|
|
(idx, "example.com", "alice", "", False, EntryType.PASSWORD)
|
|
]
|
|
|
|
em.archive_entry(idx)
|
|
assert em.retrieve_entry(idx)["archived"] is True
|
|
assert em.list_entries() == []
|
|
assert em.list_entries(include_archived=True) == [
|
|
(idx, "example.com", "alice", "", True)
|
|
]
|
|
assert em.search_entries("example") == [
|
|
(idx, "example.com", "alice", "", True, EntryType.PASSWORD)
|
|
]
|
|
|
|
em.restore_entry(idx)
|
|
assert em.retrieve_entry(idx)["archived"] is False
|
|
assert em.list_entries() == [(idx, "example.com", "alice", "", False)]
|
|
assert em.search_entries("example") == [
|
|
(idx, "example.com", "alice", "", False, EntryType.PASSWORD)
|
|
]
|
|
|
|
|
|
def test_view_archived_entries_cli(monkeypatch):
|
|
with TemporaryDirectory() as tmpdir:
|
|
tmp_path = Path(tmpdir)
|
|
vault, enc_mgr = create_vault(tmp_path, TEST_SEED, TEST_PASSWORD)
|
|
cfg_mgr = ConfigManager(vault, tmp_path)
|
|
backup_mgr = BackupManager(tmp_path, cfg_mgr)
|
|
entry_mgr = EntryManager(vault, backup_mgr)
|
|
|
|
pm = PasswordManager.__new__(PasswordManager)
|
|
pm.encryption_mode = EncryptionMode.SEED_ONLY
|
|
pm.encryption_manager = enc_mgr
|
|
pm.vault = vault
|
|
pm.entry_manager = entry_mgr
|
|
pm.backup_manager = backup_mgr
|
|
pm.parent_seed = TEST_SEED
|
|
pm.nostr_client = SimpleNamespace()
|
|
pm.fingerprint_dir = tmp_path
|
|
pm.is_dirty = False
|
|
pm.notifications = queue.Queue()
|
|
|
|
idx = entry_mgr.add_entry("example.com", 8)
|
|
|
|
monkeypatch.setattr("builtins.input", lambda *_: str(idx))
|
|
pm.handle_archive_entry()
|
|
assert entry_mgr.retrieve_entry(idx)["archived"] is True
|
|
|
|
inputs = iter([str(idx), "r", "", ""])
|
|
monkeypatch.setattr("builtins.input", lambda *_: next(inputs))
|
|
pm.handle_view_archived_entries()
|
|
assert entry_mgr.retrieve_entry(idx)["archived"] is False
|
|
|
|
|
|
def test_view_archived_entries_view_only(monkeypatch, capsys):
|
|
with TemporaryDirectory() as tmpdir:
|
|
tmp_path = Path(tmpdir)
|
|
vault, enc_mgr = create_vault(tmp_path, TEST_SEED, TEST_PASSWORD)
|
|
cfg_mgr = ConfigManager(vault, tmp_path)
|
|
backup_mgr = BackupManager(tmp_path, cfg_mgr)
|
|
entry_mgr = EntryManager(vault, backup_mgr)
|
|
|
|
pm = PasswordManager.__new__(PasswordManager)
|
|
pm.encryption_mode = EncryptionMode.SEED_ONLY
|
|
pm.encryption_manager = enc_mgr
|
|
pm.vault = vault
|
|
pm.entry_manager = entry_mgr
|
|
pm.backup_manager = backup_mgr
|
|
pm.parent_seed = TEST_SEED
|
|
pm.nostr_client = SimpleNamespace()
|
|
pm.fingerprint_dir = tmp_path
|
|
pm.is_dirty = False
|
|
pm.notifications = queue.Queue()
|
|
|
|
idx = entry_mgr.add_entry("example.com", 8)
|
|
|
|
monkeypatch.setattr("builtins.input", lambda *_: str(idx))
|
|
pm.handle_archive_entry()
|
|
assert entry_mgr.retrieve_entry(idx)["archived"] is True
|
|
|
|
inputs = iter([str(idx), "v", "", "", ""])
|
|
monkeypatch.setattr("builtins.input", lambda *_: next(inputs))
|
|
pm.handle_view_archived_entries()
|
|
assert entry_mgr.retrieve_entry(idx)["archived"] is True
|
|
out = capsys.readouterr().out
|
|
assert "example.com" in out
|
|
|
|
|
|
def test_view_archived_entries_removed_after_restore(monkeypatch, capsys):
|
|
with TemporaryDirectory() as tmpdir:
|
|
tmp_path = Path(tmpdir)
|
|
vault, enc_mgr = create_vault(tmp_path, TEST_SEED, TEST_PASSWORD)
|
|
cfg_mgr = ConfigManager(vault, tmp_path)
|
|
backup_mgr = BackupManager(tmp_path, cfg_mgr)
|
|
entry_mgr = EntryManager(vault, backup_mgr)
|
|
|
|
pm = PasswordManager.__new__(PasswordManager)
|
|
pm.encryption_mode = EncryptionMode.SEED_ONLY
|
|
pm.encryption_manager = enc_mgr
|
|
pm.vault = vault
|
|
pm.entry_manager = entry_mgr
|
|
pm.backup_manager = backup_mgr
|
|
pm.parent_seed = TEST_SEED
|
|
pm.nostr_client = SimpleNamespace()
|
|
pm.fingerprint_dir = tmp_path
|
|
pm.is_dirty = False
|
|
pm.notifications = queue.Queue()
|
|
|
|
idx = entry_mgr.add_entry("example.com", 8)
|
|
|
|
monkeypatch.setattr("builtins.input", lambda *_: str(idx))
|
|
pm.handle_archive_entry()
|
|
assert entry_mgr.retrieve_entry(idx)["archived"] is True
|
|
|
|
inputs = iter([str(idx), "r", "", ""])
|
|
monkeypatch.setattr("builtins.input", lambda *_: next(inputs))
|
|
pm.handle_view_archived_entries()
|
|
assert entry_mgr.retrieve_entry(idx)["archived"] is False
|
|
|
|
monkeypatch.setattr("builtins.input", lambda *_: "")
|
|
pm.handle_view_archived_entries()
|
|
note = pm.notifications.get_nowait()
|
|
assert note.level == "WARNING"
|
|
assert note.message == "No archived entries found."
|
|
|
|
|
|
def test_archived_entries_menu_hides_active(monkeypatch, capsys):
|
|
with TemporaryDirectory() as tmpdir:
|
|
tmp_path = Path(tmpdir)
|
|
vault, enc_mgr = create_vault(tmp_path, TEST_SEED, TEST_PASSWORD)
|
|
cfg_mgr = ConfigManager(vault, tmp_path)
|
|
backup_mgr = BackupManager(tmp_path, cfg_mgr)
|
|
entry_mgr = EntryManager(vault, backup_mgr)
|
|
|
|
pm = PasswordManager.__new__(PasswordManager)
|
|
pm.encryption_mode = EncryptionMode.SEED_ONLY
|
|
pm.encryption_manager = enc_mgr
|
|
pm.vault = vault
|
|
pm.entry_manager = entry_mgr
|
|
pm.backup_manager = backup_mgr
|
|
pm.parent_seed = TEST_SEED
|
|
pm.nostr_client = SimpleNamespace()
|
|
pm.fingerprint_dir = tmp_path
|
|
pm.is_dirty = False
|
|
pm.notifications = queue.Queue()
|
|
|
|
archived_idx = entry_mgr.add_entry("archived.com", 8)
|
|
active_idx = entry_mgr.add_entry("active.com", 8)
|
|
|
|
# Archive only the first entry
|
|
monkeypatch.setattr("builtins.input", lambda *_: str(archived_idx))
|
|
pm.handle_archive_entry()
|
|
assert entry_mgr.retrieve_entry(archived_idx)["archived"] is True
|
|
assert entry_mgr.retrieve_entry(active_idx)["archived"] is False
|
|
|
|
# View archived entries and immediately exit
|
|
inputs = iter([""])
|
|
monkeypatch.setattr("builtins.input", lambda *_: next(inputs))
|
|
pm.handle_view_archived_entries()
|
|
out = capsys.readouterr().out
|
|
assert "archived.com" in out
|
|
assert "active.com" not in out
|