Gemini fixes

Author: ishymko

tests/client/transports/test_grpc_client.py

@@ -568,9 +568,7 @@ async def test_delete_task_callback(
     sample_task_push_notification_config: TaskPushNotificationConfig,
 ) -> None:
     """Test deleting task push notification config."""
-    mock_grpc_stub.DeleteTaskPushNotificationConfig.return_value = (
-        sample_task_push_notification_config
-    )
+    mock_grpc_stub.DeleteTaskPushNotificationConfig.return_value = None
 
     await grpc_transport.delete_task_callback(
         DeleteTaskPushNotificationConfigRequest(

tests/client/transports/test_jsonrpc_client.py

@@ -416,7 +416,7 @@ async def test_delete_task_callback_success(
         )
         response = await transport.delete_task_callback(request)
 
-        mock_httpx_client.post.assert_called_once()  # Assuming mock_send_request was a typo for mock_httpx_client.post
+        mock_httpx_client.post.assert_called_once()
         assert response is None
         call_args = mock_httpx_client.post.call_args
         payload = call_args[1]['json']

tests/client/transports/test_rest_client.py

@@ -340,13 +340,7 @@ async def test_delete_task_callback_success(
         task_id = 'task-1'
         mock_response = AsyncMock(spec=httpx.Response)
         mock_response.status_code = 200
-        mock_response.json.return_value = {
-            'taskId': task_id,
-            'pushNotificationConfig': {
-                'id': 'config-1',
-                'url': 'https://example.com',
-            },
-        }
+        mock_response.json.return_value = {}
         mock_httpx_client.send.return_value = mock_response
 
         # Mock the build_request method to capture its inputs

tests/integration/test_client_server_integration.py

@@ -137,9 +137,7 @@ async def stream_side_effect(*args, **kwargs):
     handler.on_list_task_push_notification_configs.return_value = (
         ListTaskPushNotificationConfigsResponse(configs=[CALLBACK_CONFIG])
     )
-    handler.on_delete_task_push_notification_config.return_value = (
-        CALLBACK_CONFIG
-    )
+    handler.on_delete_task_push_notification_config.return_value = None
 
     async def resubscribe_side_effect(*args, **kwargs):
         yield RESUBSCRIBE_EVENT