fix tests

Author: sokoliva

tests/server/request_handlers/test_default_request_handler_v2.py

@@ -30,7 +30,7 @@
     InternalError,
     InvalidParamsError,
     TaskNotFoundError,
-    UnsupportedOperationError,
+    PushNotificationNotSupportedError,
 )
 from a2a.types.a2a_pb2 import (
     AgentCapabilities,
@@ -354,7 +354,7 @@ async def test_set_task_push_notification_config_no_notifier():
     params = TaskPushNotificationConfig(
         task_id='task1', url='http://example.com'
     )
-    with pytest.raises(UnsupportedOperationError):
+    with pytest.raises(PushNotificationNotSupportedError):
         await request_handler.on_create_task_push_notification_config(
             params, create_server_call_context()
         )
@@ -398,7 +398,7 @@ async def test_get_task_push_notification_config_no_store():
     params = GetTaskPushNotificationConfigRequest(
         task_id='task1', id='task_push_notification_config'
     )
-    with pytest.raises(UnsupportedOperationError):
+    with pytest.raises(PushNotificationNotSupportedError):
         await request_handler.on_get_task_push_notification_config(
             params, create_server_call_context()
         )
@@ -583,7 +583,7 @@ async def test_list_task_push_notification_config_no_store():
         agent_card=create_default_agent_card(),
     )
     params = ListTaskPushNotificationConfigsRequest(task_id='task1')
-    with pytest.raises(UnsupportedOperationError):
+    with pytest.raises(PushNotificationNotSupportedError):
         await request_handler.on_list_task_push_notification_configs(
             params, create_server_call_context()
         )
@@ -711,11 +711,11 @@ async def test_delete_task_push_notification_config_no_store():
     params = DeleteTaskPushNotificationConfigRequest(
         task_id='task1', id='config1'
     )
-    with pytest.raises(UnsupportedOperationError) as exc_info:
+    with pytest.raises(PushNotificationNotSupportedError) as exc_info:
         await request_handler.on_delete_task_push_notification_config(
             params, create_server_call_context()
         )
-    assert isinstance(exc_info.value, UnsupportedOperationError)
+    assert isinstance(exc_info.value, PushNotificationNotSupportedError)
 
 
 @pytest.mark.asyncio