Add a little timing.

benchmarks/test_immutable.py

@@ -8,6 +8,9 @@ TODO Parameterization (pytest?)
 - Number of needed/happy/total shares.
 """
 
+from time import time
+from contextlib import contextmanager
+
 from twisted.trial.unittest import TestCase
 
 from allmydata.util.deferredutil import async_to_deferred
@@ -16,16 +19,25 @@ from allmydata.test.common_system import SystemTestMixin
 from allmydata.immutable.upload import Data as UData
 
 
+@contextmanager
+def timeit(name):
+    start = time()
+    try:
+        yield
+    finally:
+        print(f"{name}: {time() - start:.3f}")
+
+
 class ImmutableBenchmarks(SystemTestMixin, TestCase):
     """Benchmarks for immutables."""
 
-    FORCE_FOOLSCAP_FOR_STORAGE = True
+    FORCE_FOOLSCAP_FOR_STORAGE = False
 
     @async_to_deferred
     async def test_upload_and_download(self):
         self.basedir = self.mktemp()
 
-        DATA = b"Some data to upload\n" * 2000
+        DATA = b"Some data to upload\n" * 10
 
         # 3 nodes
         await self.set_up_nodes(3)
@@ -36,11 +48,14 @@ class ImmutableBenchmarks(SystemTestMixin, TestCase):
             c.encoding_params["happy"] = 3
             c.encoding_params["n"] = 3
 
+        for i in range(5):
             # 1. Upload:
+            with timeit("upload"):
                 uploader = self.clients[0].getServiceNamed("uploader")
                 results = await uploader.upload(UData(DATA, convergence=None))
 
             # 2. Download:
+            with timeit("download"):
                 uri = results.get_uri()
                 node = self.clients[1].create_node_from_uri(uri)
                 mc = await node.read(MemoryConsumer(), 0, None)