summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--client/src/leap/soledad/client/_blobs.py10
-rw-r--r--testing/tests/blobs/test_local_backend.py15
2 files changed, 25 insertions, 0 deletions
diff --git a/client/src/leap/soledad/client/_blobs.py b/client/src/leap/soledad/client/_blobs.py
index f9f1515b..e781421e 100644
--- a/client/src/leap/soledad/client/_blobs.py
+++ b/client/src/leap/soledad/client/_blobs.py
@@ -163,6 +163,9 @@ class BlobManager(object):
data = yield self._client.get(uri)
defer.returnValue((yield data.json()))
+ def local_list(self):
+ return self.local.list()
+
@defer.inlineCallbacks
def put(self, doc, size):
fd = doc.blob_fd
@@ -289,6 +292,13 @@ class SQLiteBlobBackend(object):
if result:
defer.returnValue(BytesIO(str(result[0][0])))
+ @defer.inlineCallbacks
+ def list(self):
+ query = 'select blob_id from blobs'
+ result = yield self.dbpool.runQuery(query)
+ if result:
+ defer.returnValue([b_id[0] for b_id in result])
+
def _init_blob_table(conn):
maybe_create = (
diff --git a/testing/tests/blobs/test_local_backend.py b/testing/tests/blobs/test_local_backend.py
index 1e23606c..79f433fe 100644
--- a/testing/tests/blobs/test_local_backend.py
+++ b/testing/tests/blobs/test_local_backend.py
@@ -90,3 +90,18 @@ class SQLCipherBlobsClientTestCase(unittest.TestCase):
assert result.getvalue() == msg
assert self.manager._encrypt_and_upload.called
assert not self.manager._download_and_decrypt.called
+
+ @defer.inlineCallbacks
+ @pytest.mark.usefixtures("method_tmpdir")
+ def test_local_list_blobs(self):
+ self.manager._encrypt_and_upload = Mock(return_value=None)
+ msg = "1337"
+ doc = BlobDoc('mydoc_id', 'mydoc_rev', BytesIO(msg),
+ blob_id='myblob_id')
+ yield self.manager.put(doc, size=len(msg))
+ doc2 = BlobDoc('mydoc_id2', 'mydoc_rev2', BytesIO(msg),
+ blob_id='myblob_id2')
+ yield self.manager.put(doc2, size=len(msg))
+ blobs_list = yield self.manager.local_list()
+
+ assert 'myblob_id' in blobs_list and 'myblob_id2' in blobs_list