1 files changed, 80 insertions, 0 deletions
diff --git a/scripts/scalability/test_controller/client/test_Blobs.py b/scripts/scalability/test_controller/client/test_Blobs.py
new file mode 100644
index 00000000..5f31dbc8
--- /dev/null
+++ b/scripts/scalability/test_controller/client/test_Blobs.py
@@ -0,0 +1,80 @@
+import base64
+import os
+import subprocess
+import unittest
+import urlparse
+import threading
+
+from funkload.FunkLoadTestCase import FunkLoadTestCase
+from webunit.utility import Upload
+
+
+SIZE = 100  # KB
+DELETE_ON_SETUP = False
+DELETE_ON_TEARDOWN = False
+
+
+def _get_auth_header(user_id):
+    encoded = base64.b64encode('%s:%s-token' % (user_id, user_id))
+    return 'Token %s' % encoded
+
+
+def _ensure_template(templates_dir, size):
+    fname = '%dK-0.blob' % size
+    fpath = os.path.join(templates_dir, fname)
+    if not os.path.isfile(fpath):
+        dirname = os.path.dirname(os.path.realpath(__file__))
+        executable = os.path.join(dirname, 'templates.py')
+        code = subprocess.check_call([
+            executable,
+            '--amount', '1',
+            '--size', str(size),
+            templates_dir,
+        ])
+        assert code == 0, 'failed creating template'
+    return fpath
+
+
+class Blobs(FunkLoadTestCase):
+
+    next_user_id = 0
+    lock = threading.Lock()
+
+    def _get_next_user_id(self):
+        with Blobs.lock:
+            user_id = Blobs.next_user_id
+            Blobs.next_user_id += 1
+            Blobs.next_user_id %= 5000
+        return user_id
+
+    def setUp(self):
+        user_id = self._get_next_user_id()
+        base_url = self.conf_get('main', 'url')
+        self.url = urlparse.urljoin(base_url, 'blobs/%s/0' % user_id)
+        self.setHeader('Authorization', _get_auth_header(user_id))
+        templates_dir = self.conf_get('main', 'templates_dir')
+        size = self.conf_getInt('main', 'size')
+        fpath = _ensure_template(templates_dir, size)
+        self.upload = ['file', Upload(fpath)]
+        if DELETE_ON_SETUP:
+            ret = self.delete(self.url, description='Delete blob on setUp')
+            self.assert_(ret.code in [404, 200], 'expected 404 or 200')
+
+    def tearDown(self):
+        if DELETE_ON_TEARDOWN:
+            ret = self.delete(self.url, description='Delete blob on tearDown')
+            self.assert_(ret.code in [404, 200], 'expected 404 or 200')
+        self.clearHeaders()
+
+    def test_upload(self):
+        ret = self.put(self.url, params=[self.upload],
+                       description='Upload blob')
+        self.assert_(ret.code == 200, "expecting a 200")
+
+    def test_download(self):
+        ret = self.get(self.url, description='Download blob')
+        self.assert_(ret.code == 200, "expecting a 200")
+
+
+if __name__ == '__main__':
+    unittest.main()