style: apply ruff formatting to test files

Author: ahoblitz

tests/server/test_models.py

@@ -22,8 +22,8 @@ def test_process_bind_param_with_pydantic_model(self):
         dialect = MagicMock()
 
         result = pydantic_type.process_bind_param(status, dialect)
-        assert result["state"] == "working"
-        assert result["message"] is None
+        assert result['state'] == 'working'
+        assert result['message'] is None
         # TaskStatus may have other optional fields
 
     def test_process_bind_param_with_none(self):
@@ -37,9 +37,11 @@ def test_process_result_value(self):
         pydantic_type = PydanticType(TaskStatus)
         dialect = MagicMock()
 
-        result = pydantic_type.process_result_value({"state": "completed", "message": None}, dialect)
+        result = pydantic_type.process_result_value(
+            {'state': 'completed', 'message': None}, dialect
+        )
         assert isinstance(result, TaskStatus)
-        assert result.state == "completed"
+        assert result.state == 'completed'
 
 
 class TestPydanticListType:
@@ -48,33 +50,38 @@ class TestPydanticListType:
     def test_process_bind_param_with_list(self):
         pydantic_list_type = PydanticListType(Artifact)
         artifacts = [
-            Artifact(artifact_id="1", parts=[TextPart(type="text", text="Hello")]),
-            Artifact(artifact_id="2", parts=[TextPart(type="text", text="World")])
+            Artifact(
+                artifact_id='1', parts=[TextPart(type='text', text='Hello')]
+            ),
+            Artifact(
+                artifact_id='2', parts=[TextPart(type='text', text='World')]
+            ),
         ]
         dialect = MagicMock()
 
         result = pydantic_list_type.process_bind_param(artifacts, dialect)
         assert len(result) == 2
-        assert result[0]["artifactId"] == "1"  # JSON mode uses camelCase
-        assert result[1]["artifactId"] == "2"
+        assert result[0]['artifactId'] == '1'  # JSON mode uses camelCase
+        assert result[1]['artifactId'] == '2'
 
     def test_process_result_value_with_list(self):
         pydantic_list_type = PydanticListType(Artifact)
         dialect = MagicMock()
         data = [
-            {"artifact_id": "1", "parts": [{"type": "text", "text": "Hello"}]},
-            {"artifact_id": "2", "parts": [{"type": "text", "text": "World"}]}
+            {'artifact_id': '1', 'parts': [{'type': 'text', 'text': 'Hello'}]},
+            {'artifact_id': '2', 'parts': [{'type': 'text', 'text': 'World'}]},
         ]
 
         result = pydantic_list_type.process_result_value(data, dialect)
         assert len(result) == 2
         assert all(isinstance(art, Artifact) for art in result)
-        assert result[0].artifact_id == "1"
-        assert result[1].artifact_id == "2"
+        assert result[0].artifact_id == '1'
+        assert result[1].artifact_id == '2'
 
 
 def test_create_task_model():
     """Test dynamic task model creation."""
+
     # Create a fresh base to avoid table conflicts
     class TestBase(DeclarativeBase):
         pass
@@ -92,15 +99,20 @@ class TestBase(DeclarativeBase):
 
 def test_create_push_notification_config_model():
     """Test dynamic push notification config model creation."""
+
     # Create a fresh base to avoid table conflicts
     class TestBase(DeclarativeBase):
         pass
 
     # Create with default table name
-    default_model = create_push_notification_config_model('test_push_configs_1', TestBase)
+    default_model = create_push_notification_config_model(
+        'test_push_configs_1', TestBase
+    )
     assert default_model.__tablename__ == 'test_push_configs_1'
 
     # Create with custom table name
-    custom_model = create_push_notification_config_model('test_push_configs_2', TestBase)
+    custom_model = create_push_notification_config_model(
+        'test_push_configs_2', TestBase
+    )
     assert custom_model.__tablename__ == 'test_push_configs_2'
     assert 'test_push_configs_2' in custom_model.__name__

tests/utils/test_constants.py

@@ -5,9 +5,15 @@
 
 def test_agent_card_constants():
     """Test that agent card constants have expected values."""
-    assert constants.AGENT_CARD_WELL_KNOWN_PATH == '/.well-known/agent-card.json'
-    assert constants.PREV_AGENT_CARD_WELL_KNOWN_PATH == '/.well-known/agent.json'
-    assert constants.EXTENDED_AGENT_CARD_PATH == '/agent/authenticatedExtendedCard'
+    assert (
+        constants.AGENT_CARD_WELL_KNOWN_PATH == '/.well-known/agent-card.json'
+    )
+    assert (
+        constants.PREV_AGENT_CARD_WELL_KNOWN_PATH == '/.well-known/agent.json'
+    )
+    assert (
+        constants.EXTENDED_AGENT_CARD_PATH == '/agent/authenticatedExtendedCard'
+    )
 
 
 def test_default_rpc_url():

tests/utils/test_error_handlers.py

@@ -27,7 +27,7 @@ def __init__(self, content, status_code):
 @pytest.mark.asyncio
 async def test_rest_error_handler_server_error():
     """Test rest_error_handler with ServerError."""
-    error = InvalidRequestError(message="Bad request")
+    error = InvalidRequestError(message='Bad request')
 
     @rest_error_handler
     async def failing_func():
@@ -44,9 +44,10 @@ async def failing_func():
 @pytest.mark.asyncio
 async def test_rest_error_handler_unknown_exception():
     """Test rest_error_handler with unknown exception."""
+
     @rest_error_handler
     async def failing_func():
-        raise ValueError("Unexpected error")
+        raise ValueError('Unexpected error')
 
     with patch('a2a.utils.error_handlers.JSONResponse', MockJSONResponse):
         result = await failing_func()
@@ -59,7 +60,7 @@ async def failing_func():
 @pytest.mark.asyncio
 async def test_rest_stream_error_handler_server_error():
     """Test rest_stream_error_handler with ServerError."""
-    error = InternalError(message="Internal server error")
+    error = InternalError(message='Internal server error')
 
     @rest_stream_error_handler
     async def failing_stream():
@@ -74,11 +75,12 @@ async def failing_stream():
 @pytest.mark.asyncio
 async def test_rest_stream_error_handler_reraises_exception():
     """Test rest_stream_error_handler reraises other exceptions."""
+
     @rest_stream_error_handler
     async def failing_stream():
-        raise RuntimeError("Stream failed")
+        raise RuntimeError('Stream failed')
 
-    with pytest.raises(RuntimeError, match="Stream failed"):
+    with pytest.raises(RuntimeError, match='Stream failed'):
         await failing_stream()