Improve coverage

Author: rajeshvelicheti

src/a2a/server/apps/rest/rest_adapter.py

@@ -149,11 +149,9 @@ async def handle_get_agent_card(
         if self.card_modifier:
             card_to_serve = self.card_modifier(card_to_serve)
 
-        return JSONResponse(
-            card_to_serve.model_dump(
-                exclude_none=True,
-                by_alias=True,
-            )
+        return card_to_serve.model_dump(
+            exclude_none=True,
+            by_alias=True,
         )
 
     @rest_error_handler
@@ -187,11 +185,9 @@ async def handle_authenticated_agent_card(
             card_to_serve = self.extended_card_modifier(base_card, context)
 
         if card_to_serve:
-            return JSONResponse(
-                card_to_serve.model_dump(
-                    exclude_none=True,
-                    by_alias=True,
-                )
+            return card_to_serve.model_dump(
+                exclude_none=True,
+                by_alias=True,
             )
         # If supports_authenticated_extended_card is true, but no
         # extended_agent_card was provided, and no modifier produced a card,
@@ -253,7 +249,7 @@ def routes(self) -> dict[tuple[str, str], Callable[[Request], Any]]:
             ('/v1/tasks', 'GET'): functools.partial(
                 self._handle_request, self.handler.list_tasks
             ),
-            ('v1/well_known/agent_json', 'GET'): functools.partial(
+            ('/v1/well_known/agent_json', 'GET'): functools.partial(
                 self._handle_request, self.handle_get_agent_card
             ),
         }

tests/integration/test_client_server_integration.py

@@ -717,6 +717,33 @@ async def test_http_transport_get_card(
         await transport.close()
 
 
+@pytest.mark.asyncio
+async def test_http_transport_get_authenticated_card(
+    agent_card: AgentCard,
+    mock_request_handler: AsyncMock,
+) -> None:
+    agent_card.supports_authenticated_extended_card = True
+    extended_agent_card = agent_card.model_copy(deep=True)
+    extended_agent_card.name = 'Extended Agent Card'
+
+    app_builder = A2ARESTFastAPIApplication(
+        agent_card,
+        mock_request_handler,
+        extended_agent_card=extended_agent_card,
+    )
+    app = app_builder.build()
+    httpx_client = httpx.AsyncClient(transport=httpx.ASGITransport(app=app))
+
+    transport = RestTransport(httpx_client=httpx_client, agent_card=agent_card)
+    result = await transport.get_card()
+    assert result.name == extended_agent_card.name
+    assert transport.agent_card.name == extended_agent_card.name
+    assert transport._needs_extended_card is False
+
+    if hasattr(transport, 'close'):
+        await transport.close()
+
+
 @pytest.mark.asyncio
 async def test_grpc_transport_get_card(
     grpc_server_and_handler: tuple[str, AsyncMock],