From 7712e6b19def97cf0266146863d11d5a29767a6a Mon Sep 17 00:00:00 2001
From: drebs <drebs@leap.se>
Date: Fri, 8 Dec 2017 22:36:24 -0200
Subject: [PATCH] [refactor] make get_total_storage() return a deferred
---
src/leap/soledad/server/interfaces.py | 4 ++++
tests/blobs/test_fs_backend.py | 2 +-
2 files changed, 5 insertions(+), 1 deletion(-)
diff --git a/src/leap/soledad/server/interfaces.py b/src/leap/soledad/server/interfaces.py
index 7009cb85..c00b1d91 100644
--- a/src/leap/soledad/server/interfaces.py
+++ b/src/leap/soledad/server/interfaces.py
@@ -143,6 +143,10 @@ class IBlobsBackend(Interface):
:return: The size in units of 1024 bytes.
:rtype: int
+
+ :return: A deferred that fires with the amount of storage used in units
+ of 1024 bytes.
+ :rtype: twisted.internet.defer.Deferred
"""
def get_tag(user, blob_id, namespace=''):
diff --git a/tests/blobs/test_fs_backend.py b/tests/blobs/test_fs_backend.py
index 727ffe92..6c5665e8 100644
--- a/tests/blobs/test_fs_backend.py
+++ b/tests/blobs/test_fs_backend.py
@@ -96,7 +96,7 @@ class FilesystemBackendTestCase(unittest.TestCase):
def test_write_cannot_exceed_quota(self, isfile):
isfile.return_value = False
backend = _blobs.FilesystemBlobsBackend(blobs_path=self.tempdir)
- backend.get_total_storage = lambda x: 100
+ backend.get_total_storage = lambda x: defer.succeed(100)
backend.quota = 90
with pytest.raises(_blobs.QuotaExceeded):
fd = Mock()
--
GitLab