54 lines
1.7 KiB
Python
54 lines
1.7 KiB
Python
from pyrage import decrypt, encrypt, x25519
|
|
|
|
from halig.encryption import Encryptor
|
|
from halig.settings import Settings
|
|
|
|
|
|
def test_instance_encryptor_from_age_keys(halig_path, notebooks_path):
|
|
identity = x25519.Identity.generate()
|
|
identity_path = halig_path / "identity.key"
|
|
identity_path.touch()
|
|
recipient_path = halig_path / "recipient.key"
|
|
recipient_path.touch()
|
|
with identity_path.open("w") as f:
|
|
f.write(str(identity))
|
|
|
|
with recipient_path.open("w") as f:
|
|
f.write(str(identity.to_public()))
|
|
|
|
settings = Settings(
|
|
notebooks_root_path=notebooks_path,
|
|
identity_path=identity_path,
|
|
recipient_path=recipient_path,
|
|
)
|
|
assert Encryptor(settings)
|
|
|
|
|
|
def test_encrypt(encryptor: Encryptor, ssh_identity):
|
|
unencrypted_data = "foo"
|
|
encrypted_data = encryptor.encrypt(unencrypted_data)
|
|
|
|
assert isinstance(encrypted_data, bytes)
|
|
assert unencrypted_data == decrypt(encrypted_data, [ssh_identity]).decode()
|
|
|
|
|
|
def test_encrypt_bytes(encryptor: Encryptor, ssh_identity):
|
|
unencrypted_data = b"foo"
|
|
encrypted_data = encryptor.encrypt(unencrypted_data)
|
|
|
|
assert isinstance(encrypted_data, bytes)
|
|
assert unencrypted_data == decrypt(encrypted_data, [ssh_identity])
|
|
|
|
|
|
def test_decrypt(encryptor: Encryptor, ssh_recipient):
|
|
unencrypted_data = "foo"
|
|
encrypted_data = encrypt(unencrypted_data.encode(), [ssh_recipient])
|
|
decrypted_data = encryptor.decrypt(encrypted_data)
|
|
assert decrypted_data.decode() == unencrypted_data
|
|
|
|
|
|
def test_decrypt_bytes(encryptor: Encryptor, ssh_recipient):
|
|
unencrypted_data = b"foo"
|
|
encrypted_data = encrypt(unencrypted_data, [ssh_recipient])
|
|
decrypted_data = encryptor.decrypt(encrypted_data)
|
|
assert decrypted_data == unencrypted_data
|