From ec673a5d674c4ee666914bbad0a8c933822edd65 Mon Sep 17 00:00:00 2001
From: Andrey Taranik <andrey@cicd.team>
Date: Thu, 30 Sep 2021 20:33:00 +0300
Subject: [PATCH] bulk tenant create test added

---
 .../performance/test_bulk_tenant_create.py    | 45 +++++++++++++++++++
 1 file changed, 45 insertions(+)
 create mode 100644 test_runner/performance/test_bulk_tenant_create.py

diff --git a/test_runner/performance/test_bulk_tenant_create.py b/test_runner/performance/test_bulk_tenant_create.py
new file mode 100644
index 0000000000..e2b41aa500
--- /dev/null
+++ b/test_runner/performance/test_bulk_tenant_create.py
@@ -0,0 +1,45 @@
+import timeit
+import pytest
+
+from contextlib import closing
+from fixtures.zenith_fixtures import (
+    TenantFactory,
+    ZenithCli,
+    PostgresFactory,
+)
+
+pytest_plugins = ("fixtures.benchmark_fixture")
+
+# Run bulk tenant cration test.
+#
+# Collects metrics:
+#
+# 1. Time to create {1,10,50} tenants
+# 2. Average creation time per tenant
+
+@pytest.mark.parametrize('tenants_count', [1, 10, 50])
+def test_bulk_tenant_create(
+    zenith_cli: ZenithCli,
+    tenant_factory: TenantFactory,
+    postgres: PostgresFactory,
+    tenants_count: int,
+    zenbenchmark,
+):
+    """Measure tenant creation time"""
+    start = timeit.default_timer()
+    for i in range(tenants_count):
+        tenant = tenant_factory.create()
+        zenith_cli.run(["branch", f"test_bulk_tenant_create_{tenants_count}_{i}", "main", f"--tenantid={tenant}"])
+        pg_tenant = postgres.create_start(
+            f"test_bulk_tenant_create_{tenants_count}_{i}",
+            tenant,
+            wal_acceptors=None,
+        )
+        with closing(pg_tenant.connect()) as conn:
+            with conn.cursor() as cur:
+                cur.execute("SELECT 1;")
+                assert cur.fetchone() == (1,)
+    end = timeit.default_timer()
+
+    zenbenchmark.record('tenant_creation_time_total', end - start, 's')
+    zenbenchmark.record('tenant_creation_time_avg', (end - start) / tenants_count, 's')