50 lines
1.7 KiB
Python
50 lines
1.7 KiB
Python
|
import pytest
|
||
|
|
|
||
|
|
from olm import PkDecryption, PkDecryptionError, PkEncryption
|
||
|
|
|
||
|
|
|
||
|
|
class TestClass(object):
|
||
|
|
def test_invalid_encryption(self):
|
||
|
|
with pytest.raises(ValueError):
|
||
|
|
PkEncryption("")
|
||
|
|
|
||
|
|
def test_decrytion(self):
|
||
|
|
decryption = PkDecryption()
|
||
|
|
encryption = PkEncryption(decryption.public_key)
|
||
|
|
plaintext = "It's a secret to everybody."
|
||
|
|
message = encryption.encrypt(plaintext)
|
||
|
|
decrypted_plaintext = decryption.decrypt(message)
|
||
|
|
isinstance(decrypted_plaintext, str)
|
||
|
|
assert plaintext == decrypted_plaintext
|
||
|
|
|
||
|
|
def test_invalid_decrytion(self):
|
||
|
|
decryption = PkDecryption()
|
||
|
|
encryption = PkEncryption(decryption.public_key)
|
||
|
|
plaintext = "It's a secret to everybody."
|
||
|
|
message = encryption.encrypt(plaintext)
|
||
|
|
message.ephemeral_key = "?"
|
||
|
|
with pytest.raises(PkDecryptionError):
|
||
|
|
decryption.decrypt(message)
|
||
|
|
|
||
|
|
def test_pickling(self):
|
||
|
|
decryption = PkDecryption()
|
||
|
|
encryption = PkEncryption(decryption.public_key)
|
||
|
|
plaintext = "It's a secret to everybody."
|
||
|
|
message = encryption.encrypt(plaintext)
|
||
|
|
|
||
|
|
pickle = decryption.pickle()
|
||
|
|
unpickled = PkDecryption.from_pickle(pickle)
|
||
|
|
decrypted_plaintext = unpickled.decrypt(message)
|
||
|
|
assert plaintext == decrypted_plaintext
|
||
|
|
|
||
|
|
def test_invalid_unpickling(self):
|
||
|
|
with pytest.raises(ValueError):
|
||
|
|
PkDecryption.from_pickle("")
|
||
|
|
|
||
|
|
def test_invalid_pass_pickling(self):
|
||
|
|
decryption = PkDecryption()
|
||
|
|
pickle = decryption.pickle("Secret")
|
||
|
|
|
||
|
|
with pytest.raises(PkDecryptionError):
|
||
|
|
PkDecryption.from_pickle(pickle, "Not secret")
|