Updates

Author: ishymko

tests/client/transports/test_jsonrpc_client.py

@@ -442,7 +442,7 @@ async def test_send_message_streaming_sse_error(
         request = create_send_message_request()
         mock_event_source = AsyncMock()
         mock_event_source.response.raise_for_status = MagicMock()
-        mock_event_source.response.headers = {}
+        mock_event_source.response.headers = {'content-type': 'text/event-stream'}
         mock_event_source.aiter_sse = MagicMock(
             side_effect=SSEError('Simulated SSE error')
         )
@@ -464,7 +464,7 @@ async def test_send_message_streaming_request_error(
         request = create_send_message_request()
         mock_event_source = AsyncMock()
         mock_event_source.response.raise_for_status = MagicMock()
-        mock_event_source.response.headers = {}
+        mock_event_source.response.headers = {'content-type': 'text/event-stream'}
         mock_event_source.aiter_sse = MagicMock(
             side_effect=httpx.RequestError(
                 'Simulated request error', request=MagicMock()
@@ -488,7 +488,7 @@ async def test_send_message_streaming_timeout(
         request = create_send_message_request()
         mock_event_source = AsyncMock()
         mock_event_source.response.raise_for_status = MagicMock()
-        mock_event_source.response.headers = {}
+        mock_event_source.response.headers = {'content-type': 'text/event-stream'}
         mock_event_source.aiter_sse = MagicMock(
             side_effect=httpx.TimeoutException('Timeout')
         )

tests/client/transports/test_rest_client.py

@@ -87,6 +87,7 @@ async def test_send_message_streaming_timeout(
         )
         mock_event_source = AsyncMock(spec=EventSource)
         mock_event_source.response = MagicMock(spec=httpx.Response)
+        mock_event_source.response.headers = {'content-type': 'text/event-stream'}
         mock_event_source.response.raise_for_status.return_value = None
         mock_event_source.aiter_sse.side_effect = httpx.TimeoutException(
             'Read timed out'
@@ -295,6 +296,8 @@ async def test_send_message_streaming_with_new_extensions(
         )
 
         mock_event_source = AsyncMock(spec=EventSource)
+        mock_event_source.response = MagicMock(spec=httpx.Response)
+        mock_event_source.response.headers = {'content-type': 'text/event-stream'}
         mock_event_source.aiter_sse.return_value = async_iterable_from_list([])
         mock_aconnect_sse.return_value.__aenter__.return_value = (
             mock_event_source
@@ -708,6 +711,7 @@ async def test_rest_streaming_methods_prepend_tenant(  # noqa: PLR0913
         # 2. Setup mocks
         mock_event_source = AsyncMock(spec=EventSource)
         mock_event_source.response = MagicMock(spec=httpx.Response)
+        mock_event_source.response.headers = {'content-type': 'text/event-stream'}
         mock_event_source.response.raise_for_status.return_value = None
 
         async def empty_aiter():