Add locust.io load testing script

loadtesting/locustfile.py

@@ -0,0 +1,44 @@
+from locust import HttpLocust, TaskSet
+import tarfile
+import io
+
+patch_version = 1
+
+def index(l):
+    l.client.get("/index.yaml")
+
+def metrics(l):
+    l.client.get("/metrics")
+
+def not_found(l):
+    l.client.get("/toto")
+
+def post_new_chart(l):
+    global patch_version
+
+    # Create dummy 'chartmuseum-loadtest' chart package for which we only increment the patch version
+    chart_name = 'chartmuseum-loadtest'
+    chart_version = '0.0.%d' % patch_version
+    patch_version += 1
+    chart_fn = '%s-%s.tgz' % (chart_name, chart_version)
+
+    tgz_buf = io.BytesIO()
+    t = tarfile.open(mode = "w:gz", fileobj=tgz_buf)
+    chart_content = b'name: %s\nversion: %s\n' % (chart_name.encode('utf8'), chart_version.encode('utf8'))
+    tarinfo = tarfile.TarInfo('%s/Chart.yaml' % chart_name)
+    tarinfo.size = len(chart_content)
+    t.addfile(tarinfo=tarinfo, fileobj=io.BytesIO(chart_content))
+    t.close()
+    tgz_buf.seek(0)
+
+    l.client.post('/api/charts', files={'chartfile': (chart_fn, tgz_buf)})
+
+
+class UserBehavior(TaskSet):
+    tasks = {index: 15, metrics: 1, post_new_chart: 1}
+
+
+class WebsiteUser(HttpLocust):
+    task_set = UserBehavior
+    min_wait = 1000
+    max_wait = 3000