Fix unit test for statsd

apache · Jul 19, 2024 · 9d0784e · 9d0784e
1 parent 4ddbcc6
commit 9d0784e
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/helm_tests/other/test_statsd.py b/helm_tests/other/test_statsd.py
@@ -243,8 +243,9 @@ def test_statsd_configmap_by_default(self):
         mappings_yml = jmespath.search('data."mappings.yml"', docs[0])
         mappings_yml_obj = yaml.safe_load(mappings_yml)
 
-        assert "airflow_dagrun_dependency_check" == mappings_yml_obj["mappings"][0]["name"]
-        assert "airflow_pool_starving_tasks" == mappings_yml_obj["mappings"][-1]["name"]
+        names = [mapping["name"] for mapping in mappings_yml_obj["mappings"]]
+        assert "airflow_dagrun_dependency_check" in names
+        assert "airflow_pool_starving_tasks" in names
 
     def test_statsd_configmap_when_exist_extra_mappings(self):
         extra_mapping = {