From 3aa937def435d570d76cd2cf6f3454346bba6940 Mon Sep 17 00:00:00 2001 From: Evan <2869018789@qq.com> Date: Thu, 30 Jul 2026 17:32:56 +0800 Subject: [PATCH] test: use caplog for telemetry error logging (#39722) --- .../unit_tests/tasks/test_community_telemetry_task.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/api/tests/unit_tests/tasks/test_community_telemetry_task.py b/api/tests/unit_tests/tasks/test_community_telemetry_task.py index 7367b420483..c90ecd67109 100644 --- a/api/tests/unit_tests/tasks/test_community_telemetry_task.py +++ b/api/tests/unit_tests/tasks/test_community_telemetry_task.py @@ -1,3 +1,4 @@ +import logging from types import SimpleNamespace from unittest.mock import MagicMock, Mock @@ -25,16 +26,16 @@ def test_send_community_telemetry_heartbeat_reports_with_a_database_session(monk report_heartbeat.assert_called_once_with(session=session) -def test_send_community_telemetry_heartbeat_swallows_report_errors(monkeypatch: pytest.MonkeyPatch): +def test_send_community_telemetry_heartbeat_swallows_report_errors( + monkeypatch: pytest.MonkeyPatch, caplog: pytest.LogCaptureFixture +): _configure_task_session(monkeypatch) monkeypatch.setattr( community_telemetry_task.CommunityTelemetryService, "report_heartbeat", Mock(side_effect=RuntimeError("telemetry unavailable")), ) - log_debug = Mock() - monkeypatch.setattr(community_telemetry_task.logger, "debug", log_debug) - + caplog.set_level(logging.DEBUG, logger=community_telemetry_task.logger.name) community_telemetry_task.send_community_telemetry_heartbeat.run() - log_debug.assert_called_once_with("Failed to process community telemetry heartbeat", exc_info=True) + assert "Failed to process community telemetry heartbeat" in caplog.text