summaryrefslogtreecommitdiff
path: root/src/leap/common/tests/test_crypto.py
diff options
context:
space:
mode:
authordrebs <drebs@leap.se>2013-05-28 11:41:33 -0300
committerdrebs <drebs@leap.se>2013-05-28 11:41:33 -0300
commit7fc904d797cb3c07f593157df1126b4179fe48d8 (patch)
tree6f3beac6a6701d07aa2b8d0e94890aa235a2501f /src/leap/common/tests/test_crypto.py
parent2a81dd8b4c885870933f76dec23807f1d5a1a91c (diff)
Fix wrong iv test to account for new form of iv.
Diffstat (limited to 'src/leap/common/tests/test_crypto.py')
-rw-r--r--src/leap/common/tests/test_crypto.py12
1 files changed, 10 insertions, 2 deletions
diff --git a/src/leap/common/tests/test_crypto.py b/src/leap/common/tests/test_crypto.py
index b704c05..ae7dc71 100644
--- a/src/leap/common/tests/test_crypto.py
+++ b/src/leap/common/tests/test_crypto.py
@@ -21,6 +21,10 @@ Tests for the crypto submodule.
"""
+import os
+import binascii
+
+
from leap.common.testing.basetest import BaseLeapTest
from leap.common import crypto
from Crypto import Random
@@ -56,9 +60,13 @@ class CryptoTestCase(BaseLeapTest):
self.assertTrue(cyphertext is not None)
self.assertTrue(cyphertext != '')
self.assertTrue(cyphertext != 'data')
- iv += 1
+ # get a different iv by changing the first byte
+ rawiv = binascii.a2b_base64(iv)
+ wrongiv = rawiv
+ while wrongiv == rawiv:
+ wrongiv = os.urandom(1) + rawiv[1:]
plaintext = crypto.decrypt_sym(
- cyphertext, key, iv=iv,
+ cyphertext, key, iv=binascii.b2a_base64(wrongiv),
method=crypto.EncryptionMethods.AES_256_CTR)
self.assertNotEqual('data', plaintext)