Formatting

Author: ishymko

tests/client/transports/test_jsonrpc_client.py

@@ -442,7 +442,9 @@ 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 = {'content-type': 'text/event-stream'}
+        mock_event_source.response.headers = {
+            'content-type': 'text/event-stream'
+        }
         mock_event_source.aiter_sse = MagicMock(
             side_effect=SSEError('Simulated SSE error')
         )
@@ -464,7 +466,9 @@ 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 = {'content-type': 'text/event-stream'}
+        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 +492,9 @@ 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 = {'content-type': 'text/event-stream'}
+        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,7 +87,9 @@ 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.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'
@@ -297,7 +299,9 @@ 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.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
@@ -711,7 +715,9 @@ 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.headers = {
+            'content-type': 'text/event-stream'
+        }
         mock_event_source.response.raise_for_status.return_value = None
 
         async def empty_aiter():