def invalid_key_test(self):
key1 = encryption.generate_key()
key2 = encryption.generate_key()
data = "test data 123"
ciphertext = encryption.encrypt(data, key1)
self.assertTrue(encryption.is_encrypted(ciphertext))
self.assertRaises(encryption.InvalidKeyException,
lambda: encryption.decrypt(ciphertext, key2))
def encrypt_exact_size_test(self):
self.assertInverses("hi", 2)
self.assertRaises(
ValueError,
lambda: encryption.encrypt("hi", encryption.generate_key(), 1))
# accented i in sí, longer than 2 characters
self.assertRaises(
ValueError,
lambda: encryption.encrypt("sí", encryption.generate_key(), 2))
self.assertInverses("hi", 3)
def find_single_key_test(self):
key = encryption.generate_key()
self.assertEqual(
[key],
encryption.get_keys(
"some text some text some text {} some text some text some text"
.format(key)))
self.assertEqual([key], encryption.get_keys(key))
self.assertEqual([key] * 7, encryption.get_keys(key * 7))
def key_characters_test(self):
for _ in range(100):
self.assertRegex(encryption.generate_key(), "^[0-9a-z]+$")
self.assertTrue(encryption.is_valid_key(encryption.generate_key()))
def assertInverses(self, data, padding=None):
key = encryption.generate_key()
ciphertext = encryption.encrypt(data, key, padding)
self.assertTrue(encryption.is_encrypted(ciphertext))
self.assertEqual(encryption.decrypt(ciphertext, key), data)
def pad_to_same_size_test(self):
ct1 = encryption.encrypt("hi", encryption.generate_key(), 1000)
ct2 = encryption.encrypt("hi" * 400, encryption.generate_key(), 1000)
self.assertEqual(len(ct1), len(ct2))