Formatting

Author: holtskinner

tests/server/apps/jsonrpc/test_fastapi_app.py

@@ -39,9 +39,9 @@ def mock_app_params(self) -> dict:
         # Ensure 'url' attribute exists on the mock_agent_card, as it's accessed
         # in __init__
         mock_agent_card.url = 'http://example.com'
-        # Ensure 'supportsAuthenticatedExtendedCard' attribute exists
-        mock_agent_card.supportsAuthenticatedExtendedCard = False
-        return dict(agent_card=mock_agent_card, http_handler=mock_handler)
+        # Ensure 'supports_authenticated_agent_card' attribute exists
+        mock_agent_card.supports_authenticated_agent_card = False
+        return {'agent_card': mock_agent_card, 'http_handler': mock_handler}
 
     @pytest.fixture(scope='class')
     def mark_pkg_fastapi_not_installed(self):

tests/server/apps/jsonrpc/test_jsonrpc_app.py

@@ -13,7 +13,9 @@
     StarletteBaseUser = MagicMock()  # type: ignore
 
 from a2a.extensions.common import HTTP_EXTENSION_HEADER
-from a2a.server.apps.jsonrpc import jsonrpc_app  # Keep this import for optional deps test
+from a2a.server.apps.jsonrpc import (
+    jsonrpc_app,  # Keep this import for optional deps test
+)
 from a2a.server.apps.jsonrpc.jsonrpc_app import (
     JSONRPCApplication,
     StarletteUserProxy,
@@ -131,7 +133,7 @@ def mock_app_params(self) -> dict:
         mock_agent_card.url = 'http://example.com'
         # Ensure 'supportsAuthenticatedExtendedCard' attribute exists
         mock_agent_card.supports_authenticated_extended_card = False
-        return dict(agent_card=mock_agent_card, http_handler=mock_handler)
+        return {'agent_card': mock_agent_card, 'http_handler': mock_handler}
 
     @pytest.fixture(scope='class')
     def mark_pkg_starlette_not_installed(self):
@@ -355,4 +357,4 @@ def side_effect(request, context: ServerCallContext):
 
 
 if __name__ == '__main__':
-    pytest.main([__file__])
+    pytest.main([__file__])

tests/server/apps/jsonrpc/test_starlette_app.py

@@ -21,8 +21,8 @@ class TestA2AStarletteApplicationOptionalDeps:
     @pytest.fixture(scope='class', autouse=True)
     def ensure_pkg_starlette_is_present(self):
         try:
-            import starlette as _starlette  # noqa: F401
             import sse_starlette as _sse_starlette  # noqa: F401
+            import starlette as _starlette  # noqa: F401
         except ImportError:
             pytest.fail(
                 f'Running tests in {self.__class__.__name__} requires'
@@ -42,7 +42,7 @@ def mock_app_params(self) -> dict:
         mock_agent_card.url = 'http://example.com'
         # Ensure 'supportsAuthenticatedExtendedCard' attribute exists
         mock_agent_card.supportsAuthenticatedExtendedCard = False
-        return dict(agent_card=mock_agent_card, http_handler=mock_handler)
+        return {'agent_card': mock_agent_card, 'http_handler': mock_handler}
 
     @pytest.fixture(scope='class')
     def mark_pkg_starlette_not_installed(self):