diff --git a/src/msfocr/data/data_upload_DHIS2.py b/src/msfocr/data/data_upload_DHIS2.py index d4d2073..2707150 100644 --- 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 \ No newline at end of file + return uid \ No newline at end of file diff --git a/tests/test_data_upload_DHIS2.py b/tests/test_data_upload_DHIS2.py new file mode 100644 index 0000000..613a5d7 --- /dev/null +++ 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 +