chore(validate): add test for aoi test

Author: frozenhelium

apps/project/tests/e2e_create_validate_project_test.py

@@ -294,6 +294,32 @@ class Mutation:
             }
         """
 
+    class Query:
+        TEST_AOI_OBJECTS = """
+        query TestAoiObjects($assetId: ID, $projectId: ID, $ohsomeFilter: String) {
+            testAoiObjects(assetId: $assetId, projectId: $projectId, ohsomeFilter: $ohsomeFilter) {
+                ok
+                error
+                objectCount
+                assetId
+                projectId
+                ohsomeFilter
+            }
+        }
+        """
+
+        TEST_TASKING_MANAGER_PROJECT = """
+        query TestTaskingManagerProject($hotTmId: String, $ohsomeFilter: String) {
+            testTaskingManagerProject(hotTmId: $hotTmId, ohsomeFilter: $ohsomeFilter) {
+                ok
+                error
+                objectCount
+                hotTmId
+                ohsomeFilter
+            }
+        }
+        """
+
     @pytest.mark.vcr("assets/tests/projects/validate/cassette")
     def test_validate_project_e2e(self):
         # TODO(susilnem): Add more test with filters
@@ -398,6 +424,15 @@ def _test_project(self, filename: str):
         assert aoi_response["ok"]
         aoi_id = aoi_response["result"]["id"]
 
+        # Test AOI objects
+        ohsomeFilter = "building=* and geometry:polygon"
+        test_aoi_objects_content = self.query_check(
+            self.Query.TEST_AOI_OBJECTS,
+            variables={"assetId": aoi_id, "projectId": project_id, "ohsomeFilter": ohsomeFilter},
+        )
+        test_aoi_objects_response = test_aoi_objects_content["data"]["testAoiObjects"]
+        assert test_aoi_objects_response["ok"]
+
         # Update project
         update_project_data = test_data["update_project"]
         update_project_data["image"] = image_id

project_types/validate/tests/api_calls_test.py

@@ -7,6 +7,7 @@
 from main.tests import TestCase
 from project_types.validate.api_calls import (
     ValidateApiCallError,
+    get_object_count_from_ohsome,
     ohsome,
     query_osm,
     query_osmcha,
@@ -141,6 +142,37 @@ def test_query_osm(self, mock_retry_get):  # type: ignore[reportMissingParameter
         with pytest.raises(ValidateApiCallError):
             query_osm([12345], {})
 
+    @patch("requests.post")
+    def test_get_object_count_from_ohsome(self, mock_post):  # type: ignore[reportMissingParameterType]
+        sample_filter = "building=* and geometry:polygon"
+        sample_area = "POLYGON((8.67 49.39,8.68 49.39,8.68 49.40,8.67 49.40,8.67 49.39))"
+
+        sample_object_count = 500
+
+        mock_response_data = {
+            "attribution": {
+                "url": "https://ohsome.org/copyrights",
+                "text": "© OpenStreetMap contributors",
+            },
+            "apiVersion": "1.10.4",
+            "result": [
+                {
+                    "timestamp": "2025-10-01T12:00:00Z",
+                    "value": sample_object_count,
+                },
+            ],
+        }
+
+        mock_response = MagicMock()
+        mock_response.status_code = 200
+        mock_response.json.return_value = mock_response_data
+        mock_post.return_value = mock_response
+
+        object_count = get_object_count_from_ohsome(sample_area, sample_filter)
+
+        mock_post.assert_called_once()
+        assert object_count == sample_object_count
+
     @patch("requests.post")
     @patch("project_types.validate.api_calls.remove_noise_and_add_user_info")
     def test_ohsome(self, mock_remove_noise, mock_post):  # type: ignore[reportMissingParameterType]