80 lines
2.2 KiB
Python
80 lines
2.2 KiB
Python
from datetime import datetime
|
|
|
|
from app.model.fund import FundRiskAlert
|
|
from app.service.risk_notification_service import RiskNotificationService
|
|
|
|
|
|
class FakeSession:
|
|
def __init__(self):
|
|
self.added = []
|
|
|
|
def add(self, value):
|
|
self.added.append(value)
|
|
|
|
|
|
def alert(level: str = "高") -> FundRiskAlert:
|
|
return FundRiskAlert(
|
|
id=1,
|
|
alert_no="ALERT-001",
|
|
customer_id=9,
|
|
alert_type="适当性错配",
|
|
alert_level=level,
|
|
trigger_rule_codes=["RW-007"],
|
|
evidence_summary="C2 客户购买 R5 产品",
|
|
evidence_snapshot={},
|
|
priority_score=90,
|
|
event_status="正在发生",
|
|
status="待处理",
|
|
ack_status="未确认",
|
|
is_escalated=0,
|
|
created_at=datetime(2026, 9, 10),
|
|
updated_at=datetime(2026, 9, 10),
|
|
)
|
|
|
|
|
|
def test_in_app_notification_contains_alert_number() -> None:
|
|
session = FakeSession()
|
|
|
|
notification = RiskNotificationService(session).create_in_app(
|
|
alert(),
|
|
receiver_user_id=990000002,
|
|
title="高风险预警",
|
|
content="C2 客户购买 R5 产品",
|
|
)
|
|
|
|
assert notification.alert_id == 1
|
|
assert notification.channel == "站内提醒"
|
|
assert "预警编号：ALERT-001" in notification.content
|
|
assert notification.id > 0
|
|
assert session.added == [notification]
|
|
|
|
|
|
def test_disabled_mail_only_creates_record() -> None:
|
|
session = FakeSession()
|
|
|
|
notification = RiskNotificationService(session).create_mail_record(
|
|
alert(),
|
|
receiver_email="risk@example.com",
|
|
title="高风险预警",
|
|
content="证据摘要",
|
|
mail_enabled=False,
|
|
)
|
|
|
|
assert notification.channel == "邮件"
|
|
assert notification.send_status == "未启用"
|
|
assert notification.fail_reason == "邮件发送功能未启用"
|
|
|
|
|
|
def test_high_risk_batch_creates_in_app_and_mail_records() -> None:
|
|
session = FakeSession()
|
|
|
|
records = RiskNotificationService(session).create_high_risk_records(
|
|
[alert("高"), alert("低")],
|
|
receiver_user_id=990000002,
|
|
receiver_email="risk@example.com",
|
|
mail_enabled=False,
|
|
)
|
|
|
|
assert len(records) == 2
|
|
assert {record.channel for record in records} == {"站内提醒", "邮件"}
|