[benchmarks] add code for stressing the server

1 files changed, 2 insertions, 20 deletions

diff --git a/scripts/scalability/test_controller/client/templates.py b/scripts/scalability/test_controller/client/templates.py
index dcf51fe1..434eda67 100755
--- a/scripts/scalability/test_controller/client/templates.py
+++ b/scripts/scalability/test_controller/client/templates.py
@@ -1,10 +1,7 @@
 #!/usr/bin/env python
 
-import base64
-import errno
 import mock
 import os
-import random
 
 from argparse import ArgumentParser
 from io import BytesIO
@@ -14,6 +11,8 @@ from twisted.internet import reactor, defer
 
 from leap.soledad.client._db.blobs import BlobManager
 
+from test_controller.utils import mkdir_p, payload
+
 
 DEFAULT_TARGET_DIR = './blob-templates'
 
@@ -33,28 +32,11 @@ def _get_encrypt_function(user, path):
     return manager._encrypt_and_upload
 
 
-def payload(size):
-    random.seed(1337)  # same seed to avoid different bench results
-    payload_bytes = bytearray(random.getrandbits(8) for _ in xrange(size))
-    # encode as base64 to avoid ascii encode/decode errors
-    return base64.b64encode(payload_bytes)[:size]  # remove b64 overhead
-
-
 def _encrypt(path, data):
     encrypt = _get_encrypt_function('user-0', path)
     return encrypt('blob', BytesIO(data))
 
 
-def mkdir_p(path):
-    try:
-        os.makedirs(path)
-    except OSError as exc:  # Python >2.5
-        if exc.errno == errno.EEXIST and os.path.isdir(path):
-            pass
-        else:
-            raise
-
-
 def create_blob_templates(target_dir, amount, size):
     mkdir_p(target_dir)
     data = payload(size)