Added test for getAllUIDs

UMassCDS · Jul 9, 2024 · d11278e · d11278e
1 parent a0c563f
commit d11278e
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/src/msfocr/data/data_upload_DHIS2.py b/src/msfocr/data/data_upload_DHIS2.py
@@ -40,6 +40,6 @@ def getAllUIDs(item_type, search_items):
     if len(items) > 0:
         print(items[0])
 
-    id = [(item['displayName'], item['id']) for item in items]
+    uid = [(item['displayName'], item['id']) for item in items]
 
-    return id
+    return uid
diff --git a/tests/test_data_upload_DHIS2.py b/tests/test_data_upload_DHIS2.py
@@ -0,0 +1,9 @@
+from msfocr.data.data_upload_DHIS2 import getAllUIDs
+
+def test_getAllUIDs(test_server_config, requests_mock):
+    requests_mock.get("http://test.com/api/categoryOptions?filter=name:ilike:12-59m", json={'categoryOptions': [{'id': 'tWRttYIzvBn', 'displayName': '12-59m'}]})
+    expected_result = [('12-59m', 'tWRttYIzvBn')]
+    result = getAllUIDs("categoryOptions", ["12-59m"])
+
+    assert expected_result == result
+