2f5590bf38
This patch changes the decryption functions not to fail if there was an unicode decode error while converting the decrypted bytes plaintext into a native python string. Characters that cannot be decoded as unicode are now replaced with the unicode replacement character (U+FFFD). The old behaviour of raising an UnicodeDecodeError can be achieved by passing the "strict" error handling scheme to the decrypt function.
67 lines
2.3 KiB
Python
67 lines
2.3 KiB
Python
# -*- coding: utf-8 -*-
|
||
from builtins import bytes
|
||
|
||
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(bytes([0xed]))
|
||
plaintext = decryption.decrypt(message)
|
||
assert plaintext == "<EFBFBD>"
|