mirror of
https://github.com/PR0M3TH3AN/SeedPass.git
synced 2025-09-09 15:58:48 +00:00
92 lines
3.2 KiB
Python
92 lines
3.2 KiB
Python
import sys
|
|
from pathlib import Path
|
|
from tempfile import TemporaryDirectory
|
|
from types import SimpleNamespace
|
|
|
|
import bcrypt
|
|
|
|
sys.path.append(str(Path(__file__).resolve().parents[1]))
|
|
|
|
from seedpass.core.encryption import EncryptionManager
|
|
from seedpass.core.vault import Vault
|
|
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 utils.key_derivation import derive_index_key, derive_key_from_password
|
|
from utils.fingerprint import generate_fingerprint
|
|
|
|
SEED = "abandon abandon abandon abandon abandon abandon abandon abandon abandon abandon abandon about"
|
|
|
|
|
|
def test_password_change_and_unlock(monkeypatch):
|
|
with TemporaryDirectory() as tmpdir:
|
|
fp_name = generate_fingerprint(SEED)
|
|
fp = Path(tmpdir) / fp_name
|
|
fp.mkdir()
|
|
old_pw = "oldpw"
|
|
new_pw = "newpw"
|
|
|
|
# initial encryption setup
|
|
index_key = derive_index_key(SEED)
|
|
seed_key = derive_key_from_password(old_pw, fp_name)
|
|
enc_mgr = EncryptionManager(index_key, fp)
|
|
seed_mgr = EncryptionManager(seed_key, fp)
|
|
vault = Vault(enc_mgr, fp)
|
|
cfg_mgr = ConfigManager(vault, fp)
|
|
backup_mgr = BackupManager(fp, cfg_mgr)
|
|
entry_mgr = EntryManager(vault, backup_mgr)
|
|
|
|
vault.save_index({"entries": {}})
|
|
cfg_mgr.save_config(
|
|
{
|
|
"relays": [],
|
|
"pin_hash": "",
|
|
"password_hash": bcrypt.hashpw(
|
|
old_pw.encode(), bcrypt.gensalt()
|
|
).decode(),
|
|
}
|
|
)
|
|
seed_mgr.encrypt_parent_seed(SEED)
|
|
|
|
pm = PasswordManager.__new__(PasswordManager)
|
|
pm.encryption_mode = EncryptionMode.SEED_ONLY
|
|
pm.encryption_manager = enc_mgr
|
|
pm.entry_manager = entry_mgr
|
|
pm.config_manager = cfg_mgr
|
|
pm.vault = vault
|
|
pm.password_generator = SimpleNamespace(encryption_manager=enc_mgr)
|
|
pm.fingerprint_dir = fp
|
|
pm.current_fingerprint = fp_name
|
|
pm.parent_seed = SEED
|
|
pm.nostr_client = SimpleNamespace(
|
|
publish_snapshot=lambda *a, **k: (None, "abcd")
|
|
)
|
|
|
|
monkeypatch.setattr(
|
|
"seedpass.core.manager.prompt_existing_password", lambda *_: old_pw
|
|
)
|
|
monkeypatch.setattr("seedpass.core.manager.prompt_for_password", lambda: new_pw)
|
|
monkeypatch.setattr(
|
|
"seedpass.core.manager.NostrClient",
|
|
lambda *a, **kw: SimpleNamespace(
|
|
publish_snapshot=lambda *a, **k: (None, "abcd")
|
|
),
|
|
)
|
|
|
|
pm.change_password(old_pw, new_pw)
|
|
pm.lock_vault()
|
|
|
|
monkeypatch.setattr(
|
|
"seedpass.core.manager.prompt_existing_password", lambda *_: new_pw
|
|
)
|
|
monkeypatch.setattr(PasswordManager, "initialize_bip85", lambda self: None)
|
|
monkeypatch.setattr(PasswordManager, "initialize_managers", lambda self: None)
|
|
monkeypatch.setattr(PasswordManager, "sync_index_from_nostr", lambda self: None)
|
|
|
|
pm.unlock_vault(new_pw)
|
|
|
|
assert pm.parent_seed == SEED
|
|
assert pm.verify_password(new_pw)
|
|
assert not pm.locked
|