diff --git a/src/leap/soledad/server/_blobs.py b/src/leap/soledad/server/_blobs.py index 5670c562f094fc4c34eefb75173be3f716a16f52..5f5ae5acdb919f13fbf199e9419d6f140b8a978c 100644 --- a/src/leap/soledad/server/_blobs.py +++ b/src/leap/soledad/server/_blobs.py @@ -144,7 +144,8 @@ class FilesystemBlobsBackend(object): pass def get_blob_size(self, user, blob_id, namespace=''): - raise NotImplementedError + blob_path = self._get_path(user, blob_id, namespace) + return os.stat(blob_path).st_size def count(self, user, request, namespace=''): base_path = self._get_path(user, namespace=namespace) diff --git a/tests/blobs/test_fs_backend.py b/tests/blobs/test_fs_backend.py index 47d78ad4bc46121911e64921d38d79f072c58d70..94b628c439116686f51f5d7420460f61a0261dfc 100644 --- a/tests/blobs/test_fs_backend.py +++ b/tests/blobs/test_fs_backend.py @@ -51,6 +51,19 @@ class FilesystemBackendTestCase(unittest.TestCase): expected_method.assert_called_once_with('Tag', [expected_tag]) + @pytest.mark.usefixtures("method_tmpdir") + def test_get_blob_size(self): + # get a backend + backend = _blobs.FilesystemBlobsBackend(blobs_path=self.tempdir) + # write a blob with size=10 + path = backend._get_path('user', 'blob_id', '') + mkdir_p(os.path.split(path)[0]) + with open(path, "w") as f: + f.write("0123456789") + # check it's size + size = backend.get_blob_size('user', 'blob_id', '') + self.assertEquals(10, size) + @pytest.mark.usefixtures("method_tmpdir") @mock.patch.object(_blobs.static, 'File') @mock.patch.object(_blobs.FilesystemBlobsBackend, '_get_path',