[test] ciphertext size function tests

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