From 961e43404e0d23674360860b04ba5480e752cc94 Mon Sep 17 00:00:00 2001 From: Victor Shyba Date: Thu, 20 Jul 2017 04:31:40 -0300 Subject: [refactor] error handling and missing headers Improves error handling, add missing header to __init__.py and remove mkdirs from flags methods. This is a commit from code review on !117. --- testing/tests/server/test_blobs_server.py | 9 +++++++++ 1 file changed, 9 insertions(+) (limited to 'testing/tests') diff --git a/testing/tests/server/test_blobs_server.py b/testing/tests/server/test_blobs_server.py index dde25668..152215d1 100644 --- a/testing/tests/server/test_blobs_server.py +++ b/testing/tests/server/test_blobs_server.py @@ -31,6 +31,7 @@ from leap.soledad.server import _blobs as server_blobs from leap.soledad.client._db.blobs import BlobManager from leap.soledad.client._db.blobs import BlobAlreadyExistsError from leap.soledad.client._db.blobs import InvalidFlagsError +from leap.soledad.client._db.blobs import SoledadError class BlobServerTestCase(unittest.TestCase): @@ -68,6 +69,14 @@ class BlobServerTestCase(unittest.TestCase): flags = yield manager.get_flags('blob_id') self.assertEquals([Flags.PROCESSING], flags) + @defer.inlineCallbacks + @pytest.mark.usefixtures("method_tmpdir") + def test_set_flags_raises_if_no_blob_found(self): + manager = BlobManager('', self.uri, self.secret, + self.secret, 'user') + with pytest.raises(SoledadError): + yield manager.set_flags('missing_id', [Flags.PENDING]) + @defer.inlineCallbacks @pytest.mark.usefixtures("method_tmpdir") def test_list_filter_flag(self): -- cgit v1.2.3