summaryrefslogtreecommitdiff
path: root/tests/client/test_crypto.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/client/test_crypto.py')
-rw-r--r--tests/client/test_crypto.py13
1 files changed, 13 insertions, 0 deletions
diff --git a/tests/client/test_crypto.py b/tests/client/test_crypto.py
index 5b647b73..8a040cd4 100644
--- a/tests/client/test_crypto.py
+++ b/tests/client/test_crypto.py
@@ -127,6 +127,19 @@ class BlobTestCase(unittest.TestCase):
assert decrypted.getvalue() == snowden1
@defer.inlineCallbacks
+ def test_get_unarmored_ciphertext_size(self):
+ for size_to_test in xrange(-1, 400):
+ test_content = '\x00' * size_to_test
+ size = _crypto.get_unarmored_ciphertext_size(len(test_content))
+ inf = BytesIO(test_content)
+ blob = _crypto.BlobEncryptor(
+ self.doc_info, inf,
+ armor=False,
+ secret='A' * 96)
+ encrypted = yield blob.encrypt()
+ assert len(encrypted.getvalue()) == size, size_to_test
+
+ @defer.inlineCallbacks
def test_default_armored_blob_encrypt(self):
encrypted = yield self.blob.encrypt()
decode = base64.urlsafe_b64decode