Adjust tests to check for logger.exception

Author: holtskinner

tests/server/tasks/test_inmemory_push_notifications.py

@@ -220,12 +220,13 @@ async def test_send_notification_http_status_error(
         await self.notifier.send_notification(task_data)  # Pass only task_data
 
         self.mock_httpx_client.post.assert_awaited_once()
-        mock_logger.error.assert_called_once()
+        mock_logger.exception.assert_called_once()
         # Check that the error message contains the generic part and the specific exception string
         self.assertIn(
-            'Error sending push-notification', mock_logger.error.call_args[0][0]
+            'Error sending push-notification',
+            mock_logger.exception.call_args[0][0],
         )
-        self.assertIn(str(http_error), mock_logger.error.call_args[0][0])
+        self.assertIn(str(http_error), mock_logger.exception.call_args[0][0])
 
     @patch('a2a.server.tasks.base_push_notification_sender.logger')
     async def test_send_notification_request_error(
@@ -242,11 +243,12 @@ async def test_send_notification_request_error(
         await self.notifier.send_notification(task_data)  # Pass only task_data
 
         self.mock_httpx_client.post.assert_awaited_once()
-        mock_logger.error.assert_called_once()
+        mock_logger.exception.assert_called_once()
         self.assertIn(
-            'Error sending push-notification', mock_logger.error.call_args[0][0]
+            'Error sending push-notification',
+            mock_logger.exception.call_args[0][0],
         )
-        self.assertIn(str(request_error), mock_logger.error.call_args[0][0])
+        self.assertIn(str(request_error), mock_logger.exception.call_args[0][0])
 
     @patch('a2a.server.tasks.base_push_notification_sender.logger')
     async def test_send_notification_with_auth(self, mock_logger: MagicMock):

tests/server/tasks/test_push_notification_sender.py

@@ -123,7 +123,7 @@ async def test_send_notification_http_status_error(
             json=task_data.model_dump(mode='json', exclude_none=True),
             headers=None,
         )
-        mock_logger.error.assert_called_once()
+        mock_logger.exception.assert_called_once()
 
     async def test_send_notification_multiple_configs(self):
         task_id = 'task_multiple_configs'

tests/utils/test_telemetry.py

@@ -78,7 +78,7 @@ def foo() -> int:
             return 1
 
         foo()
-        logger.error.assert_any_call(mock.ANY)
+        logger.exception.assert_any_call(mock.ANY)
 
 
 @pytest.mark.asyncio