mirror of
https://github.com/ZhuLinsen/daily_stock_analysis
synced 2026-09-20 10:53:33 +08:00
296 lines
9.6 KiB
Python
296 lines
9.6 KiB
Python
# -*- coding: utf-8 -*-
|
||
"""Tests for extracting DecisionSignal assets from completed reports."""
|
||
|
||
from __future__ import annotations
|
||
|
||
import os
|
||
|
||
import pytest
|
||
|
||
from src.analyzer import AnalysisResult
|
||
from src.config import Config
|
||
from src.services.decision_signal_extractor import (
|
||
build_decision_signal_payload_from_report,
|
||
extract_and_persist_from_analysis_result,
|
||
)
|
||
from src.services.decision_signal_service import DecisionSignalService
|
||
from src.storage import DatabaseManager
|
||
|
||
|
||
@pytest.fixture()
|
||
def isolated_db(tmp_path):
|
||
old_database_path = os.environ.get("DATABASE_PATH")
|
||
db_path = tmp_path / "decision_signal_extractor.db"
|
||
os.environ["DATABASE_PATH"] = str(db_path)
|
||
Config.reset_instance()
|
||
DatabaseManager.reset_instance()
|
||
db = DatabaseManager.get_instance()
|
||
try:
|
||
yield db
|
||
finally:
|
||
DatabaseManager.reset_instance()
|
||
Config.reset_instance()
|
||
if old_database_path is None:
|
||
os.environ.pop("DATABASE_PATH", None)
|
||
else:
|
||
os.environ["DATABASE_PATH"] = old_database_path
|
||
|
||
|
||
def _result(**overrides) -> AnalysisResult:
|
||
result = AnalysisResult(
|
||
code="600519",
|
||
name="贵州茅台",
|
||
sentiment_score=82,
|
||
trend_prediction="看多",
|
||
operation_advice="买入",
|
||
decision_type="buy",
|
||
confidence_level="高",
|
||
analysis_summary="趋势确认，量价配合。",
|
||
risk_warning="跌破支撑需止损",
|
||
report_language="zh",
|
||
)
|
||
result.dashboard = {
|
||
"battle_plan": {
|
||
"sniper_points": {
|
||
"ideal_buy": "理想买入点：1700元",
|
||
"secondary_buy": "1680-1690（回踩MA5附近）",
|
||
"stop_loss": "止损位：1600元",
|
||
"take_profit": "目标位：1850元",
|
||
},
|
||
"action_checklist": ["放量突破前高", "回踩不破MA10"],
|
||
},
|
||
"phase_decision": {
|
||
"watch_conditions": ["盘中量能继续放大"],
|
||
},
|
||
"intelligence": {
|
||
"risk_alerts": ["估值偏高"],
|
||
"positive_catalysts": ["业绩超预期"],
|
||
},
|
||
}
|
||
for key, value in overrides.items():
|
||
setattr(result, key, value)
|
||
return result
|
||
|
||
|
||
def test_build_payload_maps_report_context_and_price_plan() -> None:
|
||
result = _result()
|
||
result.market_phase_summary = {"phase": "postmarket"}
|
||
result.analysis_context_pack_overview = {"data_quality": {"overall_score": 55, "level": "fair"}}
|
||
context_snapshot = {
|
||
"market_phase_summary": {"phase": "intraday"},
|
||
"analysis_context_pack_overview": {
|
||
"data_quality": {"overall_score": 91, "level": "good"},
|
||
},
|
||
}
|
||
|
||
payload = build_decision_signal_payload_from_report(
|
||
result,
|
||
context_snapshot=context_snapshot,
|
||
source_report_id=88,
|
||
trace_id="trace-88",
|
||
query_source="api",
|
||
report_type="full",
|
||
)
|
||
|
||
assert payload is not None
|
||
assert payload["stock_code"] == "600519"
|
||
assert payload["stock_name"] == "贵州茅台"
|
||
assert payload["market"] == "cn"
|
||
assert payload["source_type"] == "analysis"
|
||
assert payload["source_report_id"] == 88
|
||
assert payload["trace_id"] == "trace-88"
|
||
assert payload["trigger_source"] == "api"
|
||
assert payload["action"] == "buy"
|
||
assert payload["confidence"] == 0.8
|
||
assert payload["score"] == 82
|
||
assert payload["market_phase"] == "intraday"
|
||
assert payload["entry_low"] == 1690.0
|
||
assert payload["entry_high"] == 1700.0
|
||
assert payload["stop_loss"] == 1600.0
|
||
assert payload["target_price"] == 1850.0
|
||
assert payload["data_quality_summary"]["overall_score"] == 91
|
||
assert payload["watch_conditions"] == ["盘中量能继续放大"]
|
||
assert payload["risk_summary"] == ["跌破支撑需止损", "估值偏高"]
|
||
assert payload["catalyst_summary"] == ["业绩超预期"]
|
||
assert payload["metadata"]["report_confidence_level"] == "高"
|
||
|
||
|
||
def test_build_payload_uses_result_fallbacks_and_optional_catalysts() -> None:
|
||
result = _result(confidence_level="低")
|
||
result.dashboard = {
|
||
"battle_plan": {
|
||
"sniper_points": {"ideal_buy": "1700"},
|
||
"action_checklist": ["等待回踩确认"],
|
||
},
|
||
"intelligence": {},
|
||
}
|
||
result.market_phase_summary = {"phase": "postmarket"}
|
||
result.analysis_context_pack_overview = {"data_quality": {"level": "limited"}}
|
||
|
||
payload = build_decision_signal_payload_from_report(
|
||
result,
|
||
context_snapshot=None,
|
||
source_report_id=None,
|
||
trace_id="trace-fallback",
|
||
query_source="",
|
||
report_type="simple",
|
||
)
|
||
|
||
assert payload is not None
|
||
assert payload["market_phase"] == "postmarket"
|
||
assert payload["data_quality_summary"] == {"level": "limited"}
|
||
assert payload["entry_low"] == 1700.0
|
||
assert "entry_high" not in payload
|
||
assert payload["watch_conditions"] == ["等待回踩确认"]
|
||
assert "catalyst_summary" not in payload
|
||
assert payload["trigger_source"] == "system"
|
||
assert payload["confidence"] == 0.4
|
||
|
||
|
||
def test_build_payload_maps_secondary_only_entry_to_entry_high() -> None:
|
||
result = _result()
|
||
result.dashboard = {
|
||
"battle_plan": {
|
||
"sniper_points": {"secondary_buy": "次优买入点：1680元"},
|
||
},
|
||
}
|
||
|
||
payload = build_decision_signal_payload_from_report(
|
||
result,
|
||
trace_id="trace-secondary-only",
|
||
query_source="api",
|
||
report_type="simple",
|
||
)
|
||
|
||
assert payload is not None
|
||
assert "entry_low" not in payload
|
||
assert payload["entry_high"] == 1680.0
|
||
|
||
|
||
def test_build_payload_reuses_shared_sniper_fallback_paths(isolated_db) -> None:
|
||
result = _result()
|
||
result.dashboard = {}
|
||
result.raw_response = {
|
||
"dashboard": {
|
||
"battle_plan": {
|
||
"sniper_points": {
|
||
"ideal_buy": "1690",
|
||
"secondary_buy": "1705",
|
||
"stop_loss": "1620",
|
||
"take_profit": "1880",
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
payload = build_decision_signal_payload_from_report(
|
||
result,
|
||
trace_id="trace-raw-sniper",
|
||
query_source="api",
|
||
report_type="simple",
|
||
)
|
||
stored_points = isolated_db._extract_sniper_points(result)
|
||
|
||
assert payload is not None
|
||
assert stored_points == {
|
||
"ideal_buy": 1690.0,
|
||
"secondary_buy": 1705.0,
|
||
"stop_loss": 1620.0,
|
||
"take_profit": 1880.0,
|
||
}
|
||
assert payload["entry_low"] == 1690.0
|
||
assert payload["entry_high"] == 1705.0
|
||
assert payload["stop_loss"] == 1620.0
|
||
assert payload["target_price"] == 1880.0
|
||
|
||
|
||
def test_build_payload_skips_ambiguous_action_non_stock_and_unknown_market() -> None:
|
||
ambiguous = _result(operation_advice="买盘增强，继续观察", action=None)
|
||
assert build_decision_signal_payload_from_report(
|
||
ambiguous,
|
||
trace_id="trace-1",
|
||
query_source="api",
|
||
report_type="simple",
|
||
) is None
|
||
|
||
market_review = _result(operation_advice="买入", action="buy")
|
||
assert build_decision_signal_payload_from_report(
|
||
market_review,
|
||
trace_id="trace-2",
|
||
query_source="api",
|
||
report_type="market_review",
|
||
) is None
|
||
|
||
unknown_market = _result(code="UNKNOWN", operation_advice="买入", action="buy")
|
||
assert build_decision_signal_payload_from_report(
|
||
unknown_market,
|
||
trace_id="trace-3",
|
||
query_source="api",
|
||
report_type="simple",
|
||
) is None
|
||
|
||
|
||
def test_extract_and_persist_reuses_service_dedup_and_sanitization(isolated_db) -> None:
|
||
service = DecisionSignalService(db_manager=isolated_db)
|
||
result = _result(
|
||
analysis_summary="趋势确认 token=super-secret",
|
||
)
|
||
|
||
first = extract_and_persist_from_analysis_result(
|
||
result,
|
||
context_snapshot={"market_phase_summary": {"phase": "intraday"}},
|
||
source_report_id=901,
|
||
trace_id="trace-901",
|
||
query_source="api",
|
||
report_type="full",
|
||
service=service,
|
||
)
|
||
second = extract_and_persist_from_analysis_result(
|
||
result,
|
||
context_snapshot={"market_phase_summary": {"phase": "intraday"}},
|
||
source_report_id=901,
|
||
trace_id="trace-901",
|
||
query_source="api",
|
||
report_type="full",
|
||
service=service,
|
||
)
|
||
|
||
assert first is not None
|
||
assert second is not None
|
||
assert first["created"] is True
|
||
assert second["created"] is False
|
||
assert first["item"]["reason"] == "趋势确认 token=[REDACTED]"
|
||
assert first["item"]["plan_quality"] == "complete"
|
||
|
||
listed = service.list_signals(source_report_id=901)
|
||
assert listed["total"] == 1
|
||
persisted = listed["items"][0]
|
||
assert persisted["source_report_id"] == 901
|
||
assert persisted["reason"] == "趋势确认 token=[REDACTED]"
|
||
assert persisted["entry_low"] == 1690.0
|
||
assert persisted["entry_high"] == 1700.0
|
||
|
||
|
||
def test_extract_and_persist_missing_price_plan_does_not_fabricate_fields(isolated_db) -> None:
|
||
service = DecisionSignalService(db_manager=isolated_db)
|
||
result = _result()
|
||
result.dashboard = {"battle_plan": {"sniper_points": {}}, "intelligence": {}}
|
||
|
||
created = extract_and_persist_from_analysis_result(
|
||
result,
|
||
context_snapshot={"market_phase_summary": {"phase": "postmarket"}},
|
||
source_report_id=902,
|
||
trace_id="trace-902",
|
||
query_source="schedule",
|
||
report_type="simple",
|
||
service=service,
|
||
)
|
||
|
||
assert created is not None
|
||
item = created["item"]
|
||
assert item["plan_quality"] == "minimal"
|
||
assert item["entry_low"] is None
|
||
assert item["entry_high"] is None
|
||
assert item["stop_loss"] is None
|
||
assert item["target_price"] is None
|