diff options
Diffstat (limited to 'tests/client/test_crypto.py')
-rw-r--r-- | tests/client/test_crypto.py | 13 |
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 |