diff --git a/tests/test_seed_import.py b/tests/test_seed_import.py index 4f80810..b88263d 100644 --- a/tests/test_seed_import.py +++ b/tests/test_seed_import.py @@ -2,23 +2,23 @@ import sys from pathlib import Path from tempfile import TemporaryDirectory from cryptography.fernet import Fernet +from mnemonic import Mnemonic sys.path.append(str(Path(__file__).resolve().parents[1])) from password_manager.encryption import EncryptionManager 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(): with TemporaryDirectory() as tmpdir: key = Fernet.generate_key() 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() - assert decrypted == SEED + assert decrypted == seed pm = PasswordManager.__new__(PasswordManager) - assert pm.validate_bip85_seed(SEED) + assert pm.validate_bip85_seed(seed)