Merge pull request #31 from PR0M3TH3AN/codex/update-test_seed_import.py-to-use-random-seed

Generate random BIP‑39 seed for encryption test
This commit is contained in:
thePR0M3TH3AN
2025-06-29 16:29:34 -04:00
committed by GitHub

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)