summaryrefslogtreecommitdiff
path: root/testing/tests
diff options
context:
space:
mode:
Diffstat (limited to 'testing/tests')
-rw-r--r--testing/tests/client/test_crypto.py5
1 files changed, 2 insertions, 3 deletions
diff --git a/testing/tests/client/test_crypto.py b/testing/tests/client/test_crypto.py
index 7643f75d..aad588c0 100644
--- a/testing/tests/client/test_crypto.py
+++ b/testing/tests/client/test_crypto.py
@@ -52,7 +52,7 @@ class AESTest(unittest.TestCase):
key = 'A' * 32
fd = BytesIO()
- aes = _crypto.AESConsumer(key, _buffer=fd)
+ aes = _crypto.AESWriter(key, _buffer=fd)
iv = aes.iv
data = snowden1
@@ -78,8 +78,7 @@ class AESTest(unittest.TestCase):
ciphertext = _aes_encrypt(key, iv, data)
fd = BytesIO()
- operation = _crypto.AESConsumer.decrypt
- aes = _crypto.AESConsumer(key, iv, fd, operation)
+ aes = _crypto.AESWriter(key, iv, fd, encrypt=False)
for i in range(len(ciphertext) / block):
chunk = ciphertext[i * block:(i + 1) * block]