Skip to content

Commit

Permalink
test: Add helm-test for mysql keda support
Browse files Browse the repository at this point in the history
Testing that required key are present for the trigger definition and at the right value

Signed-off-by: Romain Forlot <romain.forlot@dailymotion.com>
  • Loading branch information
claneys committed Dec 19, 2023
1 parent 07ef735 commit 622e73e
Showing 1 changed file with 40 additions and 0 deletions.
40 changes: 40 additions & 0 deletions helm_tests/other/test_keda.py
Original file line number Diff line number Diff line change
Expand Up @@ -275,3 +275,43 @@ def test_default_keda_db_connection_pgbouncer_enabled_usePgbouncer_false(self):
"spec.triggers[0].metadata.connectionFromEnv", keda_autoscaler
)
assert autoscaler_connection_env_var == "KEDA_DB_CONN"

def test_mysql_keda_db_connection(self):
"""Verify keda db connection when pgbouncer is enabled."""
import base64

docs = render_chart(
values={
"data": {"metadataConnection": {"protocol": "mysql", "port":3306} },
"workers": {"keda": {"enabled": True}},
"executor": "CeleryExecutor",
},
show_only=[
"templates/workers/worker-deployment.yaml",
"templates/workers/worker-kedaautoscaler.yaml",
"templates/secrets/metadata-connection-secret.yaml",
],
)
worker_deployment = docs[0]
keda_autoscaler = docs[1]
metadata_connection_secret = docs[2]

worker_container_env_vars = jmespath.search(
"spec.template.spec.containers[?name=='worker'].env[].name", worker_deployment
)
assert "AIRFLOW_CONN_AIRFLOW_DB" in worker_container_env_vars
assert "KEDA_DB_CONN" in worker_container_env_vars

keda_autoscaler_metadata = jmespath.search(
"spec.triggers[0].metadata", keda_autoscaler
)
assert "queryValue" in keda_autoscaler_metadata

secret_data = jmespath.search("data", metadata_connection_secret)
assert "connection" in secret_data.keys()
assert "kedaConnection" in secret_data.keys()

autoscaler_connection_env_var = jmespath.search(
"spec.triggers[0].metadata.connectionStringFromEnv", keda_autoscaler
)
assert autoscaler_connection_env_var == "KEDA_DB_CONN"

0 comments on commit 622e73e

Please sign in to comment.