feat: support both urgUnits and orgunits from IASO

openhexa/toolbox/iaso/iaso.py

@@ -58,7 +58,12 @@ def get_org_units(self, page: int = 1, limit: int = 10, **kwargs) -> dict:
         params = kwargs
         params.update({"page": page, "limit": limit})
         response = self.api_client.get("/api/orgunits", params=params)
-        return response.json().get("orgunits")
+        json_response = response.json()
+        if "orgunits" in json_response:
+            return json_response.get("orgunits")
+        elif "orgUnits" in json_response:
+            return json_response.get("orgUnits")
+        return []
 
     def get_form_instances(
         self,

tests/iaso/fixtures/iaso_api_fixtures.py

@@ -43,6 +43,22 @@
 }
 
 iaso_mocked_orgunits = {
+    "orgUnits": [
+        {
+            "name": "ACEH",
+            "id": 1978297,
+            "parent_id": 1978331,
+            "org_unit_type_id": 781,
+            "org_unit_type_name": "Province",
+            "validation_status": "VALID",
+            "created_at": 1712825023.085615,
+            "updated_at": 1712828860.665764,
+        }
+    ]
+}
+
+
+iaso_mocked_orgunits_with_params = {
     "orgunits": [
         {
             "name": "ACEH",

tests/iaso/test_iaso.py

@@ -9,7 +9,7 @@
     iaso_mocked_orgunits,
     iaso_mocked_refreshed_auth_token,
     iaso_mocked_projects,
-    iaso_mocked_instances,
+    iaso_mocked_instances, iaso_mocked_orgunits_with_params,
 )
 
 
@@ -50,7 +50,16 @@ def test_get_org_units(self, mock_responses):
         iaso = IASO("https://iaso-staging.bluesquare.org", "username", "password")
         r = iaso.get_org_units()
         assert len(r) > 0
-
+    def test_get_org_units_with_params(self, mock_responses):
+        mock_responses.add(
+            responses.POST, "https://iaso-staging.bluesquare.org/api/token/", json=iaso_mocked_auth_token, status=200
+        )
+        mock_responses.add(
+            responses.GET, "https://iaso-staging.bluesquare.org/api/orgunits/", json=iaso_mocked_orgunits_with_params, status=200
+        )
+        iaso = IASO("https://iaso-staging.bluesquare.org", "username", "password")
+        r = iaso.get_org_units()
+        assert len(r) > 0
     def test_get_forms(self, mock_responses):
         mock_responses.add(
             responses.POST, "https://iaso-staging.bluesquare.org/api/token/", json=iaso_mocked_auth_token, status=200