[test] use pull producer in blobs fs backend benchmarks

1 files changed, 11 insertions, 0 deletions

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):