Expose error mappings in page iterator and ignore errors due to undefined behaviour

Author: Ndiritu

src/msgraph_core/models/page_result.py

@@ -47,7 +47,7 @@ def get_field_deserializers(self) -> Dict[str, Callable[[ParseNode], None]]:
         """
         return {
             "@odata.nextLink": lambda x: setattr(self, "odata_next_link", x.get_str_value()),
-            "value": lambda x: setattr(self, "value", x.get_collection_of_object_values(Parsable))
+            "value": lambda x: setattr(self, "value", x.get_collection_of_object_values(Parsable)) # type: ignore # Bug. Should get a collection of primitive dictionary objects
         }
 
     def serialize(self, writer: SerializationWriter) -> None:

src/msgraph_core/tasks/page_iterator.py

@@ -26,7 +26,7 @@
 from kiota_abstractions.method import Method
 from kiota_abstractions.headers_collection import HeadersCollection
 from kiota_abstractions.request_information import RequestInformation
-from kiota_abstractions.serialization.parsable import Parsable
+from kiota_abstractions.serialization import Parsable, ParsableFactory
 
 from msgraph_core.models.page_result import PageResult  # pylint: disable=no-name-in-module, import-error
 
@@ -59,12 +59,13 @@ def __init__(
         self,
         response: Union[T, list, object],
         request_adapter: RequestAdapter,
-        constructor_callable: Optional[Callable] = None
+        constructor_callable: Optional[Callable] = None,
+        error_mapping: Optional[Dict[str, type[ParsableFactory]]] = None,
     ):
         self.request_adapter = request_adapter
 
         if isinstance(response, Parsable) and not constructor_callable:
-            parsable_factory = type(response)
+            parsable_factory = type(response) # type: ignore
         elif constructor_callable is None:
             parsable_factory = PageResult
         else:
@@ -89,6 +90,7 @@ def __init__(
         if page is not None:
             self.current_page = page
             self.has_next = bool(page.odata_next_link)
+        self.error_mapping = error_mapping if error_mapping else dict()
 
     def set_headers(self, headers: dict) -> HeadersCollection:
         """
@@ -100,6 +102,7 @@ def set_headers(self, headers: dict) -> HeadersCollection:
             header names and the values are the header values.
         """
         self.headers.add_all(**headers)
+        return self.headers
 
     @property
     def delta_link(self):
@@ -181,7 +184,7 @@ def convert_to_page(response: Union[T, list, object]) -> PageResult:
         page: PageResult = PageResult(next_link, value)
         return page
 
-    async def fetch_next_page(self) -> List[Parsable]:
+    async def fetch_next_page(self) -> Optional[Union[T, PageResult]]:
         """
         Fetches the next page of items from the server.
         Returns:
@@ -202,9 +205,8 @@ async def fetch_next_page(self) -> List[Parsable]:
         request_info.headers = self.headers
         if self.request_options:
             request_info.add_request_options(*self.request_options)
-        error_map: Dict[str, int] = {}
         response = await self.request_adapter.send_async(
-            request_info, self.parsable_factory, error_map
+            request_info, self.parsable_factory, self.error_mapping # type: ignore
         )
         return response