Merge pull request #179 from ecmwf-projects/fix-free-edition-widget-l…

…abels Refactor extract_labels function to handle missing "labels" key
ecmwf-projects · Mar 7, 2024 · 09818b7 · 09818b7
2 parents 7376895 + 06e3d03
commit 09818b7
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/cads_processing_api_service/translators.py b/cads_processing_api_service/translators.py
@@ -35,11 +35,11 @@ def extract_groups_labels(
 
 
 def extract_labels(input_cds_schema: dict[str, Any]) -> dict[str, str]:
-    details = input_cds_schema["details"]
+    details: dict[str, Any] = input_cds_schema["details"]
     if "groups" in details:
         values = extract_groups_labels(details["groups"])
     else:
-        values = details["labels"]
+        values = details.get("labels", {})
     return values
 
 

diff --git a/tests/test_10_translators.py b/tests/test_10_translators.py
@@ -60,6 +60,10 @@
         },
         "type": "StringListArrayWidget",
     },
+    "free_edition_widget": {
+        "type": "FreeEditionWidget",
+        "details": {},
+    },
 }
 
 
@@ -116,6 +120,13 @@ def test_extract_labels() -> None:
     )
     assert res_output == exp_output
 
+    test_inputs_cds_schema = TEST_INPUT_CDS_SCHEMAS["free_edition_widget"]
+    exp_output = {}
+    res_output = cads_processing_api_service.translators.extract_labels(
+        test_inputs_cds_schema
+    )
+    assert res_output == exp_output
+
 
 def test_translate_string_list() -> None:
     test_input = TEST_INPUT_CDS_SCHEMAS["string_list"]