From 9ff2a99618309f22befbba2fbe654720640d0ad0 Mon Sep 17 00:00:00 2001 From: AmandaloveYang <2569718930@qq.com> Date: Mon, 11 May 2026 14:24:35 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E6=B5=8B=E8=AF=95=EF=BC=9Agu?= =?UTF-8?q?ard=20mock=20=E8=A1=A5=E5=85=85=20check=5Fdaily=5Fquery=5Flimit?= =?UTF-8?q?=20=E6=96=B9=E6=B3=95?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Co-Authored-By: Claude Opus 4.7 --- tests/test_bot_handlers.py | 15 ++++++++++++--- 1 file changed, 12 insertions(+), 3 deletions(-) diff --git a/tests/test_bot_handlers.py b/tests/test_bot_handlers.py index 254fc319..44294870 100644 --- a/tests/test_bot_handlers.py +++ b/tests/test_bot_handlers.py @@ -18,7 +18,10 @@ def _message(text: str): def test_city_handler_missing_city_shows_usage(): bot = SimpleNamespace() - guard = SimpleNamespace(ensure_access_and_points=Mock(return_value=True)) + guard = SimpleNamespace( + check_daily_query_limit=Mock(return_value=True), + ensure_access_and_points=Mock(return_value=True), + ) city_service = SimpleNamespace(resolve_city=Mock(), build_report=Mock()) io_layer = SimpleNamespace(send_query_message=Mock()) handler = CityCommandHandler( @@ -37,7 +40,10 @@ def test_city_handler_missing_city_shows_usage(): def test_city_handler_happy_path_pushes_progress_and_report(): bot = SimpleNamespace() - guard = SimpleNamespace(ensure_access_and_points=Mock(return_value=True)) + guard = SimpleNamespace( + check_daily_query_limit=Mock(return_value=True), + ensure_access_and_points=Mock(return_value=True), + ) city_service = SimpleNamespace( resolve_city=Mock( return_value=CityResolveResult( @@ -68,7 +74,10 @@ def test_city_handler_happy_path_pushes_progress_and_report(): def test_deb_handler_history_missing_returns_hint(): bot = SimpleNamespace() - guard = SimpleNamespace(ensure_access_and_points=Mock(return_value=True)) + guard = SimpleNamespace( + check_daily_query_limit=Mock(return_value=True), + ensure_access_and_points=Mock(return_value=True), + ) deb_service = SimpleNamespace( resolve_city=Mock(return_value="ankara"), has_history=Mock(return_value=False),