diff options
| author | drebs <drebs@leap.se> | 2013-05-28 11:41:33 -0300 |
|---|---|---|
| committer | drebs <drebs@leap.se> | 2013-05-28 11:41:33 -0300 |
| commit | 7fc904d797cb3c07f593157df1126b4179fe48d8 (patch) | |
| tree | 6f3beac6a6701d07aa2b8d0e94890aa235a2501f | |
| parent | 2a81dd8b4c885870933f76dec23807f1d5a1a91c (diff) | |
Fix wrong iv test to account for new form of iv.
| -rw-r--r-- | src/leap/common/tests/test_crypto.py | 12 |
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) |