add get response status codes

Author: shemogumbe

src/msgraph_core/requests/batch_response_content.py

@@ -64,7 +64,7 @@ def get_response_stream_by_id(self, request_id: str) -> Optional[BytesIO]:
         :param request_id: The request ID of the response to get
         :type request_id: str
         :return: The response Body as a stream
-        :rtype: io.BytesIO
+        :rtype: BytesIO
         """
         response_item = self.get_response_by_id(request_id)
         if response_item is None or response_item.body is None:
@@ -74,6 +74,22 @@ def get_response_stream_by_id(self, request_id: str) -> Optional[BytesIO]:
             return response_item.body
         return BytesIO(response_item.body)
 
+    def get_response_status_codes(self) -> Dict[str, int]:
+        """
+        Go through responses and for each, append {'request-id': status_code} to a dictionary.
+        :return: A dictionary with request_id as keys and status_code as values.
+        :rtype: dict
+        """
+        status_codes: Dict[str, int] = {}
+        if self._responses is None:
+            return status_codes
+
+        for request_id, response_item in self._responses.items():
+            if response_item is not None and response_item.status is not None:
+                status_codes[request_id] = response_item.status
+
+        return status_codes
+
     def response_body(self, request_id: str, type: Type[T]) -> Optional[T]:
         """ 
         Get the body of a response by its request ID from the collection

tests/requests/test_batch_response_content.py

@@ -29,6 +29,55 @@ def test_response_method(batch_response_content):
     assert batch_response_content.get_response_by_id("12345") == response_item
 
 
+def test_get_response_stream_by_id_none(batch_response_content):
+    batch_response_content.get_response_by_id = Mock(return_value=None)
+    result = batch_response_content.get_response_stream_by_id('1')
+    assert result is None
+
+
+def test_get_response_stream_by_id_body_none(batch_response_content):
+    batch_response_content.get_response_by_id = Mock(return_value=Mock(body=None))
+    result = batch_response_content.get_response_stream_by_id('1')
+    assert result is None
+
+
+def test_get_response_stream_by_id_bytesio(batch_response_content):
+    batch_response_content.get_response_by_id = Mock(
+        return_value=Mock(body=BytesIO(b'Hello, world!'))
+    )
+    result = batch_response_content.get_response_stream_by_id('2')
+    assert isinstance(result, BytesIO)
+    assert result.read() == b'Hello, world!'
+
+
+def test_get_response_stream_by_id_bytes(batch_response_content):
+    batch_response_content.get_response_by_id = Mock(return_value=Mock(body=b'Hello, world!'))
+    result = batch_response_content.get_response_stream_by_id('1')
+    assert isinstance(result, BytesIO)
+    assert result.read() == b'Hello, world!'
+
+
+def test_get_response_status_codes_none(batch_response_content):
+    batch_response_content._responses = None
+    result = batch_response_content.get_response_status_codes()
+    assert result == {}
+
+
+def test_get_response_status_codes(batch_response_content):
+    batch_response_content._responses = {
+        '1': Mock(status=200),
+        '2': Mock(status=404),
+        '3': Mock(status=500),
+    }
+    result = batch_response_content.get_response_status_codes()
+    expected = {
+        '1': 200,
+        '2': 404,
+        '3': 500,
+    }
+    assert result == expected
+
+
 def test_response_body_method(batch_response_content):
     response_item = Mock(spec=BatchResponseItem)
     response_item.request_id = "12345"