# -*- coding: utf-8 -*- # test_crypto.py # Copyright (C) 2017 LEAP # # This program is free software: you can redistribute it and/or modify # it under the terms of the GNU General Public License as published by # the Free Software Foundation, either version 3 of the License, or # (at your option) any later version. # # This program is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU General Public License for more details. # # You should have received a copy of the GNU General Public License # along with this program. If not, see . """ Tests for cryptographic related stuff. """ from twisted.trial import unittest from twisted.internet import defer from leap.soledad.client import _blobs from leap.soledad.client._blobs import DecrypterBuffer from leap.soledad.client import _crypto from io import BytesIO class BlobTestCase(unittest.TestCase): class doc_info: doc_id = 'D-deadbeef' rev = '397932e0c77f45fcb7c3732930e7e9b2:1' def setUp(self): self.cleartext = BytesIO('rosa de foc') self.secret = 'A' * 96 self.blob = _crypto.BlobEncryptor( self.doc_info, self.cleartext, armor=False, secret='A' * 96) @defer.inlineCallbacks def test_decrypt_buffer(self): encrypted = (yield self.blob.encrypt()).getvalue() doc_id, rev = self.doc_info.doc_id, self.doc_info.rev tag = encrypted[-16:] buf = DecrypterBuffer(doc_id, rev, self.secret, tag) buf.write(encrypted) fd, size = buf.close() assert fd.getvalue() == 'rosa de foc' def test_blob_manager_encrypted_upload(self): @defer.inlineCallbacks def _check_result(uri, data): decryptor = _crypto.BlobDecryptor( self.doc_info, data, armor=False, secret=self.secret) decrypted = yield decryptor.decrypt() assert decrypted.getvalue() == 'up and up' manager = _blobs.BlobManager( '', 'http://127.0.0.1/', self.secret, self.secret, 'user') doc_id, rev = self.doc_info.doc_id, self.doc_info.rev fd = BytesIO('up and up') manager._client.put = _check_result return manager._encrypt_and_upload('blob_id', doc_id, rev, fd)