diff --git a/tests/benchmarks/test_blobs_fs_backend.py b/tests/benchmarks/test_blobs_fs_backend.py index 6a74f51430bb5d166c99bea2bd6d94fd89222a8d..9f1ebf944834343a53c3ba57714be05ddf13cd14 100644 --- a/tests/benchmarks/test_blobs_fs_backend.py +++ b/tests/benchmarks/test_blobs_fs_backend.py @@ -3,6 +3,7 @@ from io import BytesIO from leap.soledad.server._blobs import FilesystemBlobsBackend from twisted.internet import defer from twisted.web.client import FileBodyProducer +from twisted.internet._producer_helpers import _PullToPush def create_write_test(amount, size): @@ -37,6 +38,16 @@ class DevNull(object): def write(self, data): pass + def registerProducer(self, producer, streaming): + producer = _PullToPush(producer, self) + producer.startStreaming() + + def unregisterProducer(self): + pass + + def finish(self): + pass + def create_read_test(amount, size):