def test_encryption_input_error(self):
text = "Spr scrt mssg, dnt tll nyn."
password = '******'
key_file = AESCrypto.create_keys(password)
aes_crypto = AESCrypto(key_file, password)
try:
aes_crypto.encrypt(text)
assert False
except TypeError:
assert True
def test_decryption_error(self):
text = b"Spr scrt mssg, dnt tll nyn."
password = '******'
key_file = AESCrypto.create_keys(password)
aes_crypto = AESCrypto(key_file, password)
ciphertext = aes_crypto.encrypt(text, 5)
try:
aes_crypto.decrypt(ciphertext[10:50])
assert False
except InvalidToken:
assert True
def test_encrypt_decrypt(self):
text = b"A very very secret message."
password = '******'
id = 1
key_file = AESCrypto.create_keys(password)
aes_crypto = AESCrypto(key_file, password)
token = aes_crypto.encrypt(text, id)
assert text != token
data_id, data = aes_crypto.decrypt(token)
assert id == data_id
assert text == data
