65 lines
2.2 KiB
Python
65 lines
2.2 KiB
Python
# -*- coding: utf-8 -*-
|
||
import pytest
|
||
|
||
from olm import (PkDecryption, PkDecryptionError, PkEncryption, PkSigning,
|
||
ed25519_verify)
|
||
|
||
|
||
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")
|
||
|
||
def test_signing(self):
|
||
seed = PkSigning.generate_seed()
|
||
signing = PkSigning(seed)
|
||
message = "This statement is true"
|
||
signature = signing.sign(message)
|
||
ed25519_verify(signing.public_key, message, signature)
|
||
|
||
def test_invalid_unicode_decrypt(self):
|
||
decryption = PkDecryption()
|
||
encryption = PkEncryption(decryption.public_key)
|
||
message = encryption.encrypt(b"\xed")
|
||
plaintext = decryption.decrypt(message)
|
||
assert plaintext == u"<EFBFBD>"
|