mirror of
https://github.com/PR0M3TH3AN/SeedPass.git
synced 2025-09-09 15:58:48 +00:00
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:
@@ -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)
|
||||||
|
Reference in New Issue
Block a user