mirror of
https://gitlab.com/pulsechaincom/staking-deposit-cli.git
synced 2024-12-25 04:47:18 +00:00
59 lines
2.3 KiB
Python
59 lines
2.3 KiB
Python
|
import os
|
||
|
|
import json
|
||
|
|
|
||
|
|
from key_handling.keystore import (
|
||
|
|
Keystore,
|
||
|
|
ScryptKeystore,
|
||
|
|
Pbkdf2Keystore,
|
||
|
|
)
|
||
|
|
|
||
|
|
test_vector_password = 'testpassword'
|
||
|
|
test_vector_secret = bytes.fromhex('000000000019d6689c085ae165831e934ff763ae46a2a6c172b3f1b60a8ce26f')
|
||
|
|
test_vector_folder = os.path.join(os.getcwd(), 'tests','test_key_handling', 'keystore_test_vectors')
|
||
|
|
_, _, test_vector_files = next(os.walk(test_vector_folder))
|
||
|
|
|
||
|
|
test_vector_keystores = [Keystore.from_json(os.path.join(test_vector_folder, f)) for f in test_vector_files]
|
||
|
|
|
||
|
|
|
||
|
|
def test_json_serialization():
|
||
|
|
for keystore, keystore_json_file in zip(test_vector_keystores, test_vector_files):
|
||
|
|
keystore_json_path = os.path.join(test_vector_folder, keystore_json_file)
|
||
|
|
with open(keystore_json_path) as f:
|
||
|
|
assert json.loads(keystore.as_json()) == json.load(f)
|
||
|
|
|
||
|
|
|
||
|
|
def test_encrypt_decrypt_test_vectors():
|
||
|
|
for tv in test_vector_keystores:
|
||
|
|
aes_iv = tv.crypto.cipher.params['iv']
|
||
|
|
kdf_salt = tv.crypto.kdf.params['salt']
|
||
|
|
keystore = Pbkdf2Keystore if 'pbkdf' in tv.crypto.kdf.function else ScryptKeystore
|
||
|
|
generated_keystore = keystore.encrypt(
|
||
|
|
secret=test_vector_secret,
|
||
|
|
password=test_vector_password,
|
||
|
|
aes_iv=aes_iv,
|
||
|
|
kdf_salt=kdf_salt)
|
||
|
|
assert generated_keystore.decrypt(test_vector_password) == test_vector_secret
|
||
|
|
|
||
|
|
|
||
|
|
def test_generated_keystores():
|
||
|
|
for tv in test_vector_keystores:
|
||
|
|
aes_iv = tv.crypto.cipher.params['iv']
|
||
|
|
kdf_salt = tv.crypto.kdf.params['salt']
|
||
|
|
keystore = Pbkdf2Keystore if 'pbkdf' in tv.crypto.kdf.function else ScryptKeystore
|
||
|
|
generated_keystore = keystore.encrypt(
|
||
|
|
secret=test_vector_secret,
|
||
|
|
password=test_vector_password,
|
||
|
|
aes_iv=aes_iv,
|
||
|
|
kdf_salt=kdf_salt)
|
||
|
|
assert generated_keystore.crypto == tv.crypto
|
||
|
|
|
||
|
|
|
||
|
|
def test_encrypt_decrypt_pbkdf2_random_iv():
|
||
|
|
generated_keystore = Pbkdf2Keystore.encrypt(secret=test_vector_secret, password=test_vector_password)
|
||
|
|
assert generated_keystore.decrypt(test_vector_password) == test_vector_secret
|
||
|
|
|
||
|
|
|
||
|
|
def test_encrypt_decrypt_scrypt_random_iv():
|
||
|
|
generated_keystore = ScryptKeystore.encrypt(secret=test_vector_secret, password=test_vector_password)
|
||
|
|
assert generated_keystore.decrypt(test_vector_password) == test_vector_secret
|