diff --git a/src/opengeodeweb_back/geode_functions.py b/src/opengeodeweb_back/geode_functions.py index 3fd4cd05..763a4a7e 100644 --- a/src/opengeodeweb_back/geode_functions.py +++ b/src/opengeodeweb_back/geode_functions.py @@ -188,7 +188,11 @@ def list_geode_objects( for geode_object in geode_objects_filtered_list: if has_creator(geode_object, file_extension): loadability_score = is_loadable(geode_object, file_absolute_path) - return_dict[geode_object] = {"is_loadable": loadability_score} + priority_score = object_priority(geode_object, file_absolute_path) + return_dict[geode_object] = { + "is_loadable": loadability_score, + "object_priority": priority_score, + } return return_dict diff --git a/tests/test_geode_functions.py b/tests/test_geode_functions.py index 3cdaa5e0..870a130a 100644 --- a/tests/test_geode_functions.py +++ b/tests/test_geode_functions.py @@ -304,12 +304,15 @@ def test_list_geode_objects(): assert len(return_dict.keys()) >= 1 for input_geode_object, input_extension_dict in return_dict.items(): assert type(input_extension_dict) is dict - for ( - input_extension, - input_extension_value, - ) in input_extension_dict.items(): - assert isinstance(input_extension_value, float) - assert 0.0 <= input_extension_value <= 1.0 + if "is_loadable" in input_extension_dict: + is_loadable_value = input_extension_dict["is_loadable"] + assert isinstance(is_loadable_value, float) + assert 0.0 <= is_loadable_value <= 1.0 + assert "object_priority" in input_extension_dict + object_priority_value = input_extension_dict[ + "object_priority" + ] + assert isinstance(object_priority_value, int) def test_geode_objects_output_extensions():