From 1bdfb12b98c6db2b54014b207fe05f235a0ae4bb Mon Sep 17 00:00:00 2001 From: Victor Shyba Date: Mon, 20 Mar 2017 21:46:10 -0300 Subject: [test] upload/download integration test --- server/src/leap/soledad/server/_blobs.py | 1 + testing/tests/server/test_blobs_server.py | 52 +++++++++++++++++++++++++++++++ 2 files changed, 53 insertions(+) create mode 100644 testing/tests/server/test_blobs_server.py diff --git a/server/src/leap/soledad/server/_blobs.py b/server/src/leap/soledad/server/_blobs.py index c9b78005..4317deff 100644 --- a/server/src/leap/soledad/server/_blobs.py +++ b/server/src/leap/soledad/server/_blobs.py @@ -178,6 +178,7 @@ class BlobsResource(resource.Resource): self._handler = kls() """ resource.Resource.__init__(self) + self._blobs_path = blobs_path self._handler = self.blobsFactoryClass(blobs_path) assert IBlobsBackend.providedBy(self._handler) diff --git a/testing/tests/server/test_blobs_server.py b/testing/tests/server/test_blobs_server.py new file mode 100644 index 00000000..2c52ea46 --- /dev/null +++ b/testing/tests/server/test_blobs_server.py @@ -0,0 +1,52 @@ +# -*- 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 . +""" +Integration tests for blobs server +""" +import pytest +from io import BytesIO +from twisted.trial import unittest +from twisted.web.server import Site +from twisted.internet import reactor +from twisted.internet import defer +from leap.soledad.server import _blobs as server_blobs +from leap.soledad.client._blobs import BlobManager + + +class BlobServerTestCase(unittest.TestCase): + + def setUp(self): + root = server_blobs.BlobsResource(self.tempdir) + site = Site(root) + self.port = reactor.listenTCP(0, site, interface='127.0.0.1') + self.host = self.port.getHost() + self.uri = 'http://%s:%s/' % (self.host.host, self.host.port) + self.secret = 'A' * 96 + + def tearDown(self): + self.port.stopListening() + + @defer.inlineCallbacks + @pytest.mark.usefixtures("method_tmpdir") + def test_upload(self): + manager = BlobManager('', self.uri, self.secret, + self.secret, 'user') + fd = BytesIO("save me") + yield manager._encrypt_and_upload('blob_id', 'mydoc', '1', fd) + blob, size = yield manager._download_and_decrypt('blob_id', + 'mydoc', '1') + assert blob.getvalue() == "save me" -- cgit v1.2.3