mirror of
https://github.com/PR0M3TH3AN/SeedPass.git
synced 2025-09-09 15:58:48 +00:00
Add test fixtures for manager and vault
This commit is contained in:
@@ -1,7 +1,18 @@
|
|||||||
import importlib.util
|
import importlib.util
|
||||||
import logging
|
import logging
|
||||||
|
import sys
|
||||||
|
from pathlib import Path
|
||||||
|
|
||||||
import pytest
|
import pytest
|
||||||
|
|
||||||
|
sys.path.append(str(Path(__file__).resolve().parents[1]))
|
||||||
|
|
||||||
|
from helpers import create_vault, TEST_PASSWORD, TEST_SEED
|
||||||
|
from seedpass.core.backup import BackupManager
|
||||||
|
from seedpass.core.config_manager import ConfigManager
|
||||||
|
from seedpass.core.entry_management import EntryManager
|
||||||
|
from seedpass.core.manager import EncryptionMode, PasswordManager
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture(
|
@pytest.fixture(
|
||||||
params=["asyncio"] + (["trio"] if importlib.util.find_spec("trio") else [])
|
params=["asyncio"] + (["trio"] if importlib.util.find_spec("trio") else [])
|
||||||
@@ -57,3 +68,29 @@ def pytest_collection_modifyitems(
|
|||||||
for item in items:
|
for item in items:
|
||||||
if "desktop" in item.keywords:
|
if "desktop" in item.keywords:
|
||||||
item.add_marker(skip_desktop)
|
item.add_marker(skip_desktop)
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture
|
||||||
|
def vault(tmp_path):
|
||||||
|
vault, _ = create_vault(tmp_path, TEST_SEED, TEST_PASSWORD)
|
||||||
|
return vault
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture
|
||||||
|
def password_manager(vault, tmp_path):
|
||||||
|
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 = vault.encryption_manager
|
||||||
|
pm.vault = vault
|
||||||
|
pm.entry_manager = entry_mgr
|
||||||
|
pm.backup_manager = backup_mgr
|
||||||
|
pm.parent_seed = TEST_SEED
|
||||||
|
pm.nostr_client = None
|
||||||
|
pm.fingerprint_dir = tmp_path
|
||||||
|
pm.is_dirty = False
|
||||||
|
pm.secret_mode_enabled = False
|
||||||
|
return pm
|
||||||
|
@@ -1,39 +1,16 @@
|
|||||||
import sys
|
import sys
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
from tempfile import TemporaryDirectory
|
|
||||||
from types import SimpleNamespace
|
from types import SimpleNamespace
|
||||||
|
|
||||||
from helpers import create_vault, TEST_SEED, TEST_PASSWORD
|
|
||||||
|
|
||||||
sys.path.append(str(Path(__file__).resolve().parents[1]))
|
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.manager import PasswordManager, EncryptionMode
|
|
||||||
from seedpass.core.config_manager import ConfigManager
|
|
||||||
|
|
||||||
|
def test_retrieve_entry_shows_custom_fields(monkeypatch, capsys, password_manager):
|
||||||
def test_retrieve_entry_shows_custom_fields(monkeypatch, capsys):
|
pm = password_manager
|
||||||
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.password_generator = SimpleNamespace(generate_password=lambda l, i: "pw")
|
pm.password_generator = SimpleNamespace(generate_password=lambda l, i: "pw")
|
||||||
pm.parent_seed = TEST_SEED
|
|
||||||
pm.nostr_client = SimpleNamespace()
|
pm.nostr_client = SimpleNamespace()
|
||||||
pm.fingerprint_dir = tmp_path
|
|
||||||
pm.secret_mode_enabled = False
|
|
||||||
|
|
||||||
entry_mgr.add_entry(
|
pm.entry_manager.add_entry(
|
||||||
"example",
|
"example",
|
||||||
8,
|
8,
|
||||||
custom_fields=[
|
custom_fields=[
|
||||||
|
@@ -1,49 +1,26 @@
|
|||||||
import sys
|
import sys
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
from tempfile import TemporaryDirectory
|
|
||||||
from types import SimpleNamespace
|
from types import SimpleNamespace
|
||||||
|
|
||||||
from helpers import create_vault, TEST_SEED, TEST_PASSWORD
|
|
||||||
|
|
||||||
sys.path.append(str(Path(__file__).resolve().parents[1]))
|
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.manager import PasswordManager, EncryptionMode
|
|
||||||
from seedpass.core.config_manager import ConfigManager
|
|
||||||
|
|
||||||
|
|
||||||
class FakePasswordGenerator:
|
class FakePasswordGenerator:
|
||||||
def generate_password(self, length: int, index: int) -> str: # noqa: D401
|
def generate_password(self, length: int, index: int) -> str: # noqa: D401
|
||||||
return "pw"
|
return "pw"
|
||||||
|
|
||||||
|
|
||||||
def test_edit_tags_from_retrieve(monkeypatch):
|
def test_edit_tags_from_retrieve(monkeypatch, password_manager):
|
||||||
with TemporaryDirectory() as tmpdir:
|
pm = password_manager
|
||||||
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.password_generator = FakePasswordGenerator()
|
pm.password_generator = FakePasswordGenerator()
|
||||||
pm.parent_seed = TEST_SEED
|
|
||||||
pm.nostr_client = SimpleNamespace()
|
pm.nostr_client = SimpleNamespace()
|
||||||
pm.fingerprint_dir = tmp_path
|
|
||||||
pm.secret_mode_enabled = False
|
|
||||||
|
|
||||||
index = entry_mgr.add_entry("example.com", 8, tags=["old"])
|
index = pm.entry_manager.add_entry("example.com", 8, tags=["old"])
|
||||||
|
|
||||||
inputs = iter([str(index), "t", "newtag", ""])
|
inputs = iter([str(index), "t", "newtag", ""])
|
||||||
monkeypatch.setattr("builtins.input", lambda *a, **k: next(inputs))
|
monkeypatch.setattr("builtins.input", lambda *a, **k: next(inputs))
|
||||||
|
|
||||||
pm.handle_retrieve_entry()
|
pm.handle_retrieve_entry()
|
||||||
|
|
||||||
entry = entry_mgr.retrieve_entry(index)
|
entry = pm.entry_manager.retrieve_entry(index)
|
||||||
assert entry.get("tags", []) == ["newtag"]
|
assert entry.get("tags", []) == ["newtag"]
|
||||||
|
@@ -1,16 +1,10 @@
|
|||||||
import sys
|
import sys
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
from tempfile import TemporaryDirectory
|
|
||||||
|
|
||||||
from helpers import create_vault, TEST_SEED, TEST_PASSWORD
|
from helpers import TEST_SEED
|
||||||
|
|
||||||
sys.path.append(str(Path(__file__).resolve().parents[1]))
|
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.manager import PasswordManager, EncryptionMode
|
|
||||||
from seedpass.core.config_manager import ConfigManager
|
|
||||||
|
|
||||||
|
|
||||||
class FakeNostrClient:
|
class FakeNostrClient:
|
||||||
def __init__(self, *args, **kwargs):
|
def __init__(self, *args, **kwargs):
|
||||||
@@ -21,32 +15,14 @@ class FakeNostrClient:
|
|||||||
return None, "abcd"
|
return None, "abcd"
|
||||||
|
|
||||||
|
|
||||||
def test_handle_display_totp_codes(monkeypatch, capsys):
|
def test_handle_display_totp_codes(monkeypatch, capsys, password_manager):
|
||||||
with TemporaryDirectory() as tmpdir:
|
pm = password_manager
|
||||||
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 = FakeNostrClient()
|
pm.nostr_client = FakeNostrClient()
|
||||||
pm.fingerprint_dir = tmp_path
|
|
||||||
pm.is_dirty = False
|
|
||||||
pm.secret_mode_enabled = False
|
|
||||||
|
|
||||||
entry_mgr.add_totp("Example", TEST_SEED)
|
pm.entry_manager.add_totp("Example", TEST_SEED)
|
||||||
|
|
||||||
monkeypatch.setattr(pm.entry_manager, "get_totp_code", lambda *a, **k: "123456")
|
monkeypatch.setattr(pm.entry_manager, "get_totp_code", lambda *a, **k: "123456")
|
||||||
monkeypatch.setattr(
|
monkeypatch.setattr(pm.entry_manager, "get_totp_time_remaining", lambda *a, **k: 30)
|
||||||
pm.entry_manager, "get_totp_time_remaining", lambda *a, **k: 30
|
|
||||||
)
|
|
||||||
|
|
||||||
# interrupt the loop after first iteration
|
# interrupt the loop after first iteration
|
||||||
monkeypatch.setattr(
|
monkeypatch.setattr(
|
||||||
@@ -61,34 +37,16 @@ def test_handle_display_totp_codes(monkeypatch, capsys):
|
|||||||
assert "123456" in out
|
assert "123456" in out
|
||||||
|
|
||||||
|
|
||||||
def test_display_totp_codes_excludes_archived(monkeypatch, capsys):
|
def test_display_totp_codes_excludes_archived(monkeypatch, capsys, password_manager):
|
||||||
with TemporaryDirectory() as tmpdir:
|
pm = password_manager
|
||||||
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 = FakeNostrClient()
|
pm.nostr_client = FakeNostrClient()
|
||||||
pm.fingerprint_dir = tmp_path
|
|
||||||
pm.is_dirty = False
|
|
||||||
pm.secret_mode_enabled = False
|
|
||||||
|
|
||||||
entry_mgr.add_totp("Visible", TEST_SEED)
|
pm.entry_manager.add_totp("Visible", TEST_SEED)
|
||||||
entry_mgr.add_totp("Hidden", TEST_SEED)
|
pm.entry_manager.add_totp("Hidden", TEST_SEED)
|
||||||
entry_mgr.modify_entry(1, archived=True)
|
pm.entry_manager.modify_entry(1, archived=True)
|
||||||
|
|
||||||
monkeypatch.setattr(pm.entry_manager, "get_totp_code", lambda *a, **k: "123456")
|
monkeypatch.setattr(pm.entry_manager, "get_totp_code", lambda *a, **k: "123456")
|
||||||
monkeypatch.setattr(
|
monkeypatch.setattr(pm.entry_manager, "get_totp_time_remaining", lambda *a, **k: 30)
|
||||||
pm.entry_manager, "get_totp_time_remaining", lambda *a, **k: 30
|
|
||||||
)
|
|
||||||
|
|
||||||
monkeypatch.setattr(
|
monkeypatch.setattr(
|
||||||
"seedpass.core.manager.timed_input",
|
"seedpass.core.manager.timed_input",
|
||||||
|
@@ -1,16 +1,10 @@
|
|||||||
import sys
|
import sys
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
from tempfile import TemporaryDirectory
|
|
||||||
|
|
||||||
from helpers import create_vault, TEST_SEED, TEST_PASSWORD
|
from helpers import TEST_SEED
|
||||||
|
|
||||||
sys.path.append(str(Path(__file__).resolve().parents[1]))
|
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.manager import PasswordManager, EncryptionMode
|
|
||||||
from seedpass.core.config_manager import ConfigManager
|
|
||||||
|
|
||||||
import pytest
|
import pytest
|
||||||
|
|
||||||
|
|
||||||
@@ -23,25 +17,12 @@ import pytest
|
|||||||
(lambda mgr: mgr.add_nostr_key("nostr", TEST_SEED), False),
|
(lambda mgr: mgr.add_nostr_key("nostr", TEST_SEED), False),
|
||||||
],
|
],
|
||||||
)
|
)
|
||||||
def test_pause_before_entry_actions(monkeypatch, adder, needs_confirm):
|
def test_pause_before_entry_actions(
|
||||||
with TemporaryDirectory() as tmpdir:
|
monkeypatch, adder, needs_confirm, password_manager
|
||||||
tmp_path = Path(tmpdir)
|
):
|
||||||
vault, enc_mgr = create_vault(tmp_path, TEST_SEED, TEST_PASSWORD)
|
pm = password_manager
|
||||||
cfg_mgr = ConfigManager(vault, tmp_path)
|
|
||||||
backup_mgr = BackupManager(tmp_path, cfg_mgr)
|
|
||||||
entry_mgr = EntryManager(vault, backup_mgr)
|
|
||||||
|
|
||||||
pm = PasswordManager.__new__(PasswordManager)
|
index = adder(pm.entry_manager)
|
||||||
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.fingerprint_dir = tmp_path
|
|
||||||
pm.secret_mode_enabled = False
|
|
||||||
|
|
||||||
index = adder(entry_mgr)
|
|
||||||
|
|
||||||
pause_calls = []
|
pause_calls = []
|
||||||
monkeypatch.setattr(
|
monkeypatch.setattr(
|
||||||
|
Reference in New Issue
Block a user