FakeDataInclusionClient: utilisation mapping thematiques

dora/data_inclusion/test_utils.py

@@ -1,6 +1,8 @@
 from typing import Optional
 from uuid import uuid4
 
+from .constants import THEMATIQUES_MAPPING_DORA_TO_DI
+
 
 def make_di_service_data(**kwargs) -> dict:
     return {
@@ -99,13 +101,18 @@ def search_services(
             services = [r for r in services if r["source"] in sources]
 
         if thematiques is not None:
+            enriched_thematiques = []
+            for thematique in thematiques:
+                enriched_thematiques += THEMATIQUES_MAPPING_DORA_TO_DI.get(
+                    thematique, [thematique]
+                )
             services = [
                 r
                 for r in services
                 if any(
                     t.startswith(requested_thematique)
                     for t in r["thematiques"]
-                    for requested_thematique in thematiques
+                    for requested_thematique in enriched_thematiques
                 )
             ]
 

dora/data_inclusion/tests.py

@@ -19,5 +19,5 @@ def test_map_service_thematiques_mapping():
     di_service_data = make_di_service_data(thematiques=input_thematiques)
     service = map_service(di_service_data, False)
 
-    assert set(service["categories"]) == set(expected_categories)
-    assert set(service["subcategories"]) == set(expected_subcategories)
+    assert sorted(service["categories"]) == sorted(expected_categories)
+    assert sorted(service["subcategories"]) == sorted(expected_subcategories)