diff --git a/chart/tests/helm_tests/airflow_core/test_worker_sets.py b/chart/tests/helm_tests/airflow_core/test_worker_sets.py index 9892cda47321d..040b0919240b7 100644 --- a/chart/tests/helm_tests/airflow_core/test_worker_sets.py +++ b/chart/tests/helm_tests/airflow_core/test_worker_sets.py @@ -1588,6 +1588,68 @@ def test_overwrite_kerberos_sidecar_disable(self): jmespath.search("spec.template.spec.containers[?name=='worker-kerberos'] | [0]", docs[0]) is None ) + @pytest.mark.parametrize( + ("startup_probe", "worker_set_startup_probe", "expected"), + [ + ( + {"enabled": False}, + {"enabled": True}, + { + "exec": {"command": ["klist", "-s"]}, + "timeoutSeconds": 5, + "initialDelaySeconds": 0, + "periodSeconds": 10, + "failureThreshold": 6, + }, + ), + ({"enabled": True}, {"enabled": False}, None), + ( + { + "timeoutSeconds": 1, + "initialDelaySeconds": 2, + "periodSeconds": 3, + "failureThreshold": 4, + }, + {"timeoutSeconds": 11}, + { + "exec": {"command": ["klist", "-s"]}, + "timeoutSeconds": 11, + "initialDelaySeconds": 2, + "periodSeconds": 3, + "failureThreshold": 4, + }, + ), + ], + ids=["enabled", "disabled", "custom"], + ) + def test_overwrite_kerberos_sidecar_startup_probe( + self, startup_probe, worker_set_startup_probe, expected + ): + docs = render_chart( + values={ + "workers": { + "celery": { + "enableDefault": False, + "kerberosSidecar": {"enabled": True, "startupProbe": startup_probe}, + "sets": [ + { + "name": "test", + "kerberosSidecar": {"startupProbe": worker_set_startup_probe}, + } + ], + } + } + }, + show_only=["templates/workers/worker-deployment.yaml"], + ) + + assert ( + jmespath.search( + "spec.template.spec.containers[?name=='worker-kerberos'] | [0].startupProbe", docs[0] + ) + == expected + ) + @pytest.mark.parametrize( "values", [