Use random BIP-39 seed in seed import test

This commit is contained in:
thePR0M3TH3AN
2025-06-29 16:28:16 -04:00
parent 4181363bbd
commit 62be23454d

View File

@@ -2,23 +2,23 @@ import sys
from pathlib import Path from pathlib import Path
from tempfile import TemporaryDirectory from tempfile import TemporaryDirectory
from cryptography.fernet import Fernet from cryptography.fernet import Fernet
from mnemonic import Mnemonic
sys.path.append(str(Path(__file__).resolve().parents[1])) sys.path.append(str(Path(__file__).resolve().parents[1]))
from password_manager.encryption import EncryptionManager from password_manager.encryption import EncryptionManager
from password_manager.manager import PasswordManager from password_manager.manager import PasswordManager
SEED = "guard rule huge draft embark case any drastic horse bargain orchard mobile"
def test_seed_encryption_round_trip(): def test_seed_encryption_round_trip():
with TemporaryDirectory() as tmpdir: with TemporaryDirectory() as tmpdir:
key = Fernet.generate_key() key = Fernet.generate_key()
enc_mgr = EncryptionManager(key, Path(tmpdir)) enc_mgr = EncryptionManager(key, Path(tmpdir))
enc_mgr.encrypt_parent_seed(SEED) seed = Mnemonic("english").generate(strength=128)
enc_mgr.encrypt_parent_seed(seed)
decrypted = enc_mgr.decrypt_parent_seed() decrypted = enc_mgr.decrypt_parent_seed()
assert decrypted == SEED assert decrypted == seed
pm = PasswordManager.__new__(PasswordManager) pm = PasswordManager.__new__(PasswordManager)
assert pm.validate_bip85_seed(SEED) assert pm.validate_bip85_seed(seed)