fix

Author: guglielmo-san

src/a2a/server/request_handlers/__init__.py

@@ -38,7 +38,6 @@ def __init__(self, *args, **kwargs):
 __all__ = [
     'DefaultRequestHandler',
     'GrpcHandler',
-    'JSONRPCHandler',
     'RequestHandler',
     'build_error_response',
     'prepare_response_object',

tests/server/routes/test_rest_dispatcher.py

@@ -139,6 +139,16 @@ async def test_build_call_context(self, rest_dispatcher_instance):
 
 @pytest.mark.asyncio
 class TestRestDispatcherEndpoints:
+    async def test_on_message_send_throws_error_for_unsupported_version(
+        self, rest_dispatcher_instance, mock_handler
+    ):
+        # 0.3 is currently not supported for direct message sending on RestDispatcher
+        req = make_mock_request(method='POST', headers={'a2a-version': '0.3.0'})
+        response = await rest_dispatcher_instance.on_message_send(req)
+        
+        # VersionNotSupportedError maps to 400 Bad Request
+        assert response.status_code == 400
+
     async def test_on_message_send_returns_message(
         self, rest_dispatcher_instance, mock_handler
     ):
@@ -286,3 +296,36 @@ async def test_on_subscribe_to_task_unsupported(
 
         response = await dispatcher.on_subscribe_to_task(req)
         assert response.status_code == 400
+
+    async def test_on_message_send_stream_success(
+        self, rest_dispatcher_instance
+    ):
+        req = make_mock_request(method='POST')
+        response = await rest_dispatcher_instance.on_message_send_stream(req)
+        
+        assert response.status_code == 200
+        
+        chunks = []
+        async for chunk in response.body_iterator:
+            chunks.append(chunk)
+            
+        assert len(chunks) == 2
+        # sse-starlette yields strings or bytes formatted as Server-Sent Events
+        assert 'chunk1' in str(chunks[0])
+        assert 'chunk2' in str(chunks[1])
+
+    async def test_on_subscribe_to_task_success(
+        self, rest_dispatcher_instance
+    ):
+        req = make_mock_request(method='GET', path_params={'id': 'test_task'})
+        response = await rest_dispatcher_instance.on_subscribe_to_task(req)
+        
+        assert response.status_code == 200
+        
+        chunks = []
+        async for chunk in response.body_iterator:
+            chunks.append(chunk)
+
+        assert len(chunks) == 2
+        assert 'chunk1' in str(chunks[0])
+        assert 'chunk2' in str(chunks[1])