mirror of
https://github.com/ZhuLinsen/daily_stock_analysis
synced 2026-09-20 02:43:35 +08:00
e866cfa48d
* feat: add bot index analysis entry * test: add Bot index online smoke * fix: guard Windows-only spawn flag in smoke tests for Linux CI * fix: clean up worker tree on parent interrupt in smoke runner * fix: surface interrupt cleanup failures in smoke runner --------- Co-authored-by: zhulinsen <42829555+ZhuLinsen@users.noreply.github.com>
364 lines
16 KiB
Python
364 lines
16 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""Tests for the shared ``/analyze`` command (Bot 指数入口, PR2).
|
|
|
|
Covers the real ``CommandDispatcher`` gate (``validate_args`` before
|
|
``execute``) with a stubbed ``TaskService`` so the tests assert the
|
|
code/target handed to ``submit_analysis`` without touching the network or
|
|
the real pipeline.
|
|
"""
|
|
|
|
import unittest
|
|
from datetime import datetime
|
|
from unittest.mock import patch
|
|
|
|
# Keep tests runnable when optional deps are missing.
|
|
try:
|
|
import litellm # noqa: F401
|
|
except ModuleNotFoundError:
|
|
from tests.litellm_stub import ensure_litellm_stub
|
|
ensure_litellm_stub()
|
|
|
|
from bot.commands.analyze import AnalyzeCommand
|
|
from bot.dispatcher import CommandDispatcher
|
|
from bot.models import BotMessage, BotResponse, ChatType
|
|
from src.services.stock_list_parser import ParseStatus
|
|
|
|
|
|
class _StubTaskService:
|
|
"""Records the last ``submit_analysis`` call; never touches the network.
|
|
|
|
Supports two canned outcomes (success by default, or failure) so tests
|
|
can assert both the success ``extra`` contract and the no-identity error
|
|
contract without touching the network or the real pipeline.
|
|
"""
|
|
|
|
def __init__(self, fail: bool = False):
|
|
self.calls = []
|
|
self._fail = fail
|
|
|
|
def submit_analysis(self, **kwargs):
|
|
self.calls.append(kwargs)
|
|
if self._fail:
|
|
return {"success": False, "task_id": "", "error": "boom"}
|
|
return {
|
|
"success": True,
|
|
"task_id": "task-1234567890abcdef",
|
|
"code": kwargs.get("code", ""),
|
|
}
|
|
|
|
|
|
def _make_message(content: str) -> BotMessage:
|
|
return BotMessage(
|
|
platform="feishu",
|
|
message_id="m1",
|
|
user_id="u1",
|
|
user_name="tester",
|
|
chat_id="c1",
|
|
chat_type=ChatType.PRIVATE,
|
|
content=content,
|
|
raw_content=content,
|
|
mentioned=False,
|
|
timestamp=datetime.now(),
|
|
)
|
|
|
|
|
|
class TestAnalyzeCommandDispatcherGate(unittest.TestCase):
|
|
"""Real dispatcher gate: ``validate_args`` runs before ``execute`` and the
|
|
command submits through the shared ``AnalyzeCommand``."""
|
|
|
|
def _dispatch(self, content: str):
|
|
dispatcher = CommandDispatcher()
|
|
command = AnalyzeCommand()
|
|
dispatcher.register(command)
|
|
stub = _StubTaskService()
|
|
with patch("src.services.task_service.get_task_service", return_value=stub):
|
|
response = dispatcher.dispatch(_make_message(content))
|
|
return response, stub
|
|
|
|
def test_registered_code_submits_index_target(self):
|
|
response, stub = self._dispatch("/analyze sh000016")
|
|
self.assertIsInstance(response, BotResponse)
|
|
self.assertIn("分析任务已提交", response.text)
|
|
self.assertEqual(len(stub.calls), 1)
|
|
call = stub.calls[0]
|
|
self.assertEqual(call["code"], "sh000016")
|
|
target = call["analysis_target"]
|
|
self.assertIsNotNone(target)
|
|
self.assertEqual(target.asset_type, ParseStatus.INDEX)
|
|
self.assertEqual(target.canonical_id, "sh000016")
|
|
|
|
def test_csi_alias_converges_to_canonical_index_target(self):
|
|
response, stub = self._dispatch("/analyze 930955.CSI")
|
|
self.assertIn("分析任务已提交", response.text)
|
|
call = stub.calls[0]
|
|
self.assertEqual(call["code"], "csi930955")
|
|
target = call["analysis_target"]
|
|
self.assertEqual(target.asset_type, ParseStatus.INDEX)
|
|
self.assertEqual(target.canonical_id, "csi930955")
|
|
|
|
def test_registered_prefix_alias_submits_index_target(self):
|
|
"""``csi930955`` is the registered canonical CSI prefix form; it must
|
|
submit with a matching INDEX target."""
|
|
response, stub = self._dispatch("/analyze csi930955")
|
|
self.assertIn("分析任务已提交", response.text)
|
|
call = stub.calls[0]
|
|
self.assertEqual(call["code"], "csi930955")
|
|
target = call["analysis_target"]
|
|
self.assertIsNotNone(target)
|
|
self.assertEqual(target.asset_type, ParseStatus.INDEX)
|
|
self.assertEqual(target.canonical_id, "csi930955")
|
|
|
|
def test_registered_suffix_alias_submits_index_target(self):
|
|
"""``000016.SH`` is the registered suffix alias of sh000016; it must
|
|
submit the registry canonical with a matching INDEX target."""
|
|
response, stub = self._dispatch("/analyze 000016.SH")
|
|
self.assertIn("分析任务已提交", response.text)
|
|
call = stub.calls[0]
|
|
self.assertEqual(call["code"], "sh000016")
|
|
target = call["analysis_target"]
|
|
self.assertIsNotNone(target)
|
|
self.assertEqual(target.asset_type, ParseStatus.INDEX)
|
|
self.assertEqual(target.canonical_id, "sh000016")
|
|
|
|
def test_dotted_us_ticker_keeps_legacy_path_without_target(self):
|
|
"""``BRK.B`` is a dotted US ticker the legacy gate accepts; it must
|
|
resolve to ``BRK.B`` with no structured target."""
|
|
response, stub = self._dispatch("/analyze BRK.B")
|
|
self.assertIn("分析任务已提交", response.text)
|
|
call = stub.calls[0]
|
|
self.assertEqual(call["code"], "BRK.B")
|
|
self.assertIsNone(call["analysis_target"])
|
|
|
|
def test_parser_index_alias_ss_dotted_prefix_submits_index_target(self):
|
|
"""``SS.000300`` is a parser-recognized dotted-prefix alias of the
|
|
registered sh000300 index (SS = Shanghai alias); the Bot must submit
|
|
the registry canonical with a matching INDEX target."""
|
|
response, stub = self._dispatch("/analyze SS.000300")
|
|
self.assertIn("分析任务已提交", response.text)
|
|
call = stub.calls[0]
|
|
self.assertEqual(call["code"], "sh000300")
|
|
target = call["analysis_target"]
|
|
self.assertIsNotNone(target)
|
|
self.assertEqual(target.asset_type, ParseStatus.INDEX)
|
|
self.assertEqual(target.canonical_id, "sh000300")
|
|
|
|
def test_dotted_prefix_sh_alias_submits_index_target(self):
|
|
"""``SH.000016`` is a parser-recognized dotted-prefix alias of the
|
|
registered sh000016 index; the Bot must submit the registry canonical
|
|
with a matching INDEX target."""
|
|
response, stub = self._dispatch("/analyze SH.000016")
|
|
self.assertIn("分析任务已提交", response.text)
|
|
call = stub.calls[0]
|
|
self.assertEqual(call["code"], "sh000016")
|
|
target = call["analysis_target"]
|
|
self.assertIsNotNone(target)
|
|
self.assertEqual(target.asset_type, ParseStatus.INDEX)
|
|
self.assertEqual(target.canonical_id, "sh000016")
|
|
|
|
def test_dotted_prefix_sz_alias_submits_index_target(self):
|
|
"""``SZ.399001`` is a parser-recognized dotted-prefix alias of the
|
|
registered sz399001 index; the Bot must submit the registry canonical
|
|
with a matching INDEX target."""
|
|
response, stub = self._dispatch("/analyze SZ.399001")
|
|
self.assertIn("分析任务已提交", response.text)
|
|
call = stub.calls[0]
|
|
self.assertEqual(call["code"], "sz399001")
|
|
target = call["analysis_target"]
|
|
self.assertIsNotNone(target)
|
|
self.assertEqual(target.asset_type, ParseStatus.INDEX)
|
|
self.assertEqual(target.canonical_id, "sz399001")
|
|
|
|
def test_registered_name_wins_over_stock_name_fallback(self):
|
|
with patch(
|
|
"src.services.name_to_code_resolver.resolve_name_to_code"
|
|
) as mock_resolve:
|
|
response, stub = self._dispatch("/analyze 上证50")
|
|
self.assertIn("分析任务已提交", response.text)
|
|
call = stub.calls[0]
|
|
self.assertEqual(call["code"], "sh000016")
|
|
target = call["analysis_target"]
|
|
self.assertEqual(target.asset_type, ParseStatus.INDEX)
|
|
self.assertEqual(target.canonical_id, "sh000016")
|
|
self.assertEqual(target.raw_input, "上证50")
|
|
mock_resolve.assert_not_called()
|
|
|
|
def test_stock_code_keeps_legacy_code_and_no_target(self):
|
|
response, stub = self._dispatch("/analyze 600519")
|
|
self.assertIn("分析任务已提交", response.text)
|
|
call = stub.calls[0]
|
|
self.assertEqual(call["code"], "600519")
|
|
self.assertIsNone(call["analysis_target"])
|
|
|
|
def test_stock_name_fallback_keeps_legacy_code(self):
|
|
"""Deterministic: the AkShare/network online map is stubbed empty so
|
|
the real top-level ``resolve_name_to_code`` runs its local tables
|
|
only and still resolves 贵州茅台 to its legacy code."""
|
|
response, stub = None, None
|
|
with patch(
|
|
"src.services.name_to_code_resolver._get_akshare_name_to_code",
|
|
return_value={},
|
|
):
|
|
response, stub = self._dispatch("/analyze 贵州茅台")
|
|
self.assertIn("分析任务已提交", response.text)
|
|
call = stub.calls[0]
|
|
self.assertEqual(call["code"], "600519")
|
|
self.assertIsNone(call["analysis_target"])
|
|
|
|
def test_hk_and_us_stock_keep_legacy_code(self):
|
|
# ``hk00700`` is uppercased to ``HK00700`` by the legacy resolver —
|
|
# the same behavior the pre-PR command already had.
|
|
for content, expected in (("/analyze hk00700", "HK00700"), ("/analyze AAPL", "AAPL")):
|
|
response, stub = self._dispatch(content)
|
|
self.assertIn("分析任务已提交", response.text)
|
|
call = stub.calls[0]
|
|
self.assertEqual(call["code"], expected)
|
|
self.assertIsNone(call["analysis_target"])
|
|
|
|
def test_lowercase_us_ticker_keeps_legacy_uppercase_path(self):
|
|
"""``usfd`` is a real US ticker the old gate accepted (case-insensitive
|
|
1-5 letters); it must resolve to ``USFD`` with no target, exactly as
|
|
before this change."""
|
|
response, stub = self._dispatch("/analyze usfd")
|
|
self.assertIn("分析任务已提交", response.text)
|
|
call = stub.calls[0]
|
|
self.assertEqual(call["code"], "USFD")
|
|
self.assertIsNone(call["analysis_target"])
|
|
|
|
def test_legacy_invalid_shapes_return_error_without_submission(self):
|
|
"""Shapes the old validate_args gate rejected must stay rejected —
|
|
explicit error and no TaskService call."""
|
|
for code in ("12345", "00700", "600519.SH", "sh999999"):
|
|
with self.subTest(code=code):
|
|
response, stub = self._dispatch(f"/analyze {code}")
|
|
self.assertIn("无效的标的代码", response.text)
|
|
self.assertEqual(len(stub.calls), 0)
|
|
|
|
def test_unregistered_csi_returns_error_and_does_not_submit(self):
|
|
response, stub = self._dispatch("/analyze 930956.CSI")
|
|
self.assertIn("无法分析", response.text)
|
|
self.assertIn("CSI", response.text)
|
|
self.assertEqual(len(stub.calls), 0)
|
|
|
|
def test_malformed_csi_numeric_forms_return_csi_error_without_submission(
|
|
self,
|
|
):
|
|
"""The full numeric explicit CSI family (``csi`` + digits, or digits +
|
|
``.csi``) is rejected with the CSI-specific error — never submitted,
|
|
never routed into stock-name resolution."""
|
|
for code in ("csi930956", "csi123", "12345.csi"):
|
|
with self.subTest(code=code):
|
|
response, stub = self._dispatch(f"/analyze {code}")
|
|
self.assertIn("无法分析", response.text)
|
|
self.assertIn("CSI", response.text)
|
|
self.assertEqual(len(stub.calls), 0)
|
|
|
|
def test_parser_unsupported_malformed_code_returns_error_without_submission(
|
|
self,
|
|
):
|
|
"""A non-CSI parser-UNSUPPORTED malformed code shape (``us1``) must
|
|
return an explicit error and never submit a task."""
|
|
response, stub = self._dispatch("/analyze us1")
|
|
self.assertIn("无法分析", response.text)
|
|
self.assertEqual(len(stub.calls), 0)
|
|
|
|
def test_unknown_name_returns_error_and_does_not_submit(self):
|
|
"""Deterministic: the AkShare/network online map is stubbed empty so
|
|
the real ``resolve_name_to_code`` runs its local tables only."""
|
|
response, stub = None, None
|
|
with patch(
|
|
"src.services.name_to_code_resolver._get_akshare_name_to_code",
|
|
return_value={},
|
|
):
|
|
response, stub = self._dispatch("/analyze 不存在标的")
|
|
self.assertIn("无法识别标的", response.text)
|
|
self.assertEqual(len(stub.calls), 0)
|
|
|
|
def test_empty_args_rejected_by_validate_args_gate(self):
|
|
response, stub = self._dispatch("/analyze")
|
|
self.assertIn("请输入", response.text)
|
|
self.assertEqual(len(stub.calls), 0)
|
|
|
|
def test_full_report_flag_still_works(self):
|
|
response, stub = self._dispatch("/analyze sh000016 full")
|
|
self.assertIn("分析任务已提交", response.text)
|
|
call = stub.calls[0]
|
|
self.assertEqual(call["code"], "sh000016")
|
|
self.assertEqual(call["report_type"].value, "full")
|
|
|
|
def test_success_text_is_unchanged_and_carries_task_identity_extra(self):
|
|
"""On a successful submission the user-visible text must stay exactly
|
|
as before, while ``extra`` carries the internal task identity
|
|
(``task_id`` + ``stock_code``) for transport-independent consumers."""
|
|
response, stub = self._dispatch("/analyze sh000016")
|
|
expected_text = (
|
|
"✅ **分析任务已提交**\n\n"
|
|
"• 标的: `sh000016`\n"
|
|
"• 报告类型: 精简报告\n"
|
|
"• 任务 ID: `task-1234567890abcde...`\n\n"
|
|
"分析完成后将自动推送结果。"
|
|
)
|
|
self.assertEqual(response.text, expected_text)
|
|
self.assertEqual(response.extra, {
|
|
"task_id": "task-1234567890abcdef",
|
|
"stock_code": "sh000016",
|
|
})
|
|
|
|
def test_success_extra_uses_normalized_code(self):
|
|
"""``extra.stock_code`` carries the normalized code the task was
|
|
submitted under (the registry canonical for an index alias)."""
|
|
response, stub = self._dispatch("/analyze 930955.CSI")
|
|
self.assertIn("分析任务已提交", response.text)
|
|
self.assertEqual(response.extra["task_id"], "task-1234567890abcdef")
|
|
self.assertEqual(response.extra["stock_code"], "csi930955")
|
|
|
|
def test_error_response_has_no_task_identity_extra(self):
|
|
"""On a submission failure there is no task identity: no task was
|
|
created, so ``extra`` must stay empty."""
|
|
dispatcher = CommandDispatcher()
|
|
dispatcher.register(AnalyzeCommand())
|
|
stub = _StubTaskService(fail=True)
|
|
with patch("src.services.task_service.get_task_service", return_value=stub):
|
|
response = dispatcher.dispatch(_make_message("/analyze sh000016"))
|
|
self.assertIn("提交分析任务失败", response.text)
|
|
self.assertEqual(response.extra, {})
|
|
|
|
|
|
class TestAnalyzeCommandAmbiguousName(unittest.TestCase):
|
|
"""Ambiguous registered display names must fail with an explicit error and
|
|
never fall back to stock-name resolution."""
|
|
|
|
def test_ambiguous_display_name_returns_error(self):
|
|
from src.services.stock_list_parser import IndexEntry, IndexRegistry
|
|
|
|
registry = IndexRegistry((
|
|
IndexEntry(
|
|
bare_code="000300",
|
|
exchange="SH",
|
|
canonical_id="sh000300",
|
|
display_name="沪深300",
|
|
),
|
|
IndexEntry(
|
|
bare_code="000999",
|
|
exchange="SH",
|
|
canonical_id="sh000999",
|
|
display_name="沪深300",
|
|
),
|
|
))
|
|
dispatcher = CommandDispatcher()
|
|
dispatcher.register(AnalyzeCommand())
|
|
stub = _StubTaskService()
|
|
with patch(
|
|
"bot.commands.analyze.default_index_registry", return_value=registry
|
|
), patch(
|
|
"src.services.name_to_code_resolver.resolve_name_to_code"
|
|
) as mock_resolve, patch(
|
|
"src.services.task_service.get_task_service", return_value=stub
|
|
):
|
|
response = dispatcher.dispatch(_make_message("/analyze 沪深300"))
|
|
self.assertIn("歧义", response.text)
|
|
self.assertEqual(stub.calls, [])
|
|
mock_resolve.assert_not_called()
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|