def test_correct_key_length(self):
service = AESEncryption()
service.configure()
key = service._correct_key_length('*' * 10)
self.assertEqual(len(key), AES.block_size)
key = service._correct_key_length('*' * 20)
self.assertEqual(len(key), AES.block_size)
key = service._correct_key_length('*'*16)
self.assertEqual(len(key), AES.block_size)
def test_correct_key_length(self):
service = AESEncryption()
service.configure()
key = service._correct_key_length('*' * 10)
self.assertEqual(len(key), AES.block_size)
key = service._correct_key_length('*' * 20)
self.assertEqual(len(key), AES.block_size)
key = service._correct_key_length('*'*16)
self.assertEqual(len(key), AES.block_size)
def test_check_sum_length(self):
service = AESEncryption()
service.configure()
self.assertEqual(service.check_sum_length(), 4)
service.configure(use_checksum=False)
self.assertEqual(service.check_sum_length(), 0)
def test_is_encrypted(self):
service = AESEncryption()
service.configure()
self.assertTrue(service.is_encrypted(b'KN+p\x80\xac\xe23IX', '123456', b'\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00'))
def test_decrypt(self):
service = AESEncryption()
service.configure()
self.assertEqual(b'secret', service.decrypt(b'KN+p\x80\xac\xe23IX', '123456',b'\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00'))
def test_check_sum_length(self):
service = AESEncryption()
service.configure()
self.assertEqual(service.check_sum_length(), 4)
service.configure(use_checksum=False)
self.assertEqual(service.check_sum_length(), 0)
def test_is_encrypted(self):
service = AESEncryption()
service.configure()
self.assertTrue(service.is_encrypted('KN+p\x80\xac\xe23IX', '123456'))
def test_decrypt(self):
service = AESEncryption()
service.configure()
self.assertEqual('secret', service.decrypt('KN+p\x80\xac\xe23IX', '123456'))