修复测试:guard mock 补充 check_daily_query_limit 方法

Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
This commit is contained in:
AmandaloveYang
2026-05-11 14:24:35 +08:00
parent 3c2d1ce6fa
commit 9ff2a99618
+12 -3
View File
@@ -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),