mirror of
https://github.com/tahoe-lafs/tahoe-lafs.git
synced 2024-12-24 07:06:41 +00:00
104 lines
2.9 KiB
Python
104 lines
2.9 KiB
Python
"""
|
|
First attempt at benchmarking uploads and downloads.
|
|
|
|
To run:
|
|
|
|
$ pytest benchmarks/upload_download.py -s -v -Wignore
|
|
|
|
TODO Parameterization (pytest?)
|
|
- Foolscap vs not foolscap
|
|
- Number of nodes
|
|
- Data size
|
|
- Number of needed/happy/total shares.
|
|
"""
|
|
|
|
from time import time, process_time
|
|
from contextlib import contextmanager
|
|
from tempfile import mkdtemp
|
|
import os
|
|
|
|
from twisted.trial.unittest import TestCase
|
|
|
|
from allmydata.util.deferredutil import async_to_deferred
|
|
from allmydata.util.consumer import MemoryConsumer
|
|
from allmydata.test.common_system import SystemTestMixin
|
|
from allmydata.immutable.upload import Data as UData
|
|
from allmydata.mutable.publish import MutableData
|
|
|
|
|
|
@contextmanager
|
|
def timeit(name):
|
|
start = time()
|
|
start_cpu = process_time()
|
|
try:
|
|
yield
|
|
finally:
|
|
print(
|
|
f"{name}: {time() - start:.3f} elapsed, {process_time() - start_cpu:.3f} CPU"
|
|
)
|
|
|
|
|
|
class ImmutableBenchmarks(SystemTestMixin, TestCase):
|
|
"""Benchmarks for immutables."""
|
|
|
|
# To use Foolscap, change to True:
|
|
FORCE_FOOLSCAP_FOR_STORAGE = False
|
|
|
|
@async_to_deferred
|
|
async def setUp(self):
|
|
SystemTestMixin.setUp(self)
|
|
self.basedir = os.path.join(mkdtemp(), "nodes")
|
|
|
|
# 2 nodes
|
|
await self.set_up_nodes(2)
|
|
|
|
# 1 share
|
|
for c in self.clients:
|
|
c.encoding_params["k"] = 1
|
|
c.encoding_params["happy"] = 1
|
|
c.encoding_params["n"] = 1
|
|
|
|
print()
|
|
|
|
@async_to_deferred
|
|
async def test_upload_and_download_immutable(self):
|
|
# To test larger files, change this:
|
|
DATA = b"Some data to upload\n" * 10
|
|
|
|
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)
|
|
self.assertEqual(b"".join(mc.chunks), DATA)
|
|
|
|
@async_to_deferred
|
|
async def test_upload_and_download_mutable(self):
|
|
# To test larger files, change this:
|
|
DATA = b"Some data to upload\n" * 10
|
|
|
|
# 1 node
|
|
await self.set_up_nodes(1)
|
|
|
|
# 1 share
|
|
for c in self.clients:
|
|
c.encoding_params["k"] = 1
|
|
c.encoding_params["happy"] = 1
|
|
c.encoding_params["n"] = 1
|
|
|
|
for i in range(5):
|
|
# 1. Upload:
|
|
with timeit(" upload"):
|
|
result = await self.clients[0].create_mutable_file(MutableData(DATA))
|
|
|
|
# 2. Download:
|
|
with timeit("download"):
|
|
data = await result.download_best_version()
|
|
self.assertEqual(data, DATA)
|