kube burner configurable

canonical · Nov 13, 2024 · 4f12cc7 · 4f12cc7
1 parent 5d2fb16
commit 4f12cc7
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/test/performance/tests/test_util/config.py b/test/performance/tests/test_util/config.py
@@ -26,6 +26,9 @@
 # SNAP_NAME is the name of the snap under test.
 SNAP_NAME = os.getenv("TEST_SNAP_NAME") or "k8s"
 
+# KUBE_BURNER_URL is the version of kube-burner to use.
+KUBE_BURNER_URL = os.getenv("TEST_KUBE_BURNER_VERSION") or "https://github.com/kube-burner/kube-burner/releases/download/v1.2/kube-burner-1.2-Linux-x86_64.tar.gz"
+
 # FLAVOR is the flavour to use for running the performance tests.
 FLAVOR = os.getenv("TEST_FLAVOR") or ""
 

diff --git a/test/performance/tests/test_util/metrics.py b/test/performance/tests/test_util/metrics.py
@@ -51,13 +51,14 @@ def configure_kube_burner(instance: harness.Instance):
         not instance.exec(["test", "-f", "/root/kube-burner"], check=False).returncode
         == 0
     ):
-        url = "https://github.com/kube-burner/kube-burner/releases/download/v1.2/kube-burner-1.2-Linux-x86_64.tar.gz"
+        url = config.KUBE_BURNER_URL
         instance.exec(["wget", url])
         instance.exec(
             ["tar", "-zxvf", "kube-burner-1.2-Linux-x86_64.tar.gz", "kube-burner"]
         )
         instance.exec(["rm", "kube-burner-1.2-Linux-x86_64.tar.gz"])
         instance.exec(["chmod", "+x", "/root/kube-burner"])
+
     instance.exec(["mkdir", "-p", "/root/templates"])
     instance.send_file(
         (config.MANIFESTS_DIR / "api-intensive.yaml").as_posix(),