mirror of
https://github.com/ZhuLinsen/daily_stock_analysis
synced 2026-09-20 10:53:33 +08:00
eb1333a678
* feat: add one-shot index analysis via --stocks with structured AnalysisTarget pipeline * fix: render index data sources in aggregate reports * fix: refresh index registry before --stocks parsing and filter indices on CN holidays --------- Co-authored-by: zhulinsen <42829555+ZhuLinsen@users.noreply.github.com>
70 lines
2.7 KiB
Python
70 lines
2.7 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""Regression tests for pipeline data-fetch error handling."""
|
|
|
|
from datetime import date, datetime, timezone
|
|
import unittest
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
from src.core.pipeline import StockAnalysisPipeline
|
|
|
|
|
|
class PipelineFetchErrorTestCase(unittest.TestCase):
|
|
"""`fetch_and_save_stock_data` should preserve the original exception."""
|
|
|
|
def test_fetch_and_save_handles_stock_name_lookup_failure(self):
|
|
pipeline = StockAnalysisPipeline.__new__(StockAnalysisPipeline)
|
|
pipeline.fetcher_manager = MagicMock()
|
|
pipeline.db = MagicMock()
|
|
pipeline.fetcher_manager.get_stock_name.side_effect = RuntimeError("name lookup failed")
|
|
|
|
success, error = StockAnalysisPipeline.fetch_and_save_stock_data(pipeline, "600519")
|
|
|
|
self.assertFalse(success)
|
|
self.assertIn("name lookup failed", error or "")
|
|
|
|
@patch.object(
|
|
StockAnalysisPipeline,
|
|
"_resolve_resume_target_date",
|
|
return_value=date(2026, 3, 27),
|
|
)
|
|
def test_fetch_and_save_uses_effective_trading_date_for_resume_check(self, _mock_target):
|
|
pipeline = StockAnalysisPipeline.__new__(StockAnalysisPipeline)
|
|
pipeline.fetcher_manager = MagicMock()
|
|
pipeline.db = MagicMock()
|
|
pipeline.fetcher_manager.get_stock_name.return_value = "贵州茅台"
|
|
pipeline.db.has_today_data.return_value = True
|
|
current_time = datetime(2026, 3, 28, 1, 0, tzinfo=timezone.utc)
|
|
|
|
success, error = StockAnalysisPipeline.fetch_and_save_stock_data(
|
|
pipeline,
|
|
"600519",
|
|
current_time=current_time,
|
|
)
|
|
|
|
self.assertTrue(success)
|
|
self.assertIsNone(error)
|
|
_mock_target.assert_called_once_with(
|
|
"600519", current_time=current_time, analysis_target=None
|
|
)
|
|
pipeline.db.has_today_data.assert_called_once_with("600519", date(2026, 3, 27))
|
|
pipeline.fetcher_manager.get_daily_data.assert_not_called()
|
|
|
|
def test_resolve_resume_target_date_normalizes_supported_a_share_formats(self):
|
|
with patch("src.core.pipeline.get_market_for_stock", return_value="cn") as mock_market, patch(
|
|
"src.core.pipeline.get_effective_trading_date",
|
|
return_value=date(2026, 3, 27),
|
|
) as mock_target:
|
|
for code in ("SH600519", "000001.SZ", "BJ920748"):
|
|
result = StockAnalysisPipeline._resolve_resume_target_date(code)
|
|
self.assertEqual(result, date(2026, 3, 27))
|
|
|
|
self.assertEqual(
|
|
[args.args[0] for args in mock_market.call_args_list],
|
|
["600519", "000001", "920748"],
|
|
)
|
|
self.assertEqual(mock_target.call_count, 3)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|