Mercurial > libervia-backend
diff src/test/test_memory_crypto.py @ 1066:cbf917a90784
test (memory): fixes crypto encryption/decryption test
author | souliane <souliane@mailoo.org> |
---|---|
date | Thu, 12 Jun 2014 15:57:22 +0200 |
parents | 127c96020022 |
children | 77cd312d32c4 |
line wrap: on
line diff
--- a/src/test/test_memory_crypto.py Thu Jun 12 15:56:41 2014 +0200 +++ b/src/test/test_memory_crypto.py Thu Jun 12 15:57:22 2014 +0200 @@ -34,14 +34,18 @@ def test_encrypt_decrypt(self): d_list = [] + + def test(key, message): + d = BlockCipher.encrypt(key, message) + d.addCallback(lambda ciphertext: BlockCipher.decrypt(key, ciphertext)) + d.addCallback(lambda decrypted: self.assertEqual(message, decrypted)) + d_list.append(d) + for key_len in (0, 2, 8, 10, 16, 24, 30, 32, 40): key = urandom(key_len) for message_len in (0, 2, 16, 24, 32, 100): message = urandom(message_len) - d = BlockCipher.encrypt(key, message) - d.addCallback(lambda ciphertext: lambda key, cipher: BlockCipher.decrypt(key, ciphertext)) - d.addCallback(lambda decrypted: lambda message, decrypted: self.assertEqual(message, decrypted)) - d_list.append(d) + test(key, message) return defer.DeferredList(d_list) def test_hash_verify(self):