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

Refactor extract_labels function to handle missing "labels" key

Author: mcucchi9

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
 
 

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"]