Formatting

Author: holtskinner

.ruff.toml

@@ -136,7 +136,7 @@ inline-quotes = "single"
     "PLR2004",
     "SLF001",
 ]
-"types.py" = ["D", "E501", "N815"]  # Ignore docstring and annotation issues in types.py
+"types.py" = ["D", "E501"]  # Ignore docstring and annotation issues in types.py
 "proto_utils.py" = ["D102", "PLR0911"]
 "helpers.py" = ["ANN001", "ANN201", "ANN202"]
 

tests/server/apps/jsonrpc/test_serialization.py

@@ -32,7 +32,7 @@ def agent_card_with_api_key():
     }
     api_key_scheme = APIKeySecurityScheme.model_validate(api_key_scheme_data)
 
-    agent_card = AgentCard(
+    return AgentCard(
         name='APIKeyAgent',
         description='An agent that uses API Key auth.',
         url='http://example.com/apikey-agent',
@@ -44,7 +44,6 @@ def agent_card_with_api_key():
         security_schemes={'api_key_auth': SecurityScheme(root=api_key_scheme)},
         security=[{'api_key_auth': []}],
     )
-    return agent_card
 
 
 def test_starlette_agent_card_with_api_key_scheme_alias(
@@ -145,7 +144,7 @@ def test_handle_oversized_payload(agent_card_with_api_key: AgentCard):
             assert response.status_code == 413
     except Exception as e:
         # Depending on server setup, it might just drop the connection for very large payloads
-        assert isinstance(e, (ConnectionResetError, RuntimeError))
+        assert isinstance(e, ConnectionResetError | RuntimeError)
 
 
 def test_handle_unicode_characters(agent_card_with_api_key: AgentCard):

tests/server/request_handlers/test_default_request_handler.py

@@ -1208,7 +1208,7 @@ async def test_get_task_push_notification_config_info_with_config_no_id():
 
     set_config_params = TaskPushNotificationConfig(
         task_id='task_1',
-        pushNotificationConfig=PushNotificationConfig(
+        push_notification_config=PushNotificationConfig(
             url='http://1.example.com'
         ),
     )
@@ -1447,7 +1447,7 @@ async def test_list_task_push_notification_config_info_with_config_and_no_id():
     # multiple calls without config id should replace the existing
     set_config_params1 = TaskPushNotificationConfig(
         task_id='task_1',
-        pushNotificationConfig=PushNotificationConfig(
+        push_notification_config=PushNotificationConfig(
             url='http://1.example.com'
         ),
     )
@@ -1457,7 +1457,7 @@ async def test_list_task_push_notification_config_info_with_config_and_no_id():
 
     set_config_params2 = TaskPushNotificationConfig(
         task_id='task_1',
-        pushNotificationConfig=PushNotificationConfig(
+        push_notification_config=PushNotificationConfig(
             url='http://2.example.com'
         ),
     )
@@ -1555,7 +1555,7 @@ async def test_delete_no_task_push_notification_config_info():
     result = await request_handler.on_delete_task_push_notification_config(
         params, create_server_call_context()
     )
-    assert result == None
+    assert result is None
 
     params = DeleteTaskPushNotificationConfigParams(
         id='task2', push_notification_config_id='config_non_existant'
@@ -1564,7 +1564,7 @@ async def test_delete_no_task_push_notification_config_info():
     result = await request_handler.on_delete_task_push_notification_config(
         params, create_server_call_context()
     )
-    assert result == None
+    assert result is None
 
 
 @pytest.mark.asyncio
@@ -1600,7 +1600,7 @@ async def test_delete_task_push_notification_config_info_with_config():
         params, create_server_call_context()
     )
 
-    assert result1 == None
+    assert result1 is None
 
     result2 = await request_handler.on_list_task_push_notification_config(
         ListTaskPushNotificationConfigParams(id='task_1'),
@@ -1640,7 +1640,7 @@ async def test_delete_task_push_notification_config_info_with_config_and_no_id()
         params, create_server_call_context()
     )
 
-    assert result == None
+    assert result is None
 
     result2 = await request_handler.on_list_task_push_notification_config(
         ListTaskPushNotificationConfigParams(id='task_1'),

tests/server/request_handlers/test_jsonrpc_handler.py

@@ -457,7 +457,7 @@ async def test_set_push_notification_success(self) -> None:
         mock_task_store.get.return_value = mock_task
         task_push_config = TaskPushNotificationConfig(
             task_id=mock_task.id,
-            pushNotificationConfig=PushNotificationConfig(
+            push_notification_config=PushNotificationConfig(
                 url='http://example.com'
             ),
         )
@@ -492,7 +492,7 @@ async def test_get_push_notification_success(self) -> None:
         mock_task_store.get.return_value = mock_task
         task_push_config = TaskPushNotificationConfig(
             task_id=mock_task.id,
-            pushNotificationConfig=PushNotificationConfig(
+            push_notification_config=PushNotificationConfig(
                 url='http://example.com'
             ),
         )
@@ -579,7 +579,7 @@ async def streaming_coro():
             )
             request.params.configuration = MessageSendConfiguration(
                 accepted_output_modes=['text'],
-                pushNotificationConfig=PushNotificationConfig(
+                push_notification_config=PushNotificationConfig(
                     url='http://example.com'
                 ),
             )
@@ -761,7 +761,7 @@ async def test_push_notifications_not_supported_error(self) -> None:
         # Act & Assert
         task_push_config = TaskPushNotificationConfig(
             task_id='task_123',
-            pushNotificationConfig=PushNotificationConfig(
+            push_notification_config=PushNotificationConfig(
                 url='http://example.com'
             ),
         )
@@ -825,7 +825,7 @@ async def test_on_set_push_notification_no_push_config_store(self) -> None:
         # Act
         task_push_config = TaskPushNotificationConfig(
             task_id=mock_task.id,
-            pushNotificationConfig=PushNotificationConfig(
+            push_notification_config=PushNotificationConfig(
                 url='http://example.com'
             ),
         )
@@ -1049,7 +1049,7 @@ async def test_on_get_push_notification(self) -> None:
         request_handler = AsyncMock(spec=DefaultRequestHandler)
         task_push_config = TaskPushNotificationConfig(
             task_id=mock_task.id,
-            pushNotificationConfig=PushNotificationConfig(
+            push_notification_config=PushNotificationConfig(
                 id='config1', url='http://example.com'
             ),
         )
@@ -1085,7 +1085,7 @@ async def test_on_list_push_notification(self) -> None:
         request_handler = AsyncMock(spec=DefaultRequestHandler)
         task_push_config = TaskPushNotificationConfig(
             task_id=mock_task.id,
-            pushNotificationConfig=PushNotificationConfig(
+            push_notification_config=PushNotificationConfig(
                 url='http://example.com'
             ),
         )
@@ -1116,9 +1116,9 @@ async def test_on_list_push_notification_error(self) -> None:
 
         # Create request handler without a push notifier
         request_handler = AsyncMock(spec=DefaultRequestHandler)
-        task_push_config = TaskPushNotificationConfig(
+        _ = TaskPushNotificationConfig(
             task_id=mock_task.id,
-            pushNotificationConfig=PushNotificationConfig(
+            push_notification_config=PushNotificationConfig(
                 url='http://example.com'
             ),
         )

tests/server/tasks/test_database_push_notification_config_store.py

@@ -355,7 +355,7 @@ async def test_decryption_error_with_wrong_key(
             PushNotificationConfigModel, (task_id, 'config-1')
         )
 
-        with pytest.raises(ValueError) as exc_info:
+        with pytest.raises(ValueError):
             store2._from_orm(db_model)  # type: ignore
 
 
@@ -386,7 +386,7 @@ async def test_decryption_error_with_no_key(
             PushNotificationConfigModel, (task_id, 'config-1')
         )
 
-        with pytest.raises(ValueError) as exc_info:
+        with pytest.raises(ValueError):
             store2._from_orm(db_model)  # type: ignore
 
 
@@ -560,5 +560,5 @@ async def test_parsing_error_after_successful_decryption(
             PushNotificationConfigModel, (task_id, config_id)
         )
 
-        with pytest.raises(ValueError) as exc_info:
+        with pytest.raises(ValueError):
             db_store_parameterized._from_orm(db_model_retrieved)  # type: ignore

tests/server/tasks/test_task_updater.py

@@ -147,8 +147,8 @@ async def test_add_artifact_generates_id(
     assert isinstance(event, TaskArtifactUpdateEvent)
     assert event.artifact.artifact_id == str(known_uuid)
     assert event.artifact.parts == sample_parts
-    assert event.append == None
-    assert event.last_chunk == None
+    assert event.append is None
+    assert event.last_chunk is None
 
 
 @pytest.mark.asyncio

tests/server/test_integration.py

@@ -457,7 +457,7 @@ def test_set_push_notification_config(
     # Setup mock response
     task_push_config = TaskPushNotificationConfig(
         task_id='t2',
-        pushNotificationConfig=PushNotificationConfig(
+        push_notification_config=PushNotificationConfig(
             url='https://example.com', token='secret-token'
         ),
     )

tests/test_types.py

@@ -741,7 +741,7 @@ def test_send_message_streaming_artifact_update_response() -> None:
 def test_set_task_push_notification_response() -> None:
     task_push_config = TaskPushNotificationConfig(
         task_id='t2',
-        pushNotificationConfig=PushNotificationConfig(
+        push_notification_config=PushNotificationConfig(
             url='https://example.com', token='token'
         ),
     )
@@ -802,7 +802,7 @@ def test_set_task_push_notification_response() -> None:
 def test_get_task_push_notification_response() -> None:
     task_push_config = TaskPushNotificationConfig(
         task_id='t2',
-        pushNotificationConfig=PushNotificationConfig(
+        push_notification_config=PushNotificationConfig(
             url='https://example.com', token='token'
         ),
     )
@@ -914,7 +914,7 @@ def test_a2a_request_root_model() -> None:
     # SetTaskPushNotificationConfigRequest
     task_push_config = TaskPushNotificationConfig(
         task_id='t2',
-        pushNotificationConfig=PushNotificationConfig(
+        push_notification_config=PushNotificationConfig(
             url='https://example.com', token='token'
         ),
     )
@@ -1022,7 +1022,7 @@ def test_a2a_request_root_model_id_validation() -> None:
     # SetTaskPushNotificationConfigRequest
     task_push_config = TaskPushNotificationConfig(
         task_id='t2',
-        pushNotificationConfig=PushNotificationConfig(
+        push_notification_config=PushNotificationConfig(
             url='https://example.com', token='token'
         ),
     )
@@ -1304,7 +1304,7 @@ def test_task_push_notification_config() -> None:
     assert push_notification_config.authentication == auth_info
 
     task_push_notification_config = TaskPushNotificationConfig(
-        task_id='task-123', pushNotificationConfig=push_notification_config
+        task_id='task-123', push_notification_config=push_notification_config
     )
     assert task_push_notification_config.task_id == 'task-123'
     assert (