[feature] add listing to local blob db

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