summaryrefslogtreecommitdiff
path: root/tests
diff options
context:
space:
mode:
Diffstat (limited to 'tests')
-rw-r--r--tests/benchmarks/test_blobs_upload_download.py2
-rw-r--r--tests/benchmarks/test_legacy_vs_blobs.py2
-rw-r--r--tests/blobs/test_blob_manager.py3
3 files changed, 4 insertions, 3 deletions
diff --git a/tests/benchmarks/test_blobs_upload_download.py b/tests/benchmarks/test_blobs_upload_download.py
index 4489842a..229c6567 100644
--- a/tests/benchmarks/test_blobs_upload_download.py
+++ b/tests/benchmarks/test_blobs_upload_download.py
@@ -9,7 +9,7 @@ from io import BytesIO
from twisted.internet.defer import gatherResults
from twisted.internet.defer import returnValue
-from leap.soledad.client._db.blobs import BlobDoc
+from leap.soledad.client._document import BlobDoc
def payload(size):
diff --git a/tests/benchmarks/test_legacy_vs_blobs.py b/tests/benchmarks/test_legacy_vs_blobs.py
index 47d6482c..1a42a9dc 100644
--- a/tests/benchmarks/test_legacy_vs_blobs.py
+++ b/tests/benchmarks/test_legacy_vs_blobs.py
@@ -59,7 +59,7 @@ from twisted.internet.defer import returnValue
from twisted.internet.defer import DeferredSemaphore
from leap.soledad.common.blobs import Flags
-from leap.soledad.client._db.blobs import BlobDoc
+from leap.soledad.client._document import BlobDoc
def payload(size):
diff --git a/tests/blobs/test_blob_manager.py b/tests/blobs/test_blob_manager.py
index c151d8c3..26c6506e 100644
--- a/tests/blobs/test_blob_manager.py
+++ b/tests/blobs/test_blob_manager.py
@@ -19,7 +19,8 @@ Tests for BlobManager.
"""
from twisted.trial import unittest
from twisted.internet import defer
-from leap.soledad.client._db.blobs import BlobManager, BlobDoc, FIXED_REV
+from leap.soledad.client._document import BlobDoc
+from leap.soledad.client._db.blobs import BlobManager, FIXED_REV
from leap.soledad.client._db.blobs import BlobAlreadyExistsError
from leap.soledad.client._db.blobs import SyncStatus
from leap.soledad.client._db.blobs import RetriableTransferError