split out more legacy config tests

apache · Dec 20, 2024 · df90bba · df90bba
1 parent b54829e
commit df90bba
Showing 1 changed file with 34 additions and 8 deletions.
diff --git a/helm_tests/airflow_aux/test_remote_logging.py b/helm_tests/airflow_aux/test_remote_logging.py
@@ -104,17 +104,16 @@ def test_scheduler_should_omit_log_port_when_elasticsearch_enabled(self):
 
         assert "ports" not in jmespath.search("spec.template.spec.containers[0]", docs[0])
 
-    def test_env_should_omit_elasticsearch_host_vars_if_es_disabled(self):
+    def test_env_should_omit_elasticsearch_host_var_if_es_disabled(self):
         docs = render_chart(
             values={},
             show_only=[SCHEDULER_DEPLOYMENT_TEMPLATE],
         )
 
         scheduler_env_keys = jmespath.search("spec.template.spec.containers[0].env[*].name", docs[0])
         assert "AIRFLOW__ELASTICSEARCH__HOST" not in scheduler_env_keys
-        assert "AIRFLOW__ELASTICSEARCH__ELASTICSEARCH_HOST" not in scheduler_env_keys
 
-    def test_env_should_add_elasticsearch_host_vars_if_es_enabled(self):
+    def test_env_should_add_elasticsearch_host_var_if_es_enabled(self):
         docs = render_chart(
             values={
                 "elasticsearch": {
@@ -125,15 +124,42 @@ def test_env_should_add_elasticsearch_host_vars_if_es_enabled(self):
             show_only=[SCHEDULER_DEPLOYMENT_TEMPLATE],
         )
 
-        expected_value_from = {"secretKeyRef": {"name": "test-elastic-secret", "key": "connection"}}
+        scheduler_env = jmespath.search("spec.template.spec.containers[0].env", docs[0])
+
+        assert {"name": "AIRFLOW__ELASTICSEARCH__HOST", "valueFrom": {
+            "secretKeyRef": {"name": "test-elastic-secret", "key": "connection"}}} in scheduler_env
+
+    def test_env_should_omit_elasticsearch_host_var_if_es_disabled_legacy(self):
+        """AIRFLOW__ELASTICSEARCH__ELASTICSEARCH_HOST was the environment key prior to Airflow 1.10.4
+        (see https://github.com/apache/airflow/pull/5048), this test can be removed when the Helm chart
+        no longer supports Airflow 1.10.3"""
+        docs = render_chart(
+            values={},
+            show_only=[SCHEDULER_DEPLOYMENT_TEMPLATE],
+        )
+
+        scheduler_env_keys = jmespath.search("spec.template.spec.containers[0].env[*].name", docs[0])
+        assert "AIRFLOW__ELASTICSEARCH__ELASTICSEARCH_HOST" not in scheduler_env_keys
 
-        container_env = jmespath.search("spec.template.spec.containers[0].env", docs[0])
+    def test_env_should_add_elasticsearch_host_var_if_es_enabled_legacy(self):
+        """AIRFLOW__ELASTICSEARCH__ELASTICSEARCH_HOST was the environment key prior to Airflow 1.10.4
+        (see https://github.com/apache/airflow/pull/5048), this test can be removed when the Helm chart
+        no longer supports Airflow 1.10.3"""
+        docs = render_chart(
+            values={
+                "elasticsearch": {
+                    "enabled": True,
+                    "secretName": "test-elastic-secret",
+                },
+            },
+            show_only=[SCHEDULER_DEPLOYMENT_TEMPLATE],
+        )
 
-        assert {"name": "AIRFLOW__ELASTICSEARCH__HOST", "valueFrom": expected_value_from} in container_env
+        scheduler_env = jmespath.search("spec.template.spec.containers[0].env", docs[0])
         assert {
             "name": "AIRFLOW__ELASTICSEARCH__ELASTICSEARCH_HOST",
-            "valueFrom": expected_value_from,
-        } in container_env
+            "valueFrom": {"secretKeyRef": {"name": "test-elastic-secret", "key": "connection"}},
+        } in scheduler_env
 
     def test_airflow_cfg_should_set_remote_logging_false_if_es_disabled(self):
         docs = render_chart(