test: add unittests for jsonrpc apps

Signed-off-by: Shingo OKAWA <[email protected]>

Author: ognis1205

src/a2a/types.py

@@ -1366,7 +1366,9 @@ class Artifact(BaseModel):
 
 
 class DeleteTaskPushNotificationConfigResponse(
-    RootModel[JSONRPCErrorResponse | DeleteTaskPushNotificationConfigSuccessResponse]
+    RootModel[
+        JSONRPCErrorResponse | DeleteTaskPushNotificationConfigSuccessResponse
+    ]
 ):
     root: JSONRPCErrorResponse | DeleteTaskPushNotificationConfigSuccessResponse
     """
@@ -1375,7 +1377,9 @@ class DeleteTaskPushNotificationConfigResponse(
 
 
 class GetTaskPushNotificationConfigResponse(
-    RootModel[JSONRPCErrorResponse | GetTaskPushNotificationConfigSuccessResponse]
+    RootModel[
+        JSONRPCErrorResponse | GetTaskPushNotificationConfigSuccessResponse
+    ]
 ):
     root: JSONRPCErrorResponse | GetTaskPushNotificationConfigSuccessResponse
     """
@@ -1384,7 +1388,9 @@ class GetTaskPushNotificationConfigResponse(
 
 
 class ListTaskPushNotificationConfigResponse(
-    RootModel[JSONRPCErrorResponse | ListTaskPushNotificationConfigSuccessResponse]
+    RootModel[
+        JSONRPCErrorResponse | ListTaskPushNotificationConfigSuccessResponse
+    ]
 ):
     root: JSONRPCErrorResponse | ListTaskPushNotificationConfigSuccessResponse
     """
@@ -1539,7 +1545,9 @@ class SendStreamingMessageRequest(BaseModel):
 
 
 class SetTaskPushNotificationConfigResponse(
-    RootModel[JSONRPCErrorResponse | SetTaskPushNotificationConfigSuccessResponse]
+    RootModel[
+        JSONRPCErrorResponse | SetTaskPushNotificationConfigSuccessResponse
+    ]
 ):
     root: JSONRPCErrorResponse | SetTaskPushNotificationConfigSuccessResponse
     """
@@ -1846,7 +1854,9 @@ class SendStreamingMessageSuccessResponse(BaseModel):
     """
 
 
-class CancelTaskResponse(RootModel[JSONRPCErrorResponse | CancelTaskSuccessResponse]):
+class CancelTaskResponse(
+    RootModel[JSONRPCErrorResponse | CancelTaskSuccessResponse]
+):
     root: JSONRPCErrorResponse | CancelTaskSuccessResponse
     """
     JSON-RPC response for the 'tasks/cancel' method.
@@ -1885,7 +1895,9 @@ class JSONRPCResponse(
     """
 
 
-class SendMessageResponse(RootModel[JSONRPCErrorResponse | SendMessageSuccessResponse]):
+class SendMessageResponse(
+    RootModel[JSONRPCErrorResponse | SendMessageSuccessResponse]
+):
     root: JSONRPCErrorResponse | SendMessageSuccessResponse
     """
     JSON-RPC response model for the 'message/send' method.

tests/server/apps/jsonrpc/test_fastapi_app.py

@@ -0,0 +1,23 @@
+from unittest.mock import MagicMock
+
+import pytest
+
+from a2a.server.apps.jsonrpc.jsonrpc_app import JSONRPCApplicationBuilder
+from a2a.server.apps.jsonrpc.fastapi_app import A2AFastAPIApplication
+from a2a.server.request_handlers.request_handler import RequestHandler
+from a2a.types import AgentCard
+
+
+def test_builder_protocol():
+    """
+    Tests that `A2AFastAPIApplication` matches the `JSONRPCApplicationBuilder` protocol.
+
+    This ensures that `A2AFastAPIApplication` can be used interchangeably where
+    `JSONRPCApplicationBuilder` is expected.
+    """
+    agent_card = MagicMock(spec=AgentCard)
+    agent_card.url = 'http://mockurl.com'
+    agent_card.supportsAuthenticatedExtendedCard = False
+    http_handler = MagicMock(spec=RequestHandler)
+    application = A2AFastAPIApplication(agent_card, http_handler)
+    assert isinstance(application, JSONRPCApplicationBuilder)

tests/server/apps/jsonrpc/test_starlette_app.py

@@ -0,0 +1,23 @@
+from unittest.mock import MagicMock
+
+import pytest
+
+from a2a.server.apps.jsonrpc.jsonrpc_app import JSONRPCApplicationBuilder
+from a2a.server.apps.jsonrpc.starlette_app import A2AStarletteApplication
+from a2a.server.request_handlers.request_handler import RequestHandler
+from a2a.types import AgentCard
+
+
+def test_builder_protocol():
+    """
+    Tests that `A2AStarletteApplication` matches the `JSONRPCApplicationBuilder` protocol.
+
+    This ensures that `A2AStarletteApplication` can be used interchangeably where
+    `JSONRPCApplicationBuilder` is expected.
+    """
+    agent_card = MagicMock(spec=AgentCard)
+    agent_card.url = 'http://mockurl.com'
+    agent_card.supportsAuthenticatedExtendedCard = False
+    http_handler = MagicMock(spec=RequestHandler)
+    application = A2AStarletteApplication(agent_card, http_handler)
+    assert isinstance(application, JSONRPCApplicationBuilder)