def test_aes_decrypt_with_unicode(self):
"""Testing aes_decrypt with Unicode data"""
expected_message = ('The data to decrypt must be of type "bytes", '
'not "<class \'str\'>"')
with self.assertRaisesMessage(TypeError, expected_message):
aes_decrypt('abc')
def test_aes_decrypt_with_unicode(self):
"""Testing aes_decrypt with Unicode data"""
expected_message = (
'The data to decrypt must be of type "bytes", not "%s"' %
six.text_type)
with self.assertRaisesMessage(TypeError, expected_message):
aes_decrypt('abc')
def test_aes_decrypt_with_custom_key_unicode(self):
"""Testing aes_decrypt with custom key as Unicode"""
encrypted = (
b'\x9cd$e\xb1\x9e\xe0z\xb8[\x9e!\xf2h\x90\x8d\x82f%G4\xc2\xf0'
b'\xda\x8dr\x81ER?S6\x12%7\x98\x89\x90'
)
expected_message = ('The encryption key must be of type "bytes", '
'not "<class \'str\'>"')
with self.assertRaisesMessage(TypeError, expected_message):
aes_decrypt(encrypted, key='abc')
def test_aes_encrypt(self):
"""Testing aes_encrypt"""
# The encrypted value will change every time, since the iv changes,
# so we can't compare a direct value. Instead, we need to ensure that
# we can decrypt what we encrypt.
self.assertEqual(aes_decrypt(aes_encrypt(self.PLAIN_TEXT)),
self.PLAIN_TEXT)
def test_aes_encrypt(self):
"""Testing aes_encrypt"""
# The encrypted value will change every time, since the iv changes,
# so we can't compare a direct value. Instead, we need to ensure that
# we can decrypt what we encrypt.
self.assertEqual(aes_decrypt(aes_encrypt(self.PLAIN_TEXT)),
self.PLAIN_TEXT)
def test_aes_encrypt_with_unicode(self):
"""Testing aes_encrypt with Unicode string"""
# The encrypted value will change every time, since the iv changes,
# so we can't compare a direct value. Instead, we need to ensure that
# we can decrypt what we encrypt.
encrypted = aes_encrypt(self.PLAIN_TEXT_UNICODE)
self.assertIsInstance(encrypted, bytes)
self.assertEqual(aes_decrypt(encrypted), self.PLAIN_TEXT_BYTES)
def test_aes_encrypt_with_custom_key(self):
"""Testing aes_encrypt with custom key"""
# The encrypted value will change every time, since the iv changes,
# so we can't compare a direct value. Instead, we need to ensure that
# we can decrypt what we encrypt.
encrypted = aes_encrypt(self.PLAIN_TEXT, key=self.CUSTOM_KEY)
self.assertEqual(aes_decrypt(encrypted, key=self.CUSTOM_KEY),
self.PLAIN_TEXT)
def test_aes_decrypt(self):
"""Testing aes_decrypt"""
# The encrypted value was made with PyCrypto, to help with
# compatibility testing from older installs.
encrypted = (
b'\xfb\xdc\xb5h\x15\xa1\xb2\xdc\xec\xf1\x14\xa9\xc6\xab\xb2J\x10'
b'\'\xd4\xf6&\xd4k9\x82\xf6\xb5\x8bmu\xc8E\x9c\xac\xc5\x04@B')
self.assertEqual(aes_decrypt(encrypted), self.PLAIN_TEXT)
def test_aes_encrypt_with_custom_key(self):
"""Testing aes_encrypt with custom key"""
# The encrypted value will change every time, since the iv changes,
# so we can't compare a direct value. Instead, we need to ensure that
# we can decrypt what we encrypt.
encrypted = aes_encrypt(self.PLAIN_TEXT, key=self.CUSTOM_KEY)
self.assertEqual(aes_decrypt(encrypted, key=self.CUSTOM_KEY),
self.PLAIN_TEXT)
def test_aes_decrypt_with_custom_key(self):
"""Testing aes_decrypt with custom key"""
# The encrypted value was made with PyCrypto, to help with
# compatibility testing from older installs.
encrypted = (
b'\x9cd$e\xb1\x9e\xe0z\xb8[\x9e!\xf2h\x90\x8d\x82f%G4\xc2\xf0'
b'\xda\x8dr\x81ER?S6\x12%7\x98\x89\x90')
self.assertEqual(aes_decrypt(encrypted, key=self.CUSTOM_KEY),
self.PLAIN_TEXT)
def test_aes_decrypt(self):
"""Testing aes_decrypt"""
# The encrypted value was made with PyCrypto, to help with
# compatibility testing from older installs.
encrypted = (
b'\xfb\xdc\xb5h\x15\xa1\xb2\xdc\xec\xf1\x14\xa9\xc6\xab\xb2J\x10'
b'\'\xd4\xf6&\xd4k9\x82\xf6\xb5\x8bmu\xc8E\x9c\xac\xc5\x04@B'
)
self.assertEqual(aes_decrypt(encrypted), self.PLAIN_TEXT)
def test_encrypt(self):
"""Testing encrypt (deprecated)"""
with warnings.catch_warnings(record=True) as w:
# The encrypted value will change every time, since the iv changes,
# so we can't compare a direct value. Instead, we need to ensure
# that we can decrypt what we encrypt.
self.assertEqual(aes_decrypt(encrypt(self.PLAIN_TEXT)),
self.PLAIN_TEXT)
self.assertEqual(
six.text_type(w[0].message),
'encrypt() is deprecated. Use aes_encrypt() instead.')
def test_encrypt(self):
"""Testing encrypt (deprecated)"""
with warnings.catch_warnings(record=True) as w:
# The encrypted value will change every time, since the iv changes,
# so we can't compare a direct value. Instead, we need to ensure
# that we can decrypt what we encrypt.
self.assertEqual(aes_decrypt(encrypt(self.PLAIN_TEXT)),
self.PLAIN_TEXT)
self.assertEqual(
six.text_type(w[0].message),
'encrypt() is deprecated. Use aes_encrypt() instead.')
def test_aes_decrypt_with_custom_key(self):
"""Testing aes_decrypt with custom key"""
# The encrypted value was made with PyCrypto, to help with
# compatibility testing from older installs.
encrypted = (
b'\x9cd$e\xb1\x9e\xe0z\xb8[\x9e!\xf2h\x90\x8d\x82f%G4\xc2\xf0'
b'\xda\x8dr\x81ER?S6\x12%7\x98\x89\x90'
)
self.assertEqual(aes_decrypt(encrypted, key=self.CUSTOM_KEY),
self.PLAIN_TEXT)
def test_encrypt(self):
"""Testing encrypt (deprecated)"""
message = 'encrypt() is deprecated. Use aes_encrypt() instead.'
with self.assertWarns(message=message):
# The encrypted value will change every time, since the iv changes,
# so we can't compare a direct value. Instead, we need to ensure
# that we can decrypt what we encrypt.
encrypted = encrypt(self.PLAIN_TEXT_BYTES)
self.assertIsInstance(encrypted, bytes)
self.assertEqual(aes_decrypt(encrypted), self.PLAIN_TEXT_BYTES)