SeedPass/src/tests/test_restore_from_nostr_setup.py

from pathlib import Path

from helpers import create_vault, TEST_SEED, TEST_PASSWORD

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


def _init_pm(dir_path: Path, client) -> PasswordManager:
    vault, enc_mgr = create_vault(dir_path, TEST_SEED, TEST_PASSWORD)
    cfg_mgr = ConfigManager(vault, dir_path)
    backup_mgr = BackupManager(dir_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.config_manager = cfg_mgr
    pm.nostr_client = client
    pm.fingerprint_dir = dir_path
    pm.current_fingerprint = "fp"
    pm.is_dirty = False
    return pm


def test_handle_new_seed_setup_restore_from_nostr(
    monkeypatch, tmp_path, capsys, make_dummy_nostr_client
):
    client, _relay = make_dummy_nostr_client(tmp_path / "srv")

    dir_a = tmp_path / "A"
    dir_b = tmp_path / "B"
    dir_a.mkdir()
    dir_b.mkdir()

    pm_src = _init_pm(dir_a, client)
    pm_src.notify = lambda *a, **k: None
    pm_src.entry_manager.add_entry("site1", 12)
    pm_src.sync_vault()

    pm_new = PasswordManager.__new__(PasswordManager)
    pm_new.encryption_mode = EncryptionMode.SEED_ONLY
    pm_new.nostr_client = client
    pm_new.notify = lambda *a, **k: None

    def finalize(seed, *, password=None):
        vault, enc_mgr = create_vault(dir_b, seed, TEST_PASSWORD)
        cfg_mgr = ConfigManager(vault, dir_b)
        backup_mgr = BackupManager(dir_b, cfg_mgr)
        entry_mgr = EntryManager(vault, backup_mgr)
        pm_new.encryption_manager = enc_mgr
        pm_new.vault = vault
        pm_new.entry_manager = entry_mgr
        pm_new.backup_manager = backup_mgr
        pm_new.config_manager = cfg_mgr
        pm_new.fingerprint_dir = dir_b
        pm_new.current_fingerprint = "fp"
        pm_new.nostr_client = client
        return "fp"

    monkeypatch.setattr(pm_new, "_finalize_existing_seed", finalize)
    monkeypatch.setattr("seedpass.core.manager.masked_input", lambda *_: TEST_SEED)

    inputs = iter(["4"])
    monkeypatch.setattr("builtins.input", lambda *a, **k: next(inputs))

    pm_new.handle_new_seed_setup()
    out = capsys.readouterr().out
    assert "Vault restored from Nostr" in out
    labels = [e[1] for e in pm_new.entry_manager.list_entries()]
    assert labels == ["site1"]


def test_handle_new_seed_setup_restore_from_local_backup(monkeypatch, tmp_path, capsys):
    dir_a = tmp_path / "A"
    dir_b = tmp_path / "B"
    dir_a.mkdir()
    dir_b.mkdir()

    pm_src = _init_pm(dir_a, None)
    pm_src.notify = lambda *a, **k: None
    pm_src.entry_manager.add_entry("site1", 12)
    pm_src.backup_manager.create_backup()
    backup_path = next(
        pm_src.backup_manager.backup_dir.glob("entries_db_backup_*.json.enc")
    )

    pm_new = PasswordManager.__new__(PasswordManager)
    pm_new.encryption_mode = EncryptionMode.SEED_ONLY
    pm_new.notify = lambda *a, **k: None

    called = {"init": False}

    def init_fp_mgr():
        called["init"] = True
        pm_new.fingerprint_manager = object()

    monkeypatch.setattr(pm_new, "initialize_fingerprint_manager", init_fp_mgr)

    def finalize(seed, *, password=None):
        assert pm_new.fingerprint_manager is not None
        vault, enc_mgr = create_vault(dir_b, seed, TEST_PASSWORD)
        cfg_mgr = ConfigManager(vault, dir_b)
        backup_mgr = BackupManager(dir_b, cfg_mgr)
        entry_mgr = EntryManager(vault, backup_mgr)
        pm_new.encryption_manager = enc_mgr
        pm_new.vault = vault
        pm_new.entry_manager = entry_mgr
        pm_new.backup_manager = backup_mgr
        pm_new.config_manager = cfg_mgr
        pm_new.fingerprint_dir = dir_b
        pm_new.current_fingerprint = "fp"
        return "fp"

    monkeypatch.setattr(pm_new, "_finalize_existing_seed", finalize)
    monkeypatch.setattr("seedpass.core.manager.masked_input", lambda *_: TEST_SEED)

    inputs = iter(["5", str(backup_path)])
    monkeypatch.setattr("builtins.input", lambda *a, **k: next(inputs))

    pm_new.handle_new_seed_setup()
    out = capsys.readouterr().out
    assert "Index file restored from backup" in out
    labels = [e[1] for e in pm_new.entry_manager.list_entries()]
    assert labels == ["site1"]
    assert called["init"]


async def _no_snapshot():
    return None


def test_restore_from_nostr_warns(
    monkeypatch, tmp_path, capsys, make_dummy_nostr_client
):
    client, _relay = make_dummy_nostr_client(tmp_path / "srv")
    monkeypatch.setattr(client, "fetch_latest_snapshot", _no_snapshot)

    pm = PasswordManager.__new__(PasswordManager)
    pm.encryption_mode = EncryptionMode.SEED_ONLY
    pm.nostr_client = client

    monkeypatch.setattr("seedpass.core.manager.confirm_action", lambda *_: True)
    monkeypatch.setattr(pm, "_finalize_existing_seed", lambda *_a, **_k: "fp")
    monkeypatch.setattr(pm, "attempt_initial_sync", lambda: False)

    pm.restore_from_nostr_with_guidance(TEST_SEED)
    out = capsys.readouterr().out
    assert "No Nostr backup" in out


def test_restore_from_nostr_abort(
    monkeypatch, tmp_path, capsys, make_dummy_nostr_client
):
    client, _relay = make_dummy_nostr_client(tmp_path / "srv")
    monkeypatch.setattr(client, "fetch_latest_snapshot", _no_snapshot)

    pm = PasswordManager.__new__(PasswordManager)
    pm.encryption_mode = EncryptionMode.SEED_ONLY
    pm.nostr_client = client
    pm.vault = None

    called = {"finalize": 0}

    def finalize(*_a, **_k):
        called["finalize"] += 1

    monkeypatch.setattr("seedpass.core.manager.confirm_action", lambda *_: False)
    monkeypatch.setattr(pm, "_finalize_existing_seed", finalize)

    pm.restore_from_nostr_with_guidance(TEST_SEED)
    out = capsys.readouterr().out
    assert "No Nostr backup" in out
    assert called["finalize"] == 0
    assert pm.vault is None