From 0f2bd936335ee64e052aba584ecff851af1418a1 Mon Sep 17 00:00:00 2001 From: drebs Date: Wed, 27 Dec 2017 14:07:52 -0200 Subject: [test] use pull producer in blobs fs backend benchmarks --- tests/benchmarks/test_blobs_fs_backend.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/tests/benchmarks/test_blobs_fs_backend.py b/tests/benchmarks/test_blobs_fs_backend.py index 6a74f514..9f1ebf94 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): -- cgit v1.2.3