mirror of https://github.com/nucypher/nucypher.git
49 lines
1.4 KiB
Python
49 lines
1.4 KiB
Python
import pytest
|
|
|
|
from umbral.keys import UmbralPrivateKey
|
|
from umbral.signing import Signer
|
|
|
|
from nucypher.config.keyring import NucypherKeyring
|
|
from nucypher.crypto.powers import DelegatingPower, DecryptingPower
|
|
|
|
|
|
def test_generate_alice_keyring(tmpdir):
|
|
password = 'x' * 16
|
|
|
|
keyring = NucypherKeyring.generate(
|
|
password=password,
|
|
encrypting=True,
|
|
rest=False,
|
|
keyring_root=tmpdir,
|
|
federated=True
|
|
)
|
|
|
|
enc_pubkey = keyring.encrypting_public_key
|
|
assert enc_pubkey is not None
|
|
|
|
with pytest.raises(NucypherKeyring.KeyringLocked):
|
|
_dec_keypair = keyring.derive_crypto_power(DecryptingPower).keypair
|
|
|
|
keyring.unlock(password)
|
|
dec_keypair = keyring.derive_crypto_power(DecryptingPower).keypair
|
|
|
|
assert enc_pubkey == dec_keypair.pubkey
|
|
|
|
label = b'test'
|
|
|
|
delegating_power = keyring.derive_crypto_power(DelegatingPower)
|
|
delegating_pubkey = delegating_power.get_pubkey_from_label(label)
|
|
|
|
bob_pubkey = UmbralPrivateKey.gen_key().get_pubkey()
|
|
signer = Signer(UmbralPrivateKey.gen_key())
|
|
delegating_pubkey_again, _kfrags = delegating_power.generate_kfrags(
|
|
bob_pubkey, signer, label, m=2, n=3
|
|
)
|
|
|
|
assert delegating_pubkey == delegating_pubkey_again
|
|
|
|
another_delegating_power = keyring.derive_crypto_power(DelegatingPower)
|
|
another_delegating_pubkey = another_delegating_power.get_pubkey_from_label(label)
|
|
|
|
assert delegating_pubkey == another_delegating_pubkey
|