summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--testing/tests/blobs/test_blobs.py6
1 files changed, 2 insertions, 4 deletions
diff --git a/testing/tests/blobs/test_blobs.py b/testing/tests/blobs/test_blobs.py
index 77eff6cc..4efd7115 100644
--- a/testing/tests/blobs/test_blobs.py
+++ b/testing/tests/blobs/test_blobs.py
@@ -23,7 +23,6 @@ from leap.soledad.client._blobs import DecrypterBuffer, BlobManager
from leap.soledad.client import _crypto
from io import BytesIO
from mock import Mock
-import mock
class BlobTestCase(unittest.TestCase):
@@ -66,6 +65,5 @@ class BlobTestCase(unittest.TestCase):
manager = BlobManager('', '', self.secret, self.secret, 'user')
doc_id, rev = self.doc_info.doc_id, self.doc_info.rev
fd = BytesIO('up and up')
- with mock.patch('leap.soledad.client._blobs.treq.put') as mock_treq:
- mock_treq.side_effect = _check_result
- yield manager._encrypt_and_upload('blob_id', doc_id, rev, fd)
+ manager._client.put = _check_result
+ yield manager._encrypt_and_upload('blob_id', doc_id, rev, fd)