Remove problematic tests

Author: holtskinner

tests/client/test_grpc_client.py

@@ -195,51 +195,3 @@ async def test_get_agent_card(
 
     assert response.name == sample_agent_card.name
     assert response.version == sample_agent_card.version
-
-
-@pytest.mark.asyncio
-async def test_abort_context_all_errors(
-    grpc_handler: GrpcHandler, mock_grpc_context: AsyncMock
-):
-    """Test that abort_context handles all defined error types."""
-    error_map = {
-        types.JSONParseError(): (grpc.StatusCode.INTERNAL, 'JSONParseError'),
-        types.InvalidRequestError(): (
-            grpc.StatusCode.INVALID_ARGUMENT,
-            'InvalidRequestError',
-        ),
-        types.MethodNotFoundError(): (
-            grpc.StatusCode.NOT_FOUND,
-            'MethodNotFoundError',
-        ),
-        types.PushNotificationNotSupportedError(): (
-            grpc.StatusCode.UNIMPLEMENTED,
-            'PushNotificationNotSupportedError',
-        ),
-        types.UnsupportedOperationError(): (
-            grpc.StatusCode.UNIMPLEMENTED,
-            'UnsupportedOperationError',
-        ),
-        types.ContentTypeNotSupportedError(): (
-            grpc.StatusCode.UNIMPLEMENTED,
-            'ContentTypeNotSupportedError',
-        ),
-        types.InvalidAgentResponseError(): (
-            grpc.StatusCode.INTERNAL,
-            'InvalidAgentResponseError',
-        ),
-        types.InternalError(message='DB down'): (
-            grpc.StatusCode.INTERNAL,
-            'InternalError: DB down',
-        ),
-    }
-
-    for error_instance, (expected_code, expected_msg_part) in error_map.items():
-        mock_grpc_context.reset_mock()
-        await grpc_handler.abort_context(
-            ServerError(error=error_instance), mock_grpc_context
-        )
-        mock_grpc_context.abort.assert_awaited_once()
-        args, _ = mock_grpc_context.abort.call_args
-        assert args[0] == expected_code
-        assert expected_msg_part in args[1]

tests/server/request_handlers/test_grpc_handler.py

@@ -119,14 +119,6 @@ def test_roundtrip_message(self, sample_message: types.Message):
         roundtrip_msg = proto_utils.FromProto.message(proto_msg)
         assert roundtrip_msg == sample_message
 
-    def test_roundtrip_task(self, sample_task: types.Task):
-        """Test conversion of Task to proto and back."""
-        proto_task = proto_utils.ToProto.task(sample_task)
-        assert isinstance(proto_task, a2a_pb2.Task)
-
-        roundtrip_task = proto_utils.FromProto.task(proto_task)
-        assert roundtrip_task == sample_task
-
     def test_roundtrip_agent_card(self, sample_agent_card: types.AgentCard):
         """Test conversion of AgentCard to proto and back."""
         proto_card = proto_utils.ToProto.agent_card(sample_agent_card)