Skip to content

Commit a52cf2f

Browse files
committed
Formatting/linting
1 parent 58b0bee commit a52cf2f

3 files changed

Lines changed: 12 additions & 6 deletions

File tree

src/a2a/server/apps/jsonrpc/jsonrpc_app.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -179,7 +179,7 @@ def _generate_error_response(
179179
status_code=200,
180180
)
181181

182-
async def _handle_requests(self, request: Request) -> Response:
182+
async def _handle_requests(self, request: Request) -> Response: # noqa: PLR0911
183183
"""Handles incoming POST requests to the main A2A endpoint.
184184
185185
Parses the request body as JSON, validates it against A2A request types,

src/a2a/server/events/event_consumer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -141,7 +141,7 @@ async def consume_all(self) -> AsyncGenerator[Event]:
141141
if self.queue.is_closed():
142142
break
143143
except ValidationError as e:
144-
logger.error(f"Invalid event format received: {e}")
144+
logger.error(f'Invalid event format received: {e}')
145145
continue
146146
except Exception as e:
147147
logger.error(

tests/server/events/test_event_consumer.py

Lines changed: 10 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,10 @@
11
import asyncio
2+
23
from typing import Any
34
from unittest.mock import AsyncMock, MagicMock, patch
45

56
import pytest
7+
68
from pydantic import ValidationError
79

810
from a2a.server.events.event_consumer import EventConsumer, QueueClosed
@@ -352,15 +354,19 @@ async def test_consume_all_handles_validation_error(
352354
"""Test that consume_all gracefully handles a pydantic.ValidationError."""
353355
# Simulate dequeue_event raising a ValidationError
354356
mock_event_queue.dequeue_event.side_effect = [
355-
ValidationError.from_exception_data(title="Test Error", line_errors=[]),
356-
asyncio.CancelledError # To stop the loop for the test
357+
ValidationError.from_exception_data(title='Test Error', line_errors=[]),
358+
asyncio.CancelledError, # To stop the loop for the test
357359
]
358360

359-
with patch("a2a.server.events.event_consumer.logger.error") as logger_error_mock:
361+
with patch(
362+
'a2a.server.events.event_consumer.logger.error'
363+
) as logger_error_mock:
360364
with pytest.raises(asyncio.CancelledError):
361365
async for _ in event_consumer.consume_all():
362366
pass
363367

364368
# Check that the specific error was logged and the consumer continued
365369
logger_error_mock.assert_called_once()
366-
assert "Invalid event format received" in logger_error_mock.call_args[0][0]
370+
assert (
371+
'Invalid event format received' in logger_error_mock.call_args[0][0]
372+
)

0 commit comments

Comments
 (0)