def test_with_invalid_version():
key = b']\x1a\xa2\nW\x97\xab)\x951\xa8t\x8b\xd8\xac\x08\xebjlY\xd0S\x90d\xcc\rR\x1f\xbf\x13\xe0:\xb5\x7f\xbf\xa7\x83|\x10bQ\x03\xd3Z]\xea\x1f2\xf6tB\x13\xaeP\xcc\x8fb\xabY\xda#\xe9QE'
encrypted_data = b'\x00M\xcdjP\xfd\xcc\xa1\xd7\xda\x11(Q \xbd\xe4w\n\x03C\x14!\x99N\xe8\xf0H\xbc\xf8\xf41\xa5\x10E\x0e\xbc\x04\x01\x85\x0b\xd5F\x1bq>\x12\x04\x11Y\x10\x8f\x0f\x06'
with pytest.raises(EncryptionError) as excinfo:
aes_decrypt(key, encrypted_data)
assert str(
excinfo.value) == "Found invalid version marker: {!r}".format(b'\x00')
def test_with_v1_corrupted_data():
data = b'test'
key = aes_generate_key()
encrypted_data = aes_encrypt(key, data)
assert encrypted_data[0:1] == b'\x01'
corrupted_encrypted_data = encrypted_data[:-3]
with pytest.raises(AuthenticationError) as excinfo:
aes_decrypt(key, corrupted_encrypted_data)
assert str(excinfo.value) == "message authentication failed"
def _decrypt(self, data):
for key in self.key_list:
try:
return aes_decrypt(key, data)
except AuthenticationError:
pass
raise AuthenticationError('message authentication failed')
def test_with_data_longer_than_aes_block():
data = b'a' * 130
key = aes_generate_key()
assert aes_decrypt(key, aes_encrypt(key, data)) == data
def test_with_data_exactly_as_long_as_aes_block():
data = b'a' * 128
key = aes_generate_key()
assert aes_decrypt(key, aes_encrypt(key, data)) == data
def test_with_v1_data():
data = b'test'
key = aes_generate_key()
encrypted_data = aes_encrypt(key, data)
assert encrypted_data[0:1] == b'\x01'
assert aes_decrypt(key, encrypted_data) == data