Add service classes and tests

src/tests/test_service_classes.py

@@ -0,0 +1,134 @@
+from tempfile import TemporaryDirectory
+from types import SimpleNamespace
+from pathlib import Path
+
+import pytest
+
+from helpers import create_vault, TEST_SEED, TEST_PASSWORD, dummy_nostr_client
+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
+from seedpass.core.entry_service import EntryService
+from seedpass.core.profile_service import ProfileService
+from constants import DEFAULT_PASSWORD_LENGTH
+
+
+class FakePasswordGenerator:
+    def generate_password(self, length: int, index: int) -> str:
+        return f"pw-{index}-{length}"
+
+
+def _setup_pm(tmp_path: Path, client) -> PasswordManager:
+    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.parent_seed = TEST_SEED
+    pm.nostr_client = client
+    pm.fingerprint_dir = tmp_path
+    pm.secret_mode_enabled = False
+    pm.is_dirty = False
+    return pm
+
+
+def test_entry_service_add_password(monkeypatch, dummy_nostr_client, capsys):
+    client, _relay = dummy_nostr_client
+    with TemporaryDirectory() as tmpdir:
+        pm = _setup_pm(Path(tmpdir), client)
+        service = EntryService(pm)
+        inputs = iter(
+            [
+                "a",
+                "Example",
+                "",
+                "",
+                "",
+                "",
+                "n",
+                "",
+                "",
+                "",
+                "",
+                "",
+                "",
+                "",
+                "",
+                "",
+            ]
+        )
+        monkeypatch.setattr("builtins.input", lambda *a, **k: next(inputs))
+        monkeypatch.setattr("seedpass.core.entry_service.pause", lambda *a, **k: None)
+        monkeypatch.setattr(pm, "start_background_vault_sync", lambda *a, **k: None)
+
+        service.handle_add_password()
+        out = capsys.readouterr().out
+        entries = pm.entry_manager.list_entries(verbose=False)
+        assert entries == [(0, "Example", "", "", False)]
+        assert f"pw-0-{DEFAULT_PASSWORD_LENGTH}" in out
+
+
+def test_menu_handler_list_entries(monkeypatch, capsys):
+    with TemporaryDirectory() as tmpdir:
+        pm = _setup_pm(Path(tmpdir), SimpleNamespace())
+        pm.entry_manager.add_totp("Example", TEST_SEED)
+        pm.entry_manager.add_entry("example.com", 12)
+        pm.entry_manager.add_key_value("API entry", "api", "abc123")
+        pm.entry_manager.add_managed_account("acct", TEST_SEED)
+        inputs = iter(["1", ""])  # list all then exit
+        monkeypatch.setattr("builtins.input", lambda *_: next(inputs))
+        pm.menu_handler.handle_list_entries()
+        out = capsys.readouterr().out
+        assert "Example" in out
+        assert "example.com" in out
+        assert "API" in out
+        assert "acct" in out
+
+
+def test_profile_service_switch(monkeypatch):
+    class DummyFingerprintManager:
+        def __init__(self):
+            self.fingerprints = ["fp1", "fp2"]
+            self.current_fingerprint = "fp1"
+
+        def list_fingerprints(self):
+            return self.fingerprints
+
+        def display_name(self, fp):
+            return fp
+
+        def get_current_fingerprint_dir(self):
+            return Path(".")
+
+    pm = PasswordManager.__new__(PasswordManager)
+    pm.fingerprint_manager = DummyFingerprintManager()
+    pm.current_fingerprint = "fp1"
+    pm.setup_encryption_manager = lambda *a, **k: True
+    pm.initialize_bip85 = lambda *a, **k: None
+    pm.initialize_managers = lambda *a, **k: None
+    pm.start_background_sync = lambda *a, **k: None
+    pm.nostr_client = SimpleNamespace()
+    pm.manifest_id = None
+    pm.delta_since = None
+    pm.encryption_manager = SimpleNamespace()
+    pm.parent_seed = TEST_SEED
+
+    service = ProfileService(pm)
+    monkeypatch.setattr("builtins.input", lambda *_: "2")
+    monkeypatch.setattr(
+        "seedpass.core.profile_service.prompt_existing_password", lambda *_: "pw"
+    )
+    monkeypatch.setattr(
+        "seedpass.core.manager.NostrClient", lambda *a, **k: SimpleNamespace()
+    )
+
+    assert service.handle_switch_fingerprint() is True
+    assert pm.current_fingerprint == "fp2"